139 lines
4.8 KiB
TypeScript
139 lines
4.8 KiB
TypeScript
/**
|
|
* Unit tests for mcp/resources.ts — registerResources, 3 resource handlers.
|
|
*/
|
|
|
|
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
|
|
// Mock platform
|
|
const mockGetPlatform = vi.fn();
|
|
const mockCheckScreenRecording = vi.fn();
|
|
const mockCheckAccessibility = vi.fn();
|
|
vi.mock('../../src/utils/platform.js', () => ({
|
|
getPlatform: () => mockGetPlatform(),
|
|
checkScreenRecordingPermission: () => mockCheckScreenRecording(),
|
|
checkAccessibilityPermission: () => mockCheckAccessibility(),
|
|
}));
|
|
|
|
import { registerResources } from '../../src/mcp/resources.js';
|
|
import { AuditLog } from '../../src/safety/auditLog.js';
|
|
import type { GuiAgentConfig } from '../../src/config.js';
|
|
|
|
// Minimal Server mock that captures request handlers
|
|
class MockServer {
|
|
handlers = new Map<any, Function>();
|
|
|
|
setRequestHandler(schema: any, handler: Function) {
|
|
// Use the schema method name or the schema itself as key
|
|
this.handlers.set(schema, handler);
|
|
}
|
|
|
|
// Get handler by schema
|
|
getHandler(schema: any): Function | undefined {
|
|
return this.handlers.get(schema);
|
|
}
|
|
}
|
|
|
|
// We need the actual schemas to register & retrieve handlers
|
|
// Import them for use as keys
|
|
import {
|
|
ListResourcesRequestSchema,
|
|
ReadResourceRequestSchema,
|
|
} from '@modelcontextprotocol/sdk/types.js';
|
|
|
|
const baseConfig: GuiAgentConfig = {
|
|
apiKey: 'test-key',
|
|
provider: 'anthropic',
|
|
model: 'claude-sonnet-4-20250514',
|
|
port: 60008,
|
|
transport: 'http' as const,
|
|
maxSteps: 50,
|
|
stepDelayMs: 1500,
|
|
stuckThreshold: 3,
|
|
jpegQuality: 75,
|
|
displayIndex: 0,
|
|
};
|
|
|
|
describe('registerResources', () => {
|
|
let server: MockServer;
|
|
let auditLog: AuditLog;
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
server = new MockServer();
|
|
auditLog = new AuditLog();
|
|
mockGetPlatform.mockReturnValue('macos');
|
|
registerResources(server as any, baseConfig, auditLog);
|
|
});
|
|
|
|
describe('ListResources', () => {
|
|
it('returns 3 resources', async () => {
|
|
const handler = server.getHandler(ListResourcesRequestSchema)!;
|
|
const result = await handler({});
|
|
expect(result.resources).toHaveLength(3);
|
|
});
|
|
|
|
it('includes gui://status, gui://permissions, gui://audit-log', async () => {
|
|
const handler = server.getHandler(ListResourcesRequestSchema)!;
|
|
const result = await handler({});
|
|
const uris = result.resources.map((r: any) => r.uri);
|
|
expect(uris).toContain('gui://status');
|
|
expect(uris).toContain('gui://permissions');
|
|
expect(uris).toContain('gui://audit-log');
|
|
});
|
|
});
|
|
|
|
describe('ReadResource: gui://status', () => {
|
|
it('returns platform, transport, port, model', async () => {
|
|
const handler = server.getHandler(ReadResourceRequestSchema)!;
|
|
const result = await handler({ params: { uri: 'gui://status' } });
|
|
const status = JSON.parse(result.contents[0].text);
|
|
expect(status.platform).toBe('macos');
|
|
expect(status.transport).toBe('http');
|
|
expect(status.port).toBe(60008);
|
|
expect(status.model).toBe('claude-sonnet-4-20250514');
|
|
expect(status.running).toBe(true);
|
|
});
|
|
});
|
|
|
|
describe('ReadResource: gui://permissions', () => {
|
|
it('returns screen recording and accessibility status', async () => {
|
|
mockCheckScreenRecording.mockResolvedValue(true);
|
|
mockCheckAccessibility.mockResolvedValue(false);
|
|
|
|
const handler = server.getHandler(ReadResourceRequestSchema)!;
|
|
const result = await handler({ params: { uri: 'gui://permissions' } });
|
|
const perms = JSON.parse(result.contents[0].text);
|
|
expect(perms.screenRecording).toBe(true);
|
|
expect(perms.accessibility).toBe(false);
|
|
expect(perms.platform).toBe('macos');
|
|
});
|
|
});
|
|
|
|
describe('ReadResource: gui://audit-log', () => {
|
|
it('returns empty entries when no audit records', async () => {
|
|
const handler = server.getHandler(ReadResourceRequestSchema)!;
|
|
const result = await handler({ params: { uri: 'gui://audit-log' } });
|
|
const entries = JSON.parse(result.contents[0].text);
|
|
expect(entries).toHaveLength(0);
|
|
});
|
|
|
|
it('returns recorded audit entries', async () => {
|
|
auditLog.record({ tool: 'gui_click', args: { x: 100 }, success: true });
|
|
auditLog.record({ tool: 'gui_type', args: { text: 'hi' }, success: true });
|
|
|
|
const handler = server.getHandler(ReadResourceRequestSchema)!;
|
|
const result = await handler({ params: { uri: 'gui://audit-log' } });
|
|
const entries = JSON.parse(result.contents[0].text);
|
|
expect(entries).toHaveLength(2);
|
|
expect(entries[0].tool).toBe('gui_type'); // most recent first
|
|
});
|
|
});
|
|
|
|
describe('ReadResource: unknown URI', () => {
|
|
it('throws for unknown resource URI', async () => {
|
|
const handler = server.getHandler(ReadResourceRequestSchema)!;
|
|
await expect(handler({ params: { uri: 'gui://unknown' } })).rejects.toThrow('Unknown resource');
|
|
});
|
|
});
|
|
});
|