mirror of
https://github.com/deepseek-ai/deepseek-harness
synced 2026-08-15 21:04:50 +00:00
97 lines
3.7 KiB
TypeScript
97 lines
3.7 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
import { Context } from 'cordis'
|
|
import { AgentMessageId, type Agent } from '@deepseek-ai/dsh-agent'
|
|
import * as AgentInvariant from '@deepseek-ai/dsh-agent/invariant'
|
|
import { scopeTarget } from '@deepseek-ai/dsh-scope'
|
|
import InvariantService from '@deepseek-ai/dsh-invariants'
|
|
|
|
async function setup(): Promise<Context> {
|
|
const ctx = new Context()
|
|
await ctx.plugin(InvariantService)
|
|
await ctx.plugin(AgentInvariant)
|
|
return ctx
|
|
}
|
|
|
|
function mockAgent(id: string): Agent {
|
|
return { id } as unknown as Agent
|
|
}
|
|
|
|
describe('agent status invariants', () => {
|
|
it('accepts lifecycle transitions through idle, running, and disposed', async () => {
|
|
const ctx = await setup()
|
|
const agent = mockAgent('a1')
|
|
expect(() => {
|
|
ctx.emit(scopeTarget(agent, agent), 'agent/status', agent, 'idle')
|
|
ctx.emit(scopeTarget(agent, agent), 'agent/status', agent, 'running')
|
|
ctx.emit(scopeTarget(agent, agent), 'agent/status', agent, 'idle')
|
|
ctx.emit(scopeTarget(agent, agent), 'agent/status', agent, 'disposed')
|
|
}).not.toThrow()
|
|
|
|
const running = mockAgent('a2')
|
|
ctx.emit(scopeTarget(running, running), 'agent/status', running, 'running')
|
|
expect(() => { ctx.emit(scopeTarget(running, running), 'agent/status', running, 'disposed') }).not.toThrow()
|
|
})
|
|
|
|
it('rejects a no-op transition', async () => {
|
|
const ctx = await setup()
|
|
const agent = mockAgent('a3')
|
|
ctx.emit(scopeTarget(agent, agent), 'agent/status', agent, 'running')
|
|
expect(() => { ctx.emit(scopeTarget(agent, agent), 'agent/status', agent, 'running') })
|
|
.toThrow(/no-op transition/)
|
|
})
|
|
|
|
it('rejects leaving the terminal disposed state', async () => {
|
|
const ctx = await setup()
|
|
const agent = mockAgent('a4')
|
|
ctx.emit(scopeTarget(agent, agent), 'agent/status', agent, 'disposed')
|
|
expect(() => { ctx.emit(scopeTarget(agent, agent), 'agent/status', agent, 'idle') })
|
|
.toThrow(/left terminal state disposed/)
|
|
})
|
|
|
|
it('tracks agents independently', async () => {
|
|
const ctx = await setup()
|
|
const a = mockAgent('a5')
|
|
const b = mockAgent('b5')
|
|
ctx.emit(scopeTarget(a, a), 'agent/status', a, 'running')
|
|
expect(() => { ctx.emit(scopeTarget(b, b), 'agent/status', b, 'running') }).not.toThrow()
|
|
})
|
|
})
|
|
|
|
describe('agent inbox invariants', () => {
|
|
const info = () => ({ id: AgentMessageId('m'), content: [], source: { kind: 'user' as const } })
|
|
|
|
it('accepts a dequeue and a discard covered by prior enqueues', async () => {
|
|
const ctx = await setup()
|
|
const agent = mockAgent('i1')
|
|
const at = scopeTarget(agent, agent)
|
|
expect(() => {
|
|
ctx.emit(at, 'agent/inbox/enqueue', agent, info())
|
|
ctx.emit(at, 'agent/inbox/enqueue', agent, info())
|
|
ctx.emit(at, 'agent/inbox/dequeue', agent, info())
|
|
ctx.emit(at, 'agent/inbox/discard', agent, [info()])
|
|
}).not.toThrow()
|
|
})
|
|
|
|
it('rejects a dequeue with no outstanding item', async () => {
|
|
const ctx = await setup()
|
|
const agent = mockAgent('i2')
|
|
expect(() => { ctx.emit(scopeTarget(agent, agent), 'agent/inbox/dequeue', agent, info()) })
|
|
.toThrow(/without a matching prior enqueue/)
|
|
})
|
|
|
|
it('rejects a discard larger than the outstanding count', async () => {
|
|
const ctx = await setup()
|
|
const agent = mockAgent('i3')
|
|
const at = scopeTarget(agent, agent)
|
|
ctx.emit(at, 'agent/inbox/enqueue', agent, info())
|
|
expect(() => { ctx.emit(at, 'agent/inbox/discard', agent, [info(), info()]) })
|
|
.toThrow(/dropped 2 items but only 1 were outstanding/)
|
|
})
|
|
|
|
it('accepts an empty discard against a fresh agent', async () => {
|
|
const ctx = await setup()
|
|
const agent = mockAgent('i4')
|
|
expect(() => { ctx.emit(scopeTarget(agent, agent), 'agent/inbox/discard', agent, []) }).not.toThrow()
|
|
})
|
|
})
|