98 lines
3.1 KiB
TypeScript
98 lines
3.1 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
|
|
import * as fs from 'node:fs/promises';
|
|
|
|
import { McpInstallationStateService } from '@main/services/extensions/state/McpInstallationStateService';
|
|
|
|
vi.mock('@main/utils/pathDecoder', () => ({
|
|
getHomeDir: () => '/tmp/mock-home',
|
|
}));
|
|
|
|
vi.mock('node:fs/promises');
|
|
|
|
describe('McpInstallationStateService', () => {
|
|
let service: McpInstallationStateService;
|
|
const mockedFs = vi.mocked(fs);
|
|
|
|
beforeEach(() => {
|
|
service = new McpInstallationStateService();
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
describe('getInstalled', () => {
|
|
it('caches results within TTL for the same project path', async () => {
|
|
mockedFs.readFile.mockImplementation(async (filePath) => {
|
|
const normalizedPath = String(filePath);
|
|
if (normalizedPath === '/tmp/mock-home/.claude.json') {
|
|
return JSON.stringify({
|
|
mcpServers: {
|
|
context7: { command: 'npx -y @upstash/context7-mcp' },
|
|
},
|
|
});
|
|
}
|
|
|
|
if (normalizedPath === '/tmp/project-a/.mcp.json') {
|
|
return JSON.stringify({
|
|
mcpServers: {
|
|
'repo-a-server': { url: 'https://repo-a.example.com/mcp' },
|
|
},
|
|
});
|
|
}
|
|
|
|
throw Object.assign(new Error('ENOENT'), { code: 'ENOENT' });
|
|
});
|
|
|
|
await service.getInstalled('/tmp/project-a');
|
|
await service.getInstalled('/tmp/project-a');
|
|
|
|
expect(mockedFs.readFile).toHaveBeenCalledTimes(2);
|
|
});
|
|
|
|
it('caches results independently per project path', async () => {
|
|
mockedFs.readFile.mockImplementation(async (filePath) => {
|
|
const normalizedPath = String(filePath);
|
|
if (normalizedPath === '/tmp/mock-home/.claude.json') {
|
|
return JSON.stringify({
|
|
mcpServers: {
|
|
context7: { command: 'npx -y @upstash/context7-mcp' },
|
|
},
|
|
});
|
|
}
|
|
|
|
if (normalizedPath === '/tmp/project-a/.mcp.json') {
|
|
return JSON.stringify({
|
|
mcpServers: {
|
|
'repo-a-server': { url: 'https://repo-a.example.com/mcp' },
|
|
},
|
|
});
|
|
}
|
|
|
|
if (normalizedPath === '/tmp/project-b/.mcp.json') {
|
|
return JSON.stringify({
|
|
mcpServers: {
|
|
'repo-b-server': { command: 'uvx repo-b-mcp' },
|
|
},
|
|
});
|
|
}
|
|
|
|
throw Object.assign(new Error('ENOENT'), { code: 'ENOENT' });
|
|
});
|
|
|
|
const projectAEntries = await service.getInstalled('/tmp/project-a');
|
|
const projectBEntries = await service.getInstalled('/tmp/project-b');
|
|
|
|
expect(projectAEntries).toEqual([
|
|
{ name: 'context7', scope: 'user', transport: 'stdio' },
|
|
{ name: 'repo-a-server', scope: 'project', transport: 'http' },
|
|
]);
|
|
expect(projectBEntries).toEqual([
|
|
{ name: 'context7', scope: 'user', transport: 'stdio' },
|
|
{ name: 'repo-b-server', scope: 'project', transport: 'stdio' },
|
|
]);
|
|
expect(mockedFs.readFile).toHaveBeenCalledTimes(4);
|
|
});
|
|
});
|
|
});
|