test(agent-presets): prove the write override where it is introduced

The regression for `PresetTree.write` lived two layers up, so this layer's own
assertion could not fail: it checked the file after an ordinary teardown, and
the Loader's unload listener returns early when the whole tree is being
disposed, so the override never ran. Moves the self-disposing-row test down to
the layer that adds the override, in a temp root so a committed fixture cannot
be damaged by the run that proves the bug.
This commit is contained in:
Yichen Jiang
2026-08-07 15:04:23 +08:00
parent bdf5e39986
commit 6435463edc
3 changed files with 62 additions and 26 deletions

View File

@@ -17,9 +17,4 @@ export function apply(ctx, config) {
order: 10,
text: `section for ${config.tool}`,
}))
// Reconfiguring a live row runs the Loader's `internal/update` waterfall,
// which persists the owning tree. That is the trigger reaching the preset
// tree's `write` while the subtree is still mounted; tearing the agent down
// instead stops earlier, in the loader's own "tree is being disposed" case.
globalThis.__RECONFIGURE__ = tool => ctx.fiber.update({ ...config, tool })
}

View File

@@ -0,0 +1,9 @@
// Disposes itself once active. The Loader treats a self-disposing entry as a
// config change and writes the tree back through `EntryTree.write()`, which is
// the exact path that once truncated a preset file to `[]`.
export const name = 'self-dispose'
export function apply(ctx) {
globalThis.__SELF_DISPOSED__ = new Promise((resolve) => {
setTimeout(() => { ctx.fiber.dispose(); resolve(undefined) }, 0)
})
}

View File

@@ -1,4 +1,5 @@
import { readFile } from 'node:fs/promises'
import { mkdir, mkdtemp, readFile, writeFile } from 'node:fs/promises'
import { tmpdir } from 'node:os'
import { dirname, join } from 'node:path'
import { fileURLToPath, pathToFileURL } from 'node:url'
import { Context } from 'cordis'
@@ -11,7 +12,7 @@ import ToolRegistry from '@deepseek-ai/dsh-tools'
import AgentRegistry, { assembleContextFor, type Agent } from '@deepseek-ai/dsh-agent'
import AgentLoop from '@deepseek-ai/dsh-agent-loop'
import { beforeEach, describe, expect, it } from 'vitest'
import AgentPresets, { leakedServices, livePresetMounts } from '@deepseek-ai/dsh-agent-presets'
import AgentPresets, { COMPOSITION_FILE, leakedServices, livePresetMounts } from '@deepseek-ai/dsh-agent-presets'
declare module 'cordis' {
interface Context {
@@ -217,29 +218,60 @@ describe('a roster with nothing in it', () => {
})
})
describe('attributing a service to a subtree', () => {
it('never writes the preset file back, however the subtree changes', async () => {
delete (globalThis as { __RECONFIGURE__?: unknown }).__RECONFIGURE__
const handle = await ctx.agents.create({
sessionId: SessionId('sess-write'),
setup: async (agentCtx: Context) => void await ctx.agentPresets.mount(agentCtx, 'standard'),
describe('the preset file is an input, never a persistence target', () => {
it('survives a row that disposes itself, which makes the Loader persist a tree', async () => {
// The preset lives in a temp root, not under `fixtures/`: without the
// `write()` override the Loader REWRITES the composition it read, so a
// committed fixture would be mutated by the very run that proves the bug
// and every later run would compare against the damaged file and pass.
const root = await mkdtemp(join(tmpdir(), 'dsh-preset-write-'))
const dir = join(root, 'self-disposing')
await mkdir(dir)
const path = join(dir, COMPOSITION_FILE)
const composition = [
'- id: tool-kept',
` name: ${join(FIXTURES, 'plugins', 'contribute.js')}`,
' config:',
' tool: kept',
'- id: goes-away',
` name: ${join(FIXTURES, 'plugins', 'self-dispose.js')}`,
'',
].join('\n')
await writeFile(path, composition)
const scoped = new Context()
scoped.baseUrl = pathToFileURL(FIXTURES).href + '/'
await scoped.plugin(Loader)
scoped.loader.builtins.include = Include
await scoped.plugin(LlmService)
await scoped.plugin(SessionStore)
await scoped.plugin(SystemPrompt, { persona: '' })
await scoped.plugin(ToolRegistry)
await scoped.plugin(AgentRegistry)
await scoped.plugin(AgentLoop, { agents: [] })
await scoped.plugin(AgentPresets, { default: 'self-disposing', roots: [{ path: root, trust: 'user' as const }] })
await scoped.agents.create({
sessionId: SessionId('sess-self-dispose'),
setup: async (agentCtx: Context) => void await scoped.agentPresets.mount(agentCtx),
})
const file = join(FIXTURES, 'system', 'standard', 'agent.cordis.yml')
const before = await readFile(file, 'utf8')
await (globalThis as { __SELF_DISPOSED__?: Promise<unknown> }).__SELF_DISPOSED__
// Slack past the deterministic signal above, not a race the number has to
// win. The write rides the Loader's fiber-unload listener, which stamps
// `disabled: true` and calls `write()` in the same synchronous step; once
// the self-dispose has settled, a regression has already written. Polling
// would not help — the assertion is an ABSENCE, and no amount of waiting
// proves one — so the wait only has to clear settlement.
await new Promise(resolve => setTimeout(resolve, 50))
// The inherited `write()` persists the whole tree whenever the Loader
// decides a row's config moved, so one row reconfiguring itself would
// rewrite the shipped composition — here, with the row's new tool name.
const reconfigure = (globalThis as { __RECONFIGURE__?: (tool: string) => Promise<void> }).__RECONFIGURE__
expect(reconfigure).toBeTypeOf('function')
await reconfigure!('rewritten')
expect(toolNames(ctx, handle.agent)).toContain('rewritten')
expect(await readFile(file, 'utf8')).toBe(before)
await handle.dispose()
// Inherited, `EntryTree.write()` persists the dying tree — stamping
// `disabled: true` onto the row and, in the shipped case, truncating the
// composition every session shares.
expect(await readFile(path, 'utf8')).toBe(composition)
})
})
describe('attributing a service to a subtree', () => {
it('attributes nothing to a subtree that is already torn down', async () => {
const handle = await ctx.agents.create({
sessionId: SessionId('sess-torn'),