packages/cli/src/commands/mcp.test.ts

/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import { describe, it, expect, vi } from 'vitest';
import { mcpCommand } from './mcp.js';
import yargs, { type Argv } from 'yargs';

describe('mcp command', () => {
  it('should have correct command definition', () => {
    expect(mcpCommand.command).toBe('mcp');
    expect(mcpCommand.describe).toBe('Manage MCP servers');
    expect(typeof mcpCommand.builder).toBe('function');
    expect(typeof mcpCommand.handler).toBe('function');
  });

  it('should show help when no subcommand is provided', async () => {
    const yargsInstance = yargs().locale('en');
    (mcpCommand.builder as (y: Argv) => Argv)(yargsInstance);

    const parser = yargsInstance.command(mcpCommand).help();

    // Mock console.log and console.error to catch help output
    const consoleLogMock = vi
      .spyOn(console, 'log')
      .mockImplementation(() => {});
    const consoleErrorMock = vi
      .spyOn(console, 'error')
      .mockImplementation(() => {});

    try {
      await parser.parse('mcp');
    } catch {
      // yargs might throw an error when demandCommand is not met
    }

    // Check if help output is shown
    const helpOutput =
      consoleLogMock.mock.calls.join('\n') +
      consoleErrorMock.mock.calls.join('\n');
    expect(helpOutput).toContain('Manage MCP servers');
    expect(helpOutput).toContain('Commands:');
    expect(helpOutput).toContain('add');
    expect(helpOutput).toContain('remove');
    expect(helpOutput).toContain('list');

    consoleLogMock.mockRestore();
    consoleErrorMock.mockRestore();
  });

  it('should register add, remove, and list subcommands', () => {
    const mockYargs = {
      command: vi.fn().mockReturnThis(),
      demandCommand: vi.fn().mockReturnThis(),
      version: vi.fn().mockReturnThis(),
      middleware: vi.fn().mockReturnThis(),
    };

    (mcpCommand.builder as (y: Argv) => Argv)(mockYargs as unknown as Argv);

    expect(mockYargs.command).toHaveBeenCalledTimes(5);

    // Verify that the specific subcommands are registered
    const commandCalls = mockYargs.command.mock.calls;
    const commandNames = commandCalls.map((call) => call[0].command);

    expect(commandNames).toContain('add <name> <commandOrUrl> [args...]');
    expect(commandNames).toContain('remove <name>');
    expect(commandNames).toContain('list');
    expect(commandNames).toContain('enable <name>');
    expect(commandNames).toContain('disable <name>');

    expect(mockYargs.demandCommand).toHaveBeenCalledWith(
      1,
      'You need at least one command before continuing.',
    );
  });
});