import { jest, describe, it, expect, beforeEach, afterEach } from '@jest/globals';
// Mock all dependencies BEFORE importing
const mockRunAuthCli = jest.fn<() => Promise<void>>();
const mockRunStatusCli = jest.fn<() => Promise<void>>();
const mockRun = jest.fn<() => Promise<void>>();
const mockShutdown = jest.fn<() => Promise<void>>();
// Mock server constructor
const MockTSMCPServer = jest.fn().mockImplementation(() => ({
run: mockRun,
shutdown: mockShutdown
}));
// Mock keychain
const MockKeychainService = jest.fn();
// Mock local auth provider
const MockLocalAuthProvider = jest.fn();
jest.unstable_mockModule('../../src/cli/auth.js', () => ({
runAuthCli: mockRunAuthCli
}));
jest.unstable_mockModule('../../src/cli/status.js', () => ({
runStatusCli: mockRunStatusCli
}));
jest.unstable_mockModule('../../src/server/mcp-server.js', () => ({
TSMCPServer: MockTSMCPServer
}));
jest.unstable_mockModule('../../src/auth/keychain.js', () => ({
KeychainService: MockKeychainService
}));
jest.unstable_mockModule('../../src/auth/local-auth-provider.js', () => ({
LocalAuthProvider: MockLocalAuthProvider
}));
// Import module after mocks are set up
const indexModule = await import('../../src/index.js');
describe('index.ts', () => {
let mockConsoleLog: jest.SpiedFunction<typeof console.log>;
let mockConsoleError: jest.SpiedFunction<typeof console.error>;
let mockProcessExit: jest.SpiedFunction<typeof process.exit>;
let mockProcessOn: jest.SpiedFunction<typeof process.on>;
beforeEach(() => {
jest.clearAllMocks();
// Mock console and process
mockConsoleLog = jest.spyOn(console, 'log').mockImplementation(() => {});
mockConsoleError = jest.spyOn(console, 'error').mockImplementation(() => {});
mockProcessExit = jest.spyOn(process, 'exit').mockImplementation((code) => {
throw new Error(`process.exit(${code})`);
});
mockProcessOn = jest.spyOn(process, 'on').mockImplementation(() => process);
// Default mock implementations
mockRunAuthCli.mockResolvedValue(undefined);
mockRunStatusCli.mockResolvedValue(undefined);
mockRun.mockResolvedValue(undefined);
});
afterEach(() => {
mockConsoleLog.mockRestore();
mockConsoleError.mockRestore();
mockProcessExit.mockRestore();
mockProcessOn.mockRestore();
});
describe('handleCommand', () => {
describe('command routing', () => {
it('routes "auth" command to runAuthCli', async () => {
await indexModule.handleCommand(['auth']);
expect(mockRunAuthCli).toHaveBeenCalled();
});
it('routes "status" command to runStatusCli', async () => {
await indexModule.handleCommand(['status']);
expect(mockRunStatusCli).toHaveBeenCalled();
});
it('exits with error for unknown command', async () => {
await expect(
indexModule.handleCommand(['unknown-command'])
).rejects.toThrow('process.exit(1)');
expect(mockConsoleError).toHaveBeenCalledWith('Unknown command: unknown-command');
expect(mockConsoleError).toHaveBeenCalledWith('Run "ts-mcp --help" for usage.');
});
});
describe('help output', () => {
it('shows help for "--help" flag', async () => {
await indexModule.handleCommand(['--help']);
expect(mockConsoleLog).toHaveBeenCalledWith('TS-MCP - Touchstone MCP Server for Claude Code');
expect(mockConsoleLog).toHaveBeenCalledWith('Usage: ts-mcp [command]');
expect(mockConsoleLog).toHaveBeenCalledWith('Commands:');
expect(mockConsoleLog).toHaveBeenCalledWith(' (none) Start MCP server');
expect(mockConsoleLog).toHaveBeenCalledWith(' auth Authenticate with Touchstone');
expect(mockConsoleLog).toHaveBeenCalledWith(' status Show authentication status');
expect(mockConsoleLog).toHaveBeenCalledWith(' --help Show this help');
});
it('shows help for "-h" flag', async () => {
await indexModule.handleCommand(['-h']);
expect(mockConsoleLog).toHaveBeenCalledWith('TS-MCP - Touchstone MCP Server for Claude Code');
});
it('shows example in help', async () => {
await indexModule.handleCommand(['--help']);
expect(mockConsoleLog).toHaveBeenCalledWith('Example:');
expect(mockConsoleLog).toHaveBeenCalledWith(' claude mcp add ts-mcp -- npx github:AEGISnetinc/TS-MCP');
expect(mockConsoleLog).toHaveBeenCalledWith(' npx github:AEGISnetinc/TS-MCP auth');
});
});
describe('default behavior', () => {
it('starts local server when no command provided', async () => {
await indexModule.handleCommand([]);
expect(MockTSMCPServer).toHaveBeenCalled();
expect(mockRun).toHaveBeenCalled();
});
it('registers SIGINT handler for local server', async () => {
await indexModule.handleCommand([]);
expect(mockProcessOn).toHaveBeenCalledWith('SIGINT', expect.any(Function));
});
it('registers SIGTERM handler for local server', async () => {
await indexModule.handleCommand([]);
expect(mockProcessOn).toHaveBeenCalledWith('SIGTERM', expect.any(Function));
});
});
});
describe('printHelp', () => {
it('prints complete help message', () => {
indexModule.printHelp();
expect(mockConsoleLog).toHaveBeenCalledWith('TS-MCP - Touchstone MCP Server for Claude Code');
expect(mockConsoleLog).toHaveBeenCalledWith('');
expect(mockConsoleLog).toHaveBeenCalledWith('Usage: ts-mcp [command]');
});
});
describe('runLocalServer', () => {
it('creates server with LocalAuthProvider', async () => {
await indexModule.runLocalServer();
expect(MockKeychainService).toHaveBeenCalled();
expect(MockLocalAuthProvider).toHaveBeenCalled();
expect(MockTSMCPServer).toHaveBeenCalled();
});
it('starts the server', async () => {
await indexModule.runLocalServer();
expect(mockRun).toHaveBeenCalled();
});
it('registers signal handlers', async () => {
await indexModule.runLocalServer();
expect(mockProcessOn).toHaveBeenCalledWith('SIGINT', expect.any(Function));
expect(mockProcessOn).toHaveBeenCalledWith('SIGTERM', expect.any(Function));
});
});
describe('main', () => {
let originalArgv: string[];
beforeEach(() => {
originalArgv = process.argv;
});
afterEach(() => {
process.argv = originalArgv;
});
it('reads args from process.argv and calls handleCommand', async () => {
process.argv = ['node', 'index.js', 'status'];
await indexModule.main();
expect(mockRunStatusCli).toHaveBeenCalled();
});
it('handles auth command from process.argv', async () => {
process.argv = ['node', 'index.js', 'auth'];
await indexModule.main();
expect(mockRunAuthCli).toHaveBeenCalled();
});
});
});
