@memoire-ai/mcp-runtime 0.1.0

package/src/index.test.ts

@@ -0,0 +1,183 @@
+import test from 'node:test';
+import assert from 'node:assert/strict';
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { InMemoryTransport } from '@modelcontextprotocol/sdk/inMemory.js';
+import type { MemoireClient } from '@memoire-ai/sdk';
+import { createMemoireMcpServer } from './index.js';
+
+test('handoff MCP tools save, list, and fetch shared handoffs', async () => {
+  const calls: Array<{ name: string; args: unknown }> = [];
+
+  const fakeClient = {
+    async assembleContext(args: unknown) {
+      calls.push({ name: 'assembleContext', args });
+      return {
+        context_block: '# Memoire Resume\n\n## Shared Memory\n- Use PKCE for OAuth callbacks',
+        sections: [],
+        token_count: 120,
+        event_ids: ['event-1', 'event-2'],
+        active_work: [],
+        conflicts: [],
+      };
+    },
+    async createHandoff(args: unknown) {
+      calls.push({ name: 'createHandoff', args });
+      return {
+        handoff: {
+          id: 'handoff-1',
+          title: 'OAuth callbacks',
+          query: 'oauth callbacks',
+          summary: 'Use PKCE for OAuth callbacks',
+          context_block: '# Memoire Resume\n\n## Shared Memory\n- Use PKCE for OAuth callbacks',
+          source_event_ids: ['event-1', 'event-2'],
+          created_by_user_id: 'user-1',
+          created_at: '2026-03-12T00:00:00.000Z',
+        },
+      };
+    },
+    async listHandoffs(args: unknown) {
+      calls.push({ name: 'listHandoffs', args });
+      return {
+        handoffs: [
+          {
+            id: 'handoff-1',
+            title: 'OAuth callbacks',
+            query: 'oauth callbacks',
+            summary: 'Use PKCE for OAuth callbacks',
+            context_block: '# Memoire Resume\n\n## Shared Memory\n- Use PKCE for OAuth callbacks',
+            source_event_ids: ['event-1', 'event-2'],
+            created_by_user_id: 'user-1',
+            created_at: '2026-03-12T00:00:00.000Z',
+          },
+        ],
+      };
+    },
+    async getHandoff(args: unknown) {
+      calls.push({ name: 'getHandoff', args });
+      return {
+        handoff: {
+          id: 'handoff-1',
+          title: 'OAuth callbacks',
+          query: 'oauth callbacks',
+          summary: 'Use PKCE for OAuth callbacks',
+          context_block: '# Memoire Resume\n\n## Shared Memory\n- Use PKCE for OAuth callbacks',
+          source_event_ids: ['event-1', 'event-2'],
+          created_by_user_id: 'user-1',
+          created_at: '2026-03-12T00:00:00.000Z',
+        },
+      };
+    },
+  } as unknown as MemoireClient;
+
+  const server = createMemoireMcpServer(
+    {
+      apiUrl: 'http://localhost:8080',
+      apiKey: 'sk_mem_test',
+      orgId: 'org-1',
+      projectId: 'project-1',
+      userId: 'user-1',
+    },
+    { client: fakeClient }
+  );
+
+  const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair();
+  const mcpClient = new Client({ name: 'memoire-runtime-test', version: '0.1.0' }, { capabilities: {} });
+
+  await Promise.all([
+    server.connect(serverTransport),
+    mcpClient.connect(clientTransport),
+  ]);
+
+  try {
+    const tools = await mcpClient.listTools();
+    const toolNames = tools.tools.map((tool) => tool.name);
+
+    assert.ok(toolNames.includes('save_handoff'));
+    assert.ok(toolNames.includes('list_handoffs'));
+    assert.ok(toolNames.includes('get_handoff'));
+
+    const saved = await mcpClient.callTool({
+      name: 'save_handoff',
+      arguments: {
+        query: 'oauth callbacks',
+        title: 'OAuth callbacks',
+        token_budget: 900,
+      },
+    });
+
+    assert.match(readToolText(saved), /Saved handoff handoff-1/);
+    assert.match(readToolText(saved), /Use PKCE for OAuth callbacks/);
+
+    const listed = await mcpClient.callTool({
+      name: 'list_handoffs',
+      arguments: { limit: 5 },
+    });
+
+    assert.match(readToolText(listed), /OAuth callbacks/);
+    assert.match(readToolText(listed), /handoff-1/);
+
+    const loaded = await mcpClient.callTool({
+      name: 'get_handoff',
+      arguments: { handoff_id: 'handoff-1' },
+    });
+
+    assert.match(readToolText(loaded), /^# OAuth callbacks/m);
+    assert.match(readToolText(loaded), /Use PKCE for OAuth callbacks/);
+
+    assert.deepEqual(calls, [
+      {
+        name: 'assembleContext',
+        args: {
+          org_id: 'org-1',
+          project_id: 'project-1',
+          viewer_user_id: 'user-1',
+          query: 'oauth callbacks',
+          token_budget: 900,
+        },
+      },
+      {
+        name: 'createHandoff',
+        args: {
+          org_id: 'org-1',
+          project_id: 'project-1',
+          viewer_user_id: 'user-1',
+          title: 'OAuth callbacks',
+          query: 'oauth callbacks',
+          context_block: '# Memoire Resume\n\n## Shared Memory\n- Use PKCE for OAuth callbacks',
+          source_event_ids: ['event-1', 'event-2'],
+        },
+      },
+      {
+        name: 'listHandoffs',
+        args: {
+          org_id: 'org-1',
+          project_id: 'project-1',
+          viewer_user_id: 'user-1',
+          limit: 5,
+        },
+      },
+      {
+        name: 'getHandoff',
+        args: {
+          org_id: 'org-1',
+          project_id: 'project-1',
+          viewer_user_id: 'user-1',
+          handoff_id: 'handoff-1',
+        },
+      },
+    ]);
+  } finally {
+    await Promise.allSettled([mcpClient.close(), server.close()]);
+  }
+});
+
+function readToolText(result: Awaited<ReturnType<Client['callTool']>>): string {
+  if (!('content' in result) || !Array.isArray(result.content)) {
+    return '';
+  }
+
+  return result.content
+    .filter((item): item is { type: 'text'; text: string } => item.type === 'text')
+    .map((item) => item.text)
+    .join('\n');
+}