mirror of
https://github.com/deepseek-ai/deepseek-harness
synced 2026-08-15 21:04:50 +00:00
72 lines
2.8 KiB
TypeScript
72 lines
2.8 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from 'vitest'
|
|
import { Context } from 'cordis'
|
|
import { mkdtemp, rm } from 'node:fs/promises'
|
|
import { tmpdir } from 'node:os'
|
|
import { join } from 'node:path'
|
|
import { credentialRef } from '@deepseek-ai/dsh-credentials'
|
|
import { CredentialsLocal } from '../src/index.ts'
|
|
|
|
// The atomic write is the gated asynchronous hold point inside a queued
|
|
// write; gating it makes the dispose-versus-queued-write race fully
|
|
// deterministic. The lock helper passes through so the gated operation still
|
|
// runs inside its real acquire/release cycle.
|
|
vi.mock('@deepseek-ai/dsh-atomic-write', async (importOriginal) => {
|
|
const actual = await importOriginal<typeof import('@deepseek-ai/dsh-atomic-write')>()
|
|
let gate: Promise<void> = Promise.resolve()
|
|
return {
|
|
...actual,
|
|
writeFileAtomic: vi.fn(() => gate),
|
|
__setGate: (next: Promise<void>) => {
|
|
gate = next
|
|
},
|
|
}
|
|
})
|
|
|
|
async function setGate(next: Promise<void>): Promise<void> {
|
|
const mocked = await import('@deepseek-ai/dsh-atomic-write') as unknown as { __setGate: (next: Promise<void>) => void }
|
|
mocked.__setGate(next)
|
|
}
|
|
|
|
const KEY = credentialRef('DSH_CRED_DRAIN_A')
|
|
const OTHER = credentialRef('DSH_CRED_DRAIN_B')
|
|
|
|
const cleanups: Array<() => Promise<void>> = []
|
|
|
|
afterEach(async () => {
|
|
await setGate(Promise.resolve())
|
|
while (cleanups.length > 0) await cleanups.pop()!()
|
|
})
|
|
|
|
describe('write-drain teardown', () => {
|
|
it('lets the in-flight write land and fails the queued one after disposal', async () => {
|
|
const dir = await mkdtemp(join(tmpdir(), 'dsh-credentials-drain-'))
|
|
cleanups.push(() => rm(dir, { recursive: true, force: true }))
|
|
const ctx = new Context()
|
|
const fiber = ctx.plugin(CredentialsLocal, { path: join(dir, '.env'), watch: false })
|
|
await fiber
|
|
const service = ctx.credentials
|
|
|
|
let release!: () => void
|
|
await setGate(new Promise<void>((resolveGate) => {
|
|
release = resolveGate
|
|
}))
|
|
const first = service.set(KEY, 'one')
|
|
// Let the first task pass its liveness checks and park on the gate, so it
|
|
// is genuinely in-flight when disposal begins.
|
|
await new Promise(resolvePause => setTimeout(resolvePause, 5))
|
|
// Attach the rejection handler up front: the queued write fails while the
|
|
// drain is still awaited, before any later `await expect` could run.
|
|
const secondRejects = expect(service.set(OTHER, 'two')).rejects.toThrow(/disposed before the queued/)
|
|
const disposal = fiber.dispose()
|
|
// Give the drain disposer its first turn (set closed) before opening the gate.
|
|
await new Promise(resolvePause => setTimeout(resolvePause, 10))
|
|
release()
|
|
await disposal
|
|
|
|
await expect(first).resolves.toBeUndefined()
|
|
await secondRejects
|
|
expect(await service.resolve(KEY)).toEqual({ value: 'one', source: 'file' })
|
|
expect(await service.resolve(OTHER)).toBeUndefined()
|
|
})
|
|
})
|