/** * Tests for the per-tool subpath plugins (`@deepseek-ai/dsh-tool-fs/read`, * `/write`, `/edit`): each registers exactly one tool, injects the same * services, and cleans up on disposal. */ import { describe, expect, it } from 'vitest' import { Context } from 'cordis' import SystemPrompt from '@deepseek-ai/dsh-system-prompt' import ToolRegistry from '@deepseek-ai/dsh-tools' import { FileSystem } from '@deepseek-ai/dsh-fs' import type { FsEditOutcome, FsReadOutcome, FsTarget, FsWriteOutcome, } from '@deepseek-ai/dsh-fs' import * as readPlugin from '@deepseek-ai/dsh-tool-fs/read' import * as writePlugin from '@deepseek-ai/dsh-tool-fs/write' import * as editPlugin from '@deepseek-ai/dsh-tool-fs/edit' class StubFs extends FileSystem { override async resolve(path: string): Promise { return { inputPath: path, targetKey: path, displayPath: path } } override async readPage(): Promise { return { offset: 1, limit: 1, lines: [], totalLines: 0, version: 'v', view: 'full' } } override async createOrReplace(): Promise { return { operation: 'create', version: 'v' } } override async applyEdit(): Promise { return { replacements: 1, replaceAll: false, version: 'v' } } } async function base() { const ctx = new Context() await ctx.plugin(SystemPrompt) await ctx.plugin(ToolRegistry) await ctx.plugin(StubFs) return ctx } describe('subpath plugins', () => { it('each registers exactly its one tool', async () => { const cases: Array<[unknown, string]> = [ [readPlugin, 'read'], [writePlugin, 'write'], [editPlugin, 'edit'], ] for (const [plugin, toolName] of cases) { const ctx = await base() await ctx.plugin(plugin as Parameters[0]) expect(ctx.tools.schemas().map(s => s.name)).toEqual([toolName]) } }) it('cleans up on disposal (HMR safety)', async () => { const ctx = await base() const fiber = await ctx.plugin(readPlugin as Parameters[0]) expect(ctx.tools.schemas()).toHaveLength(1) await fiber.dispose() expect(ctx.tools.schemas()).toHaveLength(0) }) it('stays pending without a ctx.fs provider', async () => { const ctx = new Context() await ctx.plugin(SystemPrompt) await ctx.plugin(ToolRegistry) await ctx.plugin(writePlugin as Parameters[0]) expect(ctx.tools.schemas()).toHaveLength(0) }) })