ai-cli-mcp 2.3.0 → 2.3.1

package/dist/__tests__/mocks.js

@@ -0,0 +1,32 @@
+import { vi } from 'vitest';
+// Mock Claude CLI responses
+export const mockClaudeResponse = (stdout, stderr = '', exitCode = 0) => {
+    return {
+        stdout: { on: vi.fn((event, cb) => event === 'data' && cb(stdout)) },
+        stderr: { on: vi.fn((event, cb) => event === 'data' && cb(stderr)) },
+        on: vi.fn((event, cb) => {
+            if (event === 'exit')
+                setTimeout(() => cb(exitCode), 10);
+        }),
+    };
+};
+// Mock MCP request builder
+export const createMCPRequest = (tool, args, id = 1) => ({
+    jsonrpc: '2.0',
+    method: 'tools/call',
+    params: {
+        name: tool,
+        arguments: args,
+    },
+    id,
+});
+// Mock file system operations
+export const setupTestEnvironment = () => {
+    const testFiles = new Map();
+    return {
+        writeFile: (path, content) => testFiles.set(path, content),
+        readFile: (path) => testFiles.get(path),
+        exists: (path) => testFiles.has(path),
+        cleanup: () => testFiles.clear(),
+    };
+};

package/dist/__tests__/model-alias.test.js

@@ -0,0 +1,36 @@
+import { describe, it, expect } from 'vitest';
+// Test the model alias resolution logic directly
+describe('Model Alias Resolution', () => {
+    // Define the same MODEL_ALIASES as in server.ts
+    const MODEL_ALIASES = {
+        'haiku': 'claude-3-5-haiku-20241022'
+    };
+    // Replicate the resolveModelAlias function
+    function resolveModelAlias(model) {
+        return MODEL_ALIASES[model] || model;
+    }
+    it('should resolve haiku alias to full model name', () => {
+        expect(resolveModelAlias('haiku')).toBe('claude-3-5-haiku-20241022');
+    });
+    it('should pass through non-alias model names unchanged', () => {
+        expect(resolveModelAlias('sonnet')).toBe('sonnet');
+        expect(resolveModelAlias('opus')).toBe('opus');
+        expect(resolveModelAlias('claude-3-opus-20240229')).toBe('claude-3-opus-20240229');
+    });
+    it('should pass through empty strings', () => {
+        expect(resolveModelAlias('')).toBe('');
+    });
+    it('should be case-sensitive', () => {
+        // Should not resolve uppercase version
+        expect(resolveModelAlias('Haiku')).toBe('Haiku');
+        expect(resolveModelAlias('HAIKU')).toBe('HAIKU');
+    });
+    it('should handle undefined input gracefully', () => {
+        // TypeScript would normally prevent this, but testing for runtime safety
+        expect(resolveModelAlias(undefined)).toBe(undefined);
+    });
+    it('should handle null input gracefully', () => {
+        // TypeScript would normally prevent this, but testing for runtime safety
+        expect(resolveModelAlias(null)).toBe(null);
+    });
+});