Initial open-source release

packages/context/src/wiki/knowledge-wiki.service.test.ts

@@ -0,0 +1,118 @@
+import { describe, expect, it, vi } from 'vitest';
+import { KnowledgeWikiService, type WikiFrontmatter } from './knowledge-wiki.service.js';
+
+function makeService() {
+  const pagesRepository: Record<string, ReturnType<typeof vi.fn>> = {
+    upsertPage: vi.fn().mockResolvedValue(undefined),
+    deleteByKey: vi.fn().mockResolvedValue(undefined),
+    deleteByScope: vi.fn().mockResolvedValue(undefined),
+    deleteStale: vi.fn().mockResolvedValue(undefined),
+    getExistingSearchTexts: vi.fn().mockResolvedValue(new Map()),
+    applyDiffTransactional: vi.fn().mockResolvedValue(undefined),
+  };
+  const embeddingService = {
+    computeEmbedding: vi.fn().mockResolvedValue([0.1, 0.2, 0.3]),
+    computeEmbeddingsBulk: vi.fn().mockResolvedValue([]),
+    maxBatchSize: 16,
+  };
+  const configService = {
+    forWorktree: vi.fn().mockReturnValue({
+      writeFile: vi.fn(),
+      readFile: vi.fn(),
+      deleteFile: vi.fn(),
+      listFiles: vi.fn(),
+      getFileHistory: vi.fn(),
+    }),
+    writeFile: vi.fn(),
+    readFile: vi.fn(),
+    deleteFile: vi.fn(),
+    listFiles: vi.fn(),
+    getFileHistory: vi.fn(),
+  };
+  const gitService = {
+    diffNameStatus: vi.fn().mockResolvedValue([]),
+    getFileAtCommit: vi.fn().mockResolvedValue(''),
+  };
+  const service = new KnowledgeWikiService(
+    configService as any,
+    embeddingService as any,
+    pagesRepository as any,
+    gitService as any,
+  );
+  return { service, pagesRepository, embeddingService, configService, gitService };
+}
+
+const fm: WikiFrontmatter = { summary: 'sum', usage_mode: 'auto' };
+
+describe('KnowledgeWikiService.forWorktree isolation', () => {
+  it('syncSinglePage in worktree scope does not call pagesRepository.upsertPage', async () => {
+    const { service, pagesRepository, embeddingService } = makeService();
+    const scoped = service.forWorktree('/tmp/fake-worktree');
+
+    await scoped.syncSinglePage('GLOBAL', null, 'key', fm, 'body');
+
+    expect(pagesRepository.upsertPage).not.toHaveBeenCalled();
+    expect(embeddingService.computeEmbedding).not.toHaveBeenCalled();
+  });
+
+  it('deleteFromIndex in worktree scope does not call pagesRepository.deleteByKey', async () => {
+    const { service, pagesRepository } = makeService();
+    const scoped = service.forWorktree('/tmp/fake-worktree');
+
+    await scoped.deleteFromIndex('GLOBAL', null, 'key');
+
+    expect(pagesRepository.deleteByKey).not.toHaveBeenCalled();
+  });
+
+  it('syncSinglePage in main scope still calls pagesRepository.upsertPage', async () => {
+    const { service, pagesRepository } = makeService();
+
+    await service.syncSinglePage('GLOBAL', null, 'key', fm, 'body');
+
+    expect(pagesRepository.upsertPage).toHaveBeenCalledTimes(1);
+  });
+});
+
+describe('KnowledgeWikiService.syncFromCommit', () => {
+  it('applies upserts for added/modified files and deletes for removed files in a single transactional batch', async () => {
+    const { service, pagesRepository, gitService } = makeService();
+
+    gitService.diffNameStatus.mockResolvedValue([
+      { status: 'A', path: 'knowledge/global/new-page.md' },
+      { status: 'M', path: 'knowledge/global/changed-page.md' },
+      { status: 'D', path: 'knowledge/global/gone-page.md' },
+    ]);
+    gitService.getFileAtCommit.mockImplementation((path: string) => {
+      if (path.endsWith('new-page.md')) {
+        return Promise.resolve('---\nsummary: new\nusage_mode: auto\n---\n\nbody-new\n');
+      }
+      if (path.endsWith('changed-page.md')) {
+        return Promise.resolve('---\nsummary: changed\nusage_mode: auto\n---\n\nbody-changed\n');
+      }
+      return Promise.reject(new Error(`unexpected getFileAtCommit path: ${path}`));
+    });
+
+    await service.syncFromCommit('sha-before', 'sha-after', 'run-uuid');
+
+    expect(pagesRepository.applyDiffTransactional).toHaveBeenCalledTimes(1);
+    const call = pagesRepository.applyDiffTransactional.mock.calls[0][0];
+    expect(call.runId).toBe('run-uuid');
+    expect(call.upserts).toHaveLength(2);
+    expect(call.upserts).toEqual(
+      expect.arrayContaining([
+        expect.objectContaining({ scope: 'GLOBAL', pageKey: 'new-page', summary: 'new' }),
+        expect.objectContaining({ scope: 'GLOBAL', pageKey: 'changed-page', summary: 'changed' }),
+      ]),
+    );
+    expect(call.deletes).toEqual([{ scope: 'GLOBAL', scopeId: null, pageKey: 'gone-page' }]);
+  });
+
+  it('is a no-op when the diff between shas has no knowledge changes', async () => {
+    const { service, pagesRepository, gitService } = makeService();
+    gitService.diffNameStatus.mockResolvedValue([]);
+
+    await service.syncFromCommit('sha-before', 'sha-after', 'run-uuid');
+
+    expect(pagesRepository.applyDiffTransactional).not.toHaveBeenCalled();
+  });
+});