161c675aaa
- Added new dependencies for mcp-server in pnpm-lock.yaml, including fastmcp and zod. - Updated pnpm-workspace.yaml to include mcp-server in the workspace packages. - Modified TeamDataService and TeamProvisioningService to exclude 'SendMessage' from tool usage counts, improving accuracy in tool tracking. - Enhanced ChatHistory component to display context injection percentages, improving user feedback on context usage. - Updated TeamDetailView to ensure session details are fetched periodically for active tabs, enhancing data freshness. - Improved ActivityTimeline styling for better visual separation of sessions and messages.
58 lines
1.5 KiB
TypeScript
58 lines
1.5 KiB
TypeScript
import { vi } from 'vitest';
|
|
import type { FastMCP } from 'fastmcp';
|
|
import type { ITeamctlRunner, TeamctlResult } from '../../src/teamctl-runner.js';
|
|
|
|
/**
|
|
* Creates a mock ITeamctlRunner that records calls and returns predetermined results.
|
|
*/
|
|
export function createMockRunner(
|
|
response: TeamctlResult | ((args: string[]) => TeamctlResult),
|
|
): ITeamctlRunner & { execute: ReturnType<typeof vi.fn> } {
|
|
return {
|
|
execute: vi.fn(async (args: string[]): Promise<TeamctlResult> => {
|
|
return typeof response === 'function' ? response(args) : response;
|
|
}),
|
|
};
|
|
}
|
|
|
|
/** Success response helpers */
|
|
export const ok = (stdout: string): TeamctlResult => ({
|
|
stdout,
|
|
stderr: '',
|
|
exitCode: 0,
|
|
});
|
|
|
|
export const fail = (stderr: string): TeamctlResult => ({
|
|
stdout: '',
|
|
stderr,
|
|
exitCode: 1,
|
|
});
|
|
|
|
/**
|
|
* Captures registered tools via a mock FastMCP-like server.
|
|
* Returns a map of tool name → { execute function, parameters schema }.
|
|
*/
|
|
export interface CapturedTool {
|
|
name: string;
|
|
execute: (args: Record<string, unknown>) => Promise<unknown>;
|
|
parameters: unknown;
|
|
}
|
|
|
|
export function createMockServer(): {
|
|
server: FastMCP;
|
|
tools: Map<string, CapturedTool>;
|
|
} {
|
|
const tools = new Map<string, CapturedTool>();
|
|
|
|
const server = {
|
|
addTool: (def: { name: string; execute: CapturedTool['execute']; parameters: unknown }) => {
|
|
tools.set(def.name, {
|
|
name: def.name,
|
|
execute: def.execute,
|
|
parameters: def.parameters,
|
|
});
|
|
},
|
|
} as unknown as FastMCP;
|
|
|
|
return { server, tools };
|
|
}
|