Function createMockExtensionAPI

tests/test-utils.ts:98–150

()

// Mock ExtensionAPI factory
export const createMockExtensionAPI = () => {
	const registeredTools: Map<string, any> = new Map();
	const registeredCommands: Map<string, any> = new Map();
	const registeredFlags: Map<string, any> = new Map();
	const eventHandlers: Map<string, ((...args: any[]) => any)[]> = new Map();
	const appendedEntries: any[] = [];
	let execMock = jest.fn();

	const api = {
		registerTool: jest.fn((tool: any) => {
			registeredTools.set(tool.name, tool);
		}),
		registerCommand: jest.fn((name: string, options: any) => {
			registeredCommands.set(name, options);
		}),
		registerFlag: jest.fn((name: string, options: any) => {
			registeredFlags.set(name, { ...options, value: undefined });
		}),
		registerShortcut: jest.fn(),
		registerMessageRenderer: jest.fn(),
		on: jest.fn((event: string, handler: (...args: any[]) => any) => {
			if (!eventHandlers.has(event)) {
				eventHandlers.set(event, []);
			}
			eventHandlers.get(event)!.push(handler);
		}),
		appendEntry: jest.fn((customType: string, data: any) => {
			appendedEntries.push({ customType, data });
		}),
		exec: jest.fn((...args: any[]) => execMock(...args)),
		getFlag: jest.fn((name: string) => registeredFlags.get(name)?.value),
		setActiveTools: jest.fn(),
		getActiveTools: jest.fn(() => ["read", "bash", "edit", "write"]),
		getAllTools: jest.fn(() => ["read", "bash", "edit", "write"]),
		sendMessage: jest.fn(),
		sendUserMessage: jest.fn(),
		events: {
			on: jest.fn(),
			emit: jest.fn(),
			off: jest.fn(),
		},
		_registeredTools: registeredTools,
		_registeredCommands: registeredCommands,
		_registeredFlags: registeredFlags,
		_eventHandlers: eventHandlers,
		_appendedEntries: appendedEntries,
		_setExecMock: (mock: jest.Mock) => {
			execMock = mock;
		},
	};

	return api;
};

// Load extension function
// eslint-disable-next-line @typescript-eslint/no-require-imports