56 lines
1.8 KiB
TypeScript
56 lines
1.8 KiB
TypeScript
|
/**
|
||
|
|
* @license
|
||
|
|
* Copyright 2025 Google LLC
|
||
|
|
* SPDX-License-Identifier: Apache-2.0
|
||
|
|
*/
|
||
|
|
|
||
|
|
import { describe, it, expect, vi } from 'vitest';
|
||
|
|
import { mcpCommand } from './mcp.js';
|
||
|
|
import { type Argv } from 'yargs';
|
||
|
|
import yargs 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 have exactly one option (help flag)', () => {
|
||
|
|
// Test to ensure that the global 'gemini' flags are not added to the mcp command
|
||
|
|
const yargsInstance = yargs();
|
||
|
|
const builtYargs = mcpCommand.builder(yargsInstance);
|
||
|
|
const options = builtYargs.getOptions();
|
||
|
|
|
||
|
|
// Should have exactly 1 option (help flag)
|
||
|
|
expect(Object.keys(options.key).length).toBe(1);
|
||
|
|
expect(options.key).toHaveProperty('help');
|
||
|
|
});
|
||
|
|
|
||
|
|
it('should register add, remove, and list subcommands', () => {
|
||
|
|
const mockYargs = {
|
||
|
|
command: vi.fn().mockReturnThis(),
|
||
|
|
demandCommand: vi.fn().mockReturnThis(),
|
||
|
|
version: vi.fn().mockReturnThis(),
|
||
|
|
};
|
||
|
|
|
||
|
|
mcpCommand.builder(mockYargs as unknown as Argv);
|
||
|
|
|
||
|
|
expect(mockYargs.command).toHaveBeenCalledTimes(3);
|
||
|
|
|
||
|
|
// 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(mockYargs.demandCommand).toHaveBeenCalledWith(
|
||
|
|
1,
|
||
|
|
'You need at least one command before continuing.',
|
||
|
|
);
|
||
|
|
});
|
||
|
|
});
|