@kaelio/ktx 0.5.0 → 0.7.0

package/dist/context/mcp/server.test.js

@@ -1,902 +0,0 @@
-import { access, mkdtemp, readFile, rm } from 'node:fs/promises';
-import { tmpdir } from 'node:os';
-import { join } from 'node:path';
-import { Client } from '@modelcontextprotocol/sdk/client/index.js';
-import { InMemoryTransport } from '@modelcontextprotocol/sdk/inMemory.js';
-import { afterEach, describe, expect, it, vi } from 'vitest';
-import { createLocalProjectMemoryIngest } from '../../context/memory/local-memory.js';
-import { detectCaptureSignals } from '../../context/memory/capture-signals.js';
-import { initKtxProject } from '../../context/project/project.js';
-import { jsonToolResult } from './context-tools.js';
-import { createDefaultKtxMcpServer, createKtxMcpServer } from './server.js';
-function makeFakeServer() {
-    const tools = [];
-    return {
-        tools,
-        server: {
-            registerTool(name, config, handler) {
-                tools.push({ name, config, handler });
-            },
-        },
-    };
-}
-function makeIo() {
-    let stderr = '';
-    return {
-        stdout: { isTTY: true, write() { } },
-        stderr: {
-            write(chunk) {
-                stderr += chunk;
-            },
-        },
-        stderrText: () => stderr,
-    };
-}
-function getTool(tools, name) {
-    const found = tools.find((tool) => tool.name === name);
-    if (!found) {
-        throw new Error(`Tool not registered: ${name}`);
-    }
-    return found;
-}
-const retainedToolNames = [
-    'connection_list',
-    'dictionary_search',
-    'discover_data',
-    'entity_details',
-    'memory_ingest',
-    'memory_ingest_status',
-    'sl_query',
-    'sl_read_source',
-    'sql_execution',
-    'wiki_read',
-    'wiki_search',
-];
-function makeAllContextTools() {
-    return {
-        connections: {
-            list: vi.fn().mockResolvedValue([{ id: 'warehouse', name: 'Warehouse', connectionType: 'POSTGRES' }]),
-        },
-        knowledge: {
-            search: vi.fn().mockResolvedValue({ results: [], totalFound: 0 }),
-            read: vi.fn().mockResolvedValue({
-                key: 'revenue',
-                summary: 'Paid order value',
-                content: '# Revenue',
-                scope: 'GLOBAL',
-                tags: ['finance'],
-                refs: [],
-                slRefs: ['orders'],
-            }),
-        },
-        semanticLayer: {
-            readSource: vi.fn().mockResolvedValue({
-                sourceName: 'orders',
-                yaml: 'name: orders\n',
-            }),
-            query: vi.fn().mockResolvedValue({
-                sql: 'select 1',
-                headers: ['count'],
-                rows: [[1]],
-                totalRows: 1,
-                plan: { sources: ['orders'] },
-            }),
-        },
-        entityDetails: {
-            read: vi.fn().mockResolvedValue({ results: [] }),
-        },
-        dictionarySearch: {
-            search: vi.fn().mockResolvedValue({ searched: [], results: [] }),
-        },
-        discover: {
-            search: vi.fn().mockResolvedValue([]),
-        },
-        sqlExecution: {
-            execute: vi.fn().mockResolvedValue({
-                headers: ['count'],
-                headerTypes: ['integer'],
-                rows: [[1]],
-                rowCount: 1,
-            }),
-        },
-        memoryIngest: {
-            ingest: vi.fn().mockResolvedValue({ runId: 'run-1' }),
-            status: vi.fn().mockResolvedValue({
-                runId: 'run-1',
-                status: 'done',
-                stage: 'done',
-                done: true,
-                captured: { wiki: [], sl: [], xrefs: [] },
-                error: null,
-                commitHash: null,
-                skillsLoaded: [],
-                signalDetected: false,
-            }),
-        },
-    };
-}
-async function listToolsThroughSdk(contextTools) {
-    const server = createDefaultKtxMcpServer({
-        name: 'ktx-test',
-        version: '0.0.0-test',
-        userContext: { userId: 'mcp-user' },
-        contextTools,
-    });
-    const client = new Client({ name: 'ktx-test-client', version: '0.0.0-test' });
-    const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair();
-    await Promise.all([server.connect(serverTransport), client.connect(clientTransport)]);
-    try {
-        return await client.listTools();
-    }
-    finally {
-        await client.close();
-        await server.close();
-    }
-}
-describe('createKtxMcpServer', () => {
-    afterEach(() => {
-        vi.unstubAllEnvs();
-        vi.restoreAllMocks();
-    });
-    it('registers annotations and output schemas for every retained tool', async () => {
-        const fake = makeFakeServer();
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'mcp-user' },
-            contextTools: makeAllContextTools(),
-        });
-        expect(fake.tools.map((tool) => tool.name).sort()).toEqual([...retainedToolNames].sort());
-        const expectedAnnotations = {
-            connection_list: { title: 'Connection List', readOnlyHint: true, idempotentHint: true, openWorldHint: false },
-            discover_data: { title: 'Discover Data', readOnlyHint: true, openWorldHint: false },
-            wiki_search: { title: 'Wiki Search', readOnlyHint: true, openWorldHint: false },
-            wiki_read: { title: 'Wiki Read', readOnlyHint: true, idempotentHint: true, openWorldHint: false },
-            entity_details: { title: 'Entity Details', readOnlyHint: true, idempotentHint: true, openWorldHint: false },
-            dictionary_search: { title: 'Dictionary Search', readOnlyHint: true, openWorldHint: false },
-            sl_read_source: {
-                title: 'Semantic Layer Read Source',
-                readOnlyHint: true,
-                idempotentHint: true,
-                openWorldHint: false,
-            },
-            sl_query: { title: 'Semantic Layer Query', readOnlyHint: true, openWorldHint: false },
-            sql_execution: { title: 'SQL Execution', readOnlyHint: true, openWorldHint: false },
-            memory_ingest: { title: 'Memory Ingest', destructiveHint: true, openWorldHint: false },
-            memory_ingest_status: { title: 'Memory Ingest Status', readOnlyHint: true, openWorldHint: false },
-        };
-        for (const toolName of retainedToolNames) {
-            const tool = getTool(fake.tools, toolName);
-            expect(tool.config.title).toBe(expectedAnnotations[toolName]?.title);
-            expect(tool.config.annotations).toEqual(expectedAnnotations[toolName]);
-            expect(tool.config.outputSchema).toBeDefined();
-            const inputShape = tool.config.inputSchema;
-            for (const inputSchema of Object.values(inputShape)) {
-                expect(inputSchema.description).toEqual(expect.any(String));
-            }
-        }
-    });
-    it('exposes annotations and output schemas through the SDK tools/list response', async () => {
-        const result = await listToolsThroughSdk(makeAllContextTools());
-        const toolNames = result.tools.map((tool) => tool.name).sort();
-        expect(toolNames).toEqual([...retainedToolNames].sort());
-        await expect(`${JSON.stringify(result.tools, null, 2)}\n`).toMatchFileSnapshot('__snapshots__/mcp-tools-list.json');
-    });
-    it('registers context tools without memory capture tools when memory capture is omitted', async () => {
-        const fake = makeFakeServer();
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'local-user' },
-            contextTools: {
-                connections: {
-                    async list() {
-                        return [{ id: 'warehouse', name: 'warehouse', connectionType: 'postgres' }];
-                    },
-                },
-            },
-        });
-        expect(fake.tools.map((tool) => tool.name)).toEqual(['connection_list']);
-        await expect(getTool(fake.tools, 'connection_list').handler({})).resolves.toMatchObject({
-            structuredContent: {
-                connections: [{ id: 'warehouse', name: 'warehouse', connectionType: 'postgres' }],
-            },
-        });
-    });
-    it('emits sampled debug telemetry for MCP tool requests', async () => {
-        vi.spyOn(Math, 'random').mockReturnValue(0);
-        vi.stubEnv('KTX_TELEMETRY_DEBUG', '1');
-        vi.stubEnv('CI', '');
-        const fake = makeFakeServer();
-        const io = makeIo();
-        const projectDir = '/tmp/ktx-mcp-telemetry';
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'local-user' },
-            projectDir,
-            io,
-            contextTools: {
-                knowledge: {
-                    search: vi.fn().mockResolvedValue({ results: [], totalFound: 0 }),
-                    read: vi.fn().mockResolvedValue(null),
-                },
-            },
-        });
-        await expect(getTool(fake.tools, 'wiki_search').handler({ query: 'revenue recognition', limit: 5 })).resolves.toMatchObject({
-            structuredContent: { results: [], totalFound: 0 },
-        });
-        expect(io.stderrText()).toContain('"event":"mcp_request_completed"');
-        expect(io.stderrText()).toContain('"toolName":"wiki_search"');
-        expect(io.stderrText()).toContain('"sampleRate":0.1');
-        expect(io.stderrText()).not.toContain(projectDir);
-    });
-    it('registers parser-gated sql_execution when the host provides a SQL execution port', async () => {
-        const fake = makeFakeServer();
-        const response = {
-            headers: ['status', 'count'],
-            headerTypes: ['text', 'bigint'],
-            rows: [['paid', 42]],
-            rowCount: 1,
-        };
-        const sqlExecution = {
-            execute: vi.fn().mockResolvedValue(response),
-        };
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'local-user' },
-            contextTools: {
-                sqlExecution,
-            },
-        });
-        expect(fake.tools.map((tool) => tool.name)).toEqual(['sql_execution']);
-        await expect(getTool(fake.tools, 'sql_execution').handler({
-            connectionId: 'warehouse',
-            sql: 'select status, count(*) from public.orders group by status',
-            maxRows: 50,
-        })).resolves.toEqual({
-            content: [
-                {
-                    type: 'text',
-                    text: JSON.stringify({
-                        headers: ['status', 'count'],
-                        headerTypes: ['text', 'bigint'],
-                        rows: [['paid', 42]],
-                        rowCount: 1,
-                    }, null, 2),
-                },
-            ],
-            structuredContent: {
-                headers: ['status', 'count'],
-                headerTypes: ['text', 'bigint'],
-                rows: [['paid', 42]],
-                rowCount: 1,
-            },
-        });
-        expect(sqlExecution.execute).toHaveBeenCalledWith({
-            connectionId: 'warehouse',
-            sql: 'select status, count(*) from public.orders group by status',
-            maxRows: 50,
-        }, undefined);
-    });
-    it('registers entity_details when the host provides an entity-details port', async () => {
-        const fake = makeFakeServer();
-        const entityDetails = {
-            read: vi.fn().mockResolvedValue({
-                results: [
-                    {
-                        ok: true,
-                        connectionId: 'warehouse',
-                        tableRef: { catalog: null, db: 'public', name: 'orders' },
-                        display: 'public.orders',
-                        kind: 'table',
-                        comment: 'Customer orders',
-                        estimatedRows: 12,
-                        columns: [
-                            {
-                                name: 'id',
-                                nativeType: 'integer',
-                                normalizedType: 'integer',
-                                dimensionType: 'number',
-                                nullable: false,
-                                primaryKey: true,
-                                comment: null,
-                            },
-                        ],
-                        foreignKeys: [],
-                        snapshot: {
-                            syncId: 'sync-1',
-                            extractedAt: '2026-05-14T09:00:00.000Z',
-                            scanRunId: 'scan-1',
-                        },
-                    },
-                ],
-            }),
-        };
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'local-user' },
-            contextTools: { entityDetails },
-        });
-        expect(fake.tools.map((tool) => tool.name)).toEqual(['entity_details']);
-        await expect(getTool(fake.tools, 'entity_details').handler({
-            connectionId: 'warehouse',
-            entities: [{ table: 'public.orders', columns: ['id'] }],
-        })).resolves.toMatchObject({
-            structuredContent: {
-                results: [
-                    {
-                        ok: true,
-                        connectionId: 'warehouse',
-                        display: 'public.orders',
-                        columns: [{ name: 'id' }],
-                    },
-                ],
-            },
-        });
-        expect(entityDetails.read).toHaveBeenCalledWith({
-            connectionId: 'warehouse',
-            entities: [{ table: 'public.orders', columns: ['id'] }],
-        });
-    });
-    it('registers dictionary_search when the host provides a dictionary-search port', async () => {
-        const fake = makeFakeServer();
-        const dictionarySearch = {
-            search: vi.fn().mockResolvedValue({
-                searched: [
-                    {
-                        connectionId: 'warehouse',
-                        coverage: {
-                            sampledRows: null,
-                            valuesPerColumn: null,
-                            profiledColumns: 1,
-                            syncId: 'sync-1',
-                            profiledAt: null,
-                        },
-                        status: 'ready',
-                    },
-                ],
-                results: [
-                    {
-                        value: 'paid',
-                        matches: [
-                            {
-                                connectionId: 'warehouse',
-                                sourceName: 'orders',
-                                columnName: 'status',
-                                matchedValue: 'paid',
-                                cardinality: 3,
-                            },
-                        ],
-                        misses: [],
-                    },
-                ],
-            }),
-        };
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'local-user' },
-            contextTools: { dictionarySearch },
-        });
-        expect(fake.tools.map((tool) => tool.name)).toEqual(['dictionary_search']);
-        await expect(getTool(fake.tools, 'dictionary_search').handler({
-            connectionId: 'warehouse',
-            values: ['paid'],
-        })).resolves.toMatchObject({
-            structuredContent: {
-                searched: [{ connectionId: 'warehouse', status: 'ready' }],
-                results: [
-                    {
-                        value: 'paid',
-                        matches: [{ connectionId: 'warehouse', sourceName: 'orders', columnName: 'status' }],
-                        misses: [],
-                    },
-                ],
-            },
-        });
-        expect(dictionarySearch.search).toHaveBeenCalledWith({
-            connectionId: 'warehouse',
-            values: ['paid'],
-        });
-    });
-    it('sl_query normalizes order_by from cube-style {id, desc} and bare strings to {field, direction}', async () => {
-        const fake = makeFakeServer();
-        const semanticLayer = {
-            readSource: vi.fn(),
-            query: vi.fn().mockResolvedValue({
-                sql: '',
-                headers: [],
-                rows: [],
-                totalRows: 0,
-            }),
-        };
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'local-user' },
-            contextTools: { semanticLayer },
-        });
-        await getTool(fake.tools, 'sl_query').handler({
-            connectionId: 'warehouse',
-            measures: ['orders.count'],
-            order_by: [
-                { field: 'orders.total', direction: 'desc' },
-                { id: 'orders.quarter_label', desc: false },
-                { id: 'orders.created_at', desc: true },
-                'orders.segment',
-            ],
-        });
-        expect(semanticLayer.query).toHaveBeenCalledWith({
-            connectionId: 'warehouse',
-            query: expect.objectContaining({
-                order_by: [
-                    { field: 'orders.total', direction: 'desc' },
-                    { field: 'orders.quarter_label', direction: 'asc' },
-                    { field: 'orders.created_at', direction: 'desc' },
-                    { field: 'orders.segment', direction: 'asc' },
-                ],
-            }),
-        }, undefined);
-    });
-    it('sl_query normalizes cube-style dimensions to field dimensions', async () => {
-        const fake = makeFakeServer();
-        const semanticLayer = makeAllContextTools().semanticLayer;
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'local-user' },
-            contextTools: { semanticLayer },
-        });
-        await getTool(fake.tools, 'sl_query').handler({
-            connectionId: 'warehouse',
-            measures: ['orders.count'],
-            dimensions: [{ dimension: 'orders.created_at', granularity: 'month' }, 'orders.status'],
-        });
-        expect(semanticLayer.query).toHaveBeenCalledWith({
-            connectionId: 'warehouse',
-            query: expect.objectContaining({
-                dimensions: [{ field: 'orders.created_at', granularity: 'month' }, { field: 'orders.status' }],
-            }),
-        }, undefined);
-    });
-    it('entity_details normalizes sql-style schema table refs', async () => {
-        const fake = makeFakeServer();
-        const entityDetails = makeAllContextTools().entityDetails;
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'local-user' },
-            contextTools: { entityDetails },
-        });
-        await getTool(fake.tools, 'entity_details').handler({
-            connectionId: 'warehouse',
-            entities: [{ table: { schema: 'public', table: 'orders' }, columns: ['id'] }],
-        });
-        expect(entityDetails.read).toHaveBeenCalledWith({
-            connectionId: 'warehouse',
-            entities: [{ table: { catalog: null, db: 'public', name: 'orders' }, columns: ['id'] }],
-        });
-    });
-    it('wraps handler exceptions in-band for non-sql tools', async () => {
-        const fake = makeFakeServer();
-        const knowledge = {
-            search: vi.fn().mockRejectedValue(new Error('wiki index unavailable')),
-            read: vi.fn(),
-        };
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'local-user' },
-            contextTools: { knowledge },
-        });
-        await expect(getTool(fake.tools, 'wiki_search').handler({ query: 'revenue' })).resolves.toEqual({
-            content: [{ type: 'text', text: 'wiki index unavailable' }],
-            isError: true,
-        });
-    });
-    it('wires sql_execution progress to MCP notifications when a progress token is present', async () => {
-        const fake = makeFakeServer();
-        const notifications = [];
-        const sqlExecution = {
-            execute: vi.fn().mockImplementation(async (_input, options) => {
-                await options?.onProgress?.({ progress: 0, message: 'Validating SQL' });
-                await options?.onProgress?.({ progress: 0.3, message: 'Executing' });
-                await options?.onProgress?.({ progress: 1, message: 'Fetched 1 rows' });
-                return { headers: ['count'], rows: [[1]], rowCount: 1 };
-            }),
-        };
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'local-user' },
-            contextTools: { sqlExecution },
-        });
-        await getTool(fake.tools, 'sql_execution').handler({ connectionId: 'warehouse', sql: 'select 1' }, {
-            _meta: { progressToken: 'progress-1' },
-            sendNotification: async (notification) => {
-                notifications.push(notification);
-            },
-        });
-        expect(notifications).toEqual([
-            {
-                method: 'notifications/progress',
-                params: { progressToken: 'progress-1', progress: 0, message: 'Validating SQL' },
-            },
-            {
-                method: 'notifications/progress',
-                params: { progressToken: 'progress-1', progress: 0.3, message: 'Executing' },
-            },
-            {
-                method: 'notifications/progress',
-                params: { progressToken: 'progress-1', progress: 1, message: 'Fetched 1 rows' },
-            },
-        ]);
-    });
-    it('registers discover_data when the host provides a discover port', async () => {
-        const fake = makeFakeServer();
-        const discover = {
-            search: vi.fn().mockResolvedValue([
-                {
-                    kind: 'table',
-                    id: 'public.orders',
-                    score: 1,
-                    summary: 'Orders table',
-                    snippet: 'id, status',
-                    matchedOn: 'name',
-                    connectionId: 'warehouse',
-                    tableRef: { catalog: null, db: 'public', name: 'orders' },
-                },
-            ]),
-        };
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'local-user' },
-            contextTools: { discover },
-        });
-        expect(fake.tools.map((tool) => tool.name)).toEqual(['discover_data']);
-        await expect(getTool(fake.tools, 'discover_data').handler({
-            query: 'orders',
-            connectionId: 'warehouse',
-            kinds: ['table'],
-            limit: 5,
-        })).resolves.toMatchObject({
-            structuredContent: {
-                refs: [
-                    {
-                        kind: 'table',
-                        id: 'public.orders',
-                        connectionId: 'warehouse',
-                        tableRef: { catalog: null, db: 'public', name: 'orders' },
-                    },
-                ],
-            },
-        });
-        expect(discover.search).toHaveBeenCalledWith({
-            query: 'orders',
-            connectionId: 'warehouse',
-            kinds: ['table'],
-            limit: 5,
-        });
-    });
-    it('registers memory ingest tools through the context tool surface', async () => {
-        const fake = makeFakeServer();
-        let receivedInput;
-        const ingest = {
-            ingest: vi.fn().mockImplementation(async (input) => {
-                receivedInput = input;
-                return { runId: 'run-1' };
-            }),
-            status: vi.fn().mockResolvedValue({
-                runId: 'run-1',
-                status: 'done',
-                stage: 'done',
-                done: true,
-                captured: { wiki: ['revenue'], sl: [], xrefs: [] },
-                error: null,
-                commitHash: 'abc123',
-                skillsLoaded: ['wiki_capture'],
-                signalDetected: true,
-            }),
-        };
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'mcp-user' },
-            contextTools: { memoryIngest: ingest },
-        });
-        expect(fake.tools.map((tool) => tool.name).sort()).toEqual(['memory_ingest', 'memory_ingest_status']);
-        const content = [
-            'view: orders {',
-            '  sql_table_name: public.orders ;;',
-            '  measure: gross_revenue {',
-            '    type: sum',
-            '    sql: ${TABLE}.gross_revenue_cents ;;',
-            '  }',
-            '}',
-        ].join('\n');
-        const memoryIngest = getTool(fake.tools, 'memory_ingest');
-        await expect(memoryIngest.handler({
-            content,
-            connectionId: '00000000-0000-4000-8000-000000000001',
-        })).resolves.toEqual({
-            content: [{ type: 'text', text: JSON.stringify({ runId: 'run-1' }, null, 2) }],
-            structuredContent: { runId: 'run-1' },
-        });
-        expect(ingest.ingest).toHaveBeenCalledWith({
-            userId: 'mcp-user',
-            chatId: expect.stringMatching(/^mcp-/),
-            userMessage: 'Ingest external knowledge into KTX memory.',
-            assistantMessage: content,
-            connectionId: '00000000-0000-4000-8000-000000000001',
-            sourceType: 'external_ingest',
-        });
-        const cliEquivalentInput = {
-            userId: 'mcp-user',
-            chatId: 'cli-text-ingest-test-1',
-            userMessage: 'Ingest external text artifact "orders lookml" into KTX memory.',
-            assistantMessage: content,
-            connectionId: '00000000-0000-4000-8000-000000000001',
-            sourceType: 'external_ingest',
-        };
-        expect(detectCaptureSignals(receivedInput)).toEqual(detectCaptureSignals(cliEquivalentInput));
-        const memoryStatus = getTool(fake.tools, 'memory_ingest_status');
-        await expect(memoryStatus.handler({ runId: 'run-1' })).resolves.toEqual({
-            content: [
-                {
-                    type: 'text',
-                    text: JSON.stringify({
-                        runId: 'run-1',
-                        status: 'done',
-                        stage: 'done',
-                        done: true,
-                        captured: { wiki: ['revenue'], sl: [], xrefs: [] },
-                        error: null,
-                        commitHash: 'abc123',
-                        skillsLoaded: ['wiki_capture'],
-                        signalDetected: true,
-                    }, null, 2),
-                },
-            ],
-            structuredContent: {
-                runId: 'run-1',
-                status: 'done',
-                stage: 'done',
-                done: true,
-                captured: { wiki: ['revenue'], sl: [], xrefs: [] },
-                error: null,
-                commitHash: 'abc123',
-                skillsLoaded: ['wiki_capture'],
-                signalDetected: true,
-            },
-        });
-    });
-    it('returns an in-band error when a memory ingest run is missing', async () => {
-        const fake = makeFakeServer();
-        const ingest = {
-            ingest: vi.fn(),
-            status: vi.fn().mockResolvedValue(null),
-        };
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'mcp-user' },
-            contextTools: { memoryIngest: ingest },
-        });
-        const memoryStatus = getTool(fake.tools, 'memory_ingest_status');
-        await expect(memoryStatus.handler({ runId: 'missing-run' })).resolves.toEqual({
-            content: [{ type: 'text', text: 'Memory ingest run "missing-run" was not found.' }],
-            isError: true,
-        });
-    });
-    it('runs MCP memory_ingest against a local project memory port', async () => {
-        const tempDir = await mkdtemp(join(tmpdir(), 'ktx-mcp-local-memory-'));
-        try {
-            const project = await initKtxProject({ projectDir: tempDir });
-            const agentRunner = {
-                runLoop: async ({ toolSet, }) => {
-                    await toolSet.load_skill.execute({ name: 'wiki_capture' });
-                    await toolSet.wiki_write.execute({
-                        key: 'arr',
-                        summary: 'ARR definition',
-                        content: 'ARR means annual recurring revenue.',
-                    }, { toolCallId: 'wiki-write' });
-                    return { stopReason: 'natural' };
-                },
-            };
-            const memoryIngest = createLocalProjectMemoryIngest(project, {
-                agentRunner: agentRunner,
-                runIdFactory: () => 'memory-run-mcp',
-            });
-            const ingestSpy = vi.spyOn(memoryIngest, 'ingest');
-            const fake = makeFakeServer();
-            createKtxMcpServer({
-                server: fake.server,
-                userContext: { userId: 'local' },
-                contextTools: { memoryIngest },
-            });
-            const capture = await getTool(fake.tools, 'memory_ingest').handler({
-                content: 'Revenue means paid order value.',
-                connectionId: 'warehouse',
-            });
-            expect(capture).toMatchObject({
-                structuredContent: { runId: 'memory-run-mcp' },
-            });
-            await memoryIngest.waitForRun('memory-run-mcp');
-            expect(ingestSpy).toHaveBeenCalledWith({
-                userId: 'local',
-                chatId: expect.stringMatching(/^mcp-/),
-                userMessage: 'Ingest external knowledge into KTX memory.',
-                assistantMessage: 'Revenue means paid order value.',
-                connectionId: 'warehouse',
-                sourceType: 'external_ingest',
-            });
-            await expect(getTool(fake.tools, 'memory_ingest_status').handler({ runId: 'memory-run-mcp' })).resolves.toMatchObject({
-                structuredContent: {
-                    runId: 'memory-run-mcp',
-                    status: 'done',
-                    done: true,
-                    captured: { wiki: ['arr'], sl: [], xrefs: [] },
-                },
-            });
-            await expect(access(join(project.projectDir, '.ktx/db.sqlite'))).resolves.toBeUndefined();
-            await expect(access(join(project.projectDir, '.ktx/memory-runs/memory-run-mcp.json'))).rejects.toThrow();
-            await expect(readFile(join(project.projectDir, 'wiki/global/arr.md'), 'utf-8')).resolves.toContain('ARR means annual recurring revenue.');
-        }
-        finally {
-            await rm(tempDir, { recursive: true, force: true });
-        }
-    });
-    it('registers KTX context MCP tools when context ports are supplied', async () => {
-        const fake = makeFakeServer();
-        const contextTools = {
-            connections: {
-                list: vi.fn().mockResolvedValue([
-                    {
-                        id: '00000000-0000-4000-8000-000000000001',
-                        name: 'Warehouse',
-                        connectionType: 'POSTGRES',
-                    },
-                ]),
-            },
-            knowledge: {
-                search: vi.fn().mockResolvedValue({
-                    results: [
-                        {
-                            key: 'revenue',
-                            path: 'wiki/global/revenue.md',
-                            scope: 'GLOBAL',
-                            summary: 'Paid order value',
-                            score: 0.42,
-                            matchReasons: ['lexical'],
-                        },
-                    ],
-                    totalFound: 1,
-                }),
-                read: vi.fn().mockResolvedValue({
-                    key: 'revenue',
-                    summary: 'Paid order value',
-                    content: '# Revenue',
-                    scope: 'GLOBAL',
-                    tags: ['finance'],
-                    refs: [],
-                    slRefs: ['orders'],
-                }),
-            },
-            semanticLayer: {
-                readSource: vi.fn().mockResolvedValue({
-                    sourceName: 'orders',
-                    yaml: 'name: orders\n',
-                }),
-                query: vi.fn().mockResolvedValue({
-                    sql: 'select 1',
-                    headers: ['count'],
-                    rows: [[1]],
-                    totalRows: 1,
-                    plan: { sources: ['orders'] },
-                }),
-            },
-            entityDetails: {
-                read: vi.fn().mockResolvedValue({ results: [] }),
-            },
-            dictionarySearch: {
-                search: vi.fn().mockResolvedValue({
-                    searched: [],
-                    results: [],
-                }),
-            },
-            discover: {
-                search: vi.fn().mockResolvedValue([]),
-            },
-            sqlExecution: {
-                execute: vi.fn().mockResolvedValue({
-                    headers: ['count'],
-                    headerTypes: ['integer'],
-                    rows: [[1]],
-                    rowCount: 1,
-                }),
-            },
-            memoryIngest: {
-                ingest: vi.fn().mockResolvedValue({ runId: 'run-1' }),
-                status: vi.fn().mockResolvedValue(null),
-            },
-        };
-        createKtxMcpServer({
-            server: fake.server,
-            userContext: { userId: 'mcp-user' },
-            contextTools,
-        });
-        expect(fake.tools.map((tool) => tool.name).sort()).toEqual([
-            'connection_list',
-            'dictionary_search',
-            'discover_data',
-            'entity_details',
-            'memory_ingest',
-            'memory_ingest_status',
-            'sl_query',
-            'sl_read_source',
-            'sql_execution',
-            'wiki_read',
-            'wiki_search',
-        ]);
-        await expect(getTool(fake.tools, 'connection_list').handler({})).resolves.toEqual({
-            content: [
-                {
-                    type: 'text',
-                    text: JSON.stringify({
-                        connections: [
-                            {
-                                id: '00000000-0000-4000-8000-000000000001',
-                                name: 'Warehouse',
-                                connectionType: 'POSTGRES',
-                            },
-                        ],
-                    }, null, 2),
-                },
-            ],
-            structuredContent: {
-                connections: [
-                    {
-                        id: '00000000-0000-4000-8000-000000000001',
-                        name: 'Warehouse',
-                        connectionType: 'POSTGRES',
-                    },
-                ],
-            },
-        });
-        await getTool(fake.tools, 'wiki_search').handler({ query: 'revenue', limit: 5 });
-        expect(contextTools.knowledge?.search).toHaveBeenCalledWith({
-            userId: 'mcp-user',
-            query: 'revenue',
-            limit: 5,
-        });
-        await getTool(fake.tools, 'wiki_read').handler({ key: 'revenue' });
-        expect(contextTools.knowledge?.read).toHaveBeenCalledWith({
-            userId: 'mcp-user',
-            key: 'revenue',
-        });
-        await getTool(fake.tools, 'sl_read_source').handler({
-            connectionId: 'warehouse',
-            sourceName: 'orders',
-        });
-        expect(contextTools.semanticLayer?.readSource).toHaveBeenCalledWith({
-            connectionId: 'warehouse',
-            sourceName: 'orders',
-        });
-        await getTool(fake.tools, 'sl_query').handler({
-            connectionId: '00000000-0000-4000-8000-000000000001',
-            measures: ['orders.count'],
-            dimensions: ['orders.created_at'],
-            filters: ['orders.status = paid'],
-            limit: 25,
-        });
-        expect(contextTools.semanticLayer?.query).toHaveBeenCalledWith({
-            connectionId: '00000000-0000-4000-8000-000000000001',
-            query: {
-                measures: ['orders.count'],
-                dimensions: [{ field: 'orders.created_at' }],
-                filters: ['orders.status = paid'],
-                segments: [],
-                order_by: [],
-                limit: 25,
-                include_empty: true,
-            },
-        }, undefined);
-    });
-    it('keeps jsonToolResult typed to non-array objects', () => {
-        expect(jsonToolResult({ ok: true }).structuredContent).toEqual({ ok: true });
-        if (false) {
-            // @ts-expect-error bare arrays are not valid MCP structuredContent objects in KTX
-            jsonToolResult([]);
-        }
-    });
-});