@deimoscloud/coreai 0.1.0

package/src/adapters/mcp/mapper.test.ts

@@ -0,0 +1,218 @@
+import {
+  findKnownServerMappings,
+  autoDiscoverMappings,
+  getMappingsForServer,
+  groupMappingsByAdapter,
+} from './mapper.js';
+import type { McpTool, ToolMapping } from './index.js';
+
+describe('MCP Tool Mapper', () => {
+  describe('findKnownServerMappings', () => {
+    it('should find mappings for GitHub server', () => {
+      const mappings = findKnownServerMappings('@modelcontextprotocol/server-github');
+
+      expect(mappings.length).toBeGreaterThan(0);
+      expect(mappings.some((m) => m.method === 'createPullRequest')).toBe(true);
+      expect(mappings.some((m) => m.method === 'getIssue')).toBe(true);
+    });
+
+    it('should find mappings for server name containing github', () => {
+      const mappings = findKnownServerMappings('github');
+      expect(mappings.length).toBeGreaterThan(0);
+    });
+
+    it('should find mappings for mcp-server-github', () => {
+      const mappings = findKnownServerMappings('mcp-server-github');
+      expect(mappings.length).toBeGreaterThan(0);
+    });
+
+    it('should find mappings for Jira server', () => {
+      const mappings = findKnownServerMappings('jira');
+      expect(mappings.length).toBeGreaterThan(0);
+      expect(mappings.some((m) => m.adapterType === 'issue_tracker')).toBe(true);
+    });
+
+    it('should find mappings for Confluence server', () => {
+      const mappings = findKnownServerMappings('confluence');
+      expect(mappings.length).toBeGreaterThan(0);
+      expect(mappings.some((m) => m.adapterType === 'documentation')).toBe(true);
+    });
+
+    it('should find mappings for filesystem server', () => {
+      const mappings = findKnownServerMappings('filesystem');
+      expect(mappings.length).toBeGreaterThan(0);
+      expect(mappings.some((m) => m.adapterType === 'state')).toBe(true);
+    });
+
+    it('should return empty array for unknown server', () => {
+      const mappings = findKnownServerMappings('unknown-server');
+      expect(mappings).toHaveLength(0);
+    });
+  });
+
+  describe('autoDiscoverMappings', () => {
+    it('should discover issue tracker tools', () => {
+      const tools: McpTool[] = [
+        { name: 'get_issue', inputSchema: { type: 'object' } },
+        { name: 'list_issues', inputSchema: { type: 'object' } },
+        { name: 'create_ticket', inputSchema: { type: 'object' } },
+      ];
+
+      const mappings = autoDiscoverMappings(tools);
+
+      expect(mappings.length).toBe(3);
+      expect(mappings.every((m) => m.adapterType === 'issue_tracker')).toBe(true);
+    });
+
+    it('should discover git provider tools', () => {
+      const tools: McpTool[] = [
+        { name: 'create_pull_request', inputSchema: { type: 'object' } },
+        { name: 'merge_pr', inputSchema: { type: 'object' } },
+        { name: 'add_review', inputSchema: { type: 'object' } },
+      ];
+
+      const mappings = autoDiscoverMappings(tools);
+
+      expect(mappings.length).toBe(3);
+      expect(mappings.every((m) => m.adapterType === 'git')).toBe(true);
+    });
+
+    it('should discover documentation provider tools', () => {
+      const tools: McpTool[] = [
+        { name: 'get_page', inputSchema: { type: 'object' } },
+        { name: 'search_docs', inputSchema: { type: 'object' } },
+      ];
+
+      const mappings = autoDiscoverMappings(tools);
+
+      expect(mappings.length).toBe(2);
+      expect(mappings.every((m) => m.adapterType === 'documentation')).toBe(true);
+    });
+
+    it('should discover state provider tools', () => {
+      const tools: McpTool[] = [
+        { name: 'read_file', inputSchema: { type: 'object' } },
+        { name: 'write_file', inputSchema: { type: 'object' } },
+        { name: 'list_directory', inputSchema: { type: 'object' } },
+      ];
+
+      const mappings = autoDiscoverMappings(tools);
+
+      expect(mappings.length).toBe(3);
+      expect(mappings.every((m) => m.adapterType === 'state')).toBe(true);
+    });
+
+    it('should set confidence score for auto-discovered mappings', () => {
+      const tools: McpTool[] = [{ name: 'get_issue', inputSchema: { type: 'object' } }];
+
+      const mappings = autoDiscoverMappings(tools);
+
+      expect(mappings[0].confidence).toBeDefined();
+      expect(mappings[0].confidence).toBeGreaterThan(0);
+    });
+
+    it('should return empty array for unrecognized tools', () => {
+      const tools: McpTool[] = [
+        { name: 'do_something', inputSchema: { type: 'object' } },
+        { name: 'calculate', inputSchema: { type: 'object' } },
+      ];
+
+      const mappings = autoDiscoverMappings(tools);
+      expect(mappings).toHaveLength(0);
+    });
+
+    it('should discover tool that matches method pattern', () => {
+      // Tool name must match a known method pattern for successful discovery
+      const tools: McpTool[] = [
+        {
+          name: 'get_ticket',
+          description: 'Fetch a ticket from the tracker',
+          inputSchema: { type: 'object' },
+        },
+      ];
+
+      const mappings = autoDiscoverMappings(tools);
+      expect(mappings.length).toBe(1);
+      expect(mappings[0].adapterType).toBe('issue_tracker');
+      expect(mappings[0].method).toBe('getIssue');
+    });
+  });
+
+  describe('getMappingsForServer', () => {
+    it('should return known mappings filtered by available tools', () => {
+      const tools: McpTool[] = [
+        { name: 'create_pull_request', inputSchema: { type: 'object' } },
+        { name: 'get_issue', inputSchema: { type: 'object' } },
+        // Note: not including merge_pull_request
+      ];
+
+      const mappings = getMappingsForServer('github', tools);
+
+      expect(mappings.some((m) => m.toolName === 'create_pull_request')).toBe(true);
+      expect(mappings.some((m) => m.toolName === 'get_issue')).toBe(true);
+      expect(mappings.some((m) => m.toolName === 'merge_pull_request')).toBe(false);
+    });
+
+    it('should fall back to auto-discovery for unknown servers', () => {
+      const tools: McpTool[] = [
+        { name: 'get_issue', inputSchema: { type: 'object' } },
+        { name: 'create_pr', inputSchema: { type: 'object' } },
+      ];
+
+      const mappings = getMappingsForServer('custom-server', tools);
+
+      expect(mappings.length).toBe(2);
+      expect(mappings.some((m) => m.adapterType === 'issue_tracker')).toBe(true);
+      expect(mappings.some((m) => m.adapterType === 'git')).toBe(true);
+    });
+  });
+
+  describe('groupMappingsByAdapter', () => {
+    it('should group mappings by adapter type', () => {
+      const mappings: ToolMapping[] = [
+        { toolName: 'get_issue', adapterType: 'issue_tracker', method: 'getIssue' },
+        { toolName: 'create_issue', adapterType: 'issue_tracker', method: 'createIssue' },
+        { toolName: 'get_pr', adapterType: 'git', method: 'getPullRequest' },
+        { toolName: 'get_page', adapterType: 'documentation', method: 'getPage' },
+      ];
+
+      const grouped = groupMappingsByAdapter(mappings);
+
+      expect(grouped.get('issue_tracker')?.length).toBe(2);
+      expect(grouped.get('git')?.length).toBe(1);
+      expect(grouped.get('documentation')?.length).toBe(1);
+      expect(grouped.get('state')).toBeUndefined();
+    });
+
+    it('should handle empty mappings', () => {
+      const grouped = groupMappingsByAdapter([]);
+      expect(grouped.size).toBe(0);
+    });
+  });
+
+  describe('known server mappings correctness', () => {
+    it('GitHub mappings should have correct parameter maps', () => {
+      const mappings = findKnownServerMappings('github');
+
+      const createPR = mappings.find((m) => m.method === 'createPullRequest');
+      expect(createPR).toBeDefined();
+      expect(createPR?.parameterMap?.title).toBe('title');
+      expect(createPR?.parameterMap?.head).toBe('source_branch');
+      expect(createPR?.parameterMap?.base).toBe('target_branch');
+
+      const getIssue = mappings.find(
+        (m) => m.method === 'getIssue' && m.adapterType === 'issue_tracker'
+      );
+      expect(getIssue).toBeDefined();
+      expect(getIssue?.parameterMap?.issue_number).toBe('idOrKey');
+    });
+
+    it('Jira mappings should have correct parameter maps', () => {
+      const mappings = findKnownServerMappings('jira');
+
+      const createIssue = mappings.find((m) => m.method === 'createIssue');
+      expect(createIssue).toBeDefined();
+      expect(createIssue?.parameterMap?.summary).toBe('title');
+    });
+  });
+});