import { describe, it, expect, vi, beforeEach } from 'vitest';
import { parseArgs } from '../args.js';
// Mock process.exit
vi.spyOn(process, 'exit').mockImplementation((code?: number | string | null | undefined) => {
throw new Error(`process.exit(${code})`);
});
// Mock console.error
const mockConsoleError = vi.spyOn(console, 'error').mockImplementation(() => { });
describe('CLI Args', () => {
beforeEach(() => {
vi.clearAllMocks();
vi.unstubAllEnvs();
delete process.env['MYSQL_HOST'];
delete process.env['MYSQL_USER'];
delete process.env['MYSQL_PASSWORD'];
delete process.env['MYSQL_DATABASE'];
delete process.env['MYSQL_MCP_TOOL_FILTER'];
delete process.env['OAUTH_ENABLED'];
});
describe('parseArgs', () => {
it('should parse mysql connection string flag', () => {
const result = parseArgs(['--mysql', 'mysql://user:pass@host:3306/db']);
expect(result.databases).toHaveLength(1);
expect(result.databases[0]).toMatchObject({
host: 'host',
username: 'user',
password: 'pass',
database: 'db',
port: 3306
});
});
it('should parse individual mysql flags', () => {
const result = parseArgs([
'--mysql-host', 'localhost',
'--mysql-user', 'root',
'--mysql-password', 'secret',
'--mysql-database', 'testdb',
'--mysql-port', '3307'
]);
expect(result.databases).toHaveLength(1);
expect(result.databases[0]).toMatchObject({
host: 'localhost',
username: 'root',
password: 'secret',
database: 'testdb',
port: 3307
});
});
it('should use environment variables for fallback', () => {
vi.stubEnv('MYSQL_HOST', 'env-host');
vi.stubEnv('MYSQL_USER', 'env-user');
const result = parseArgs([]);
// Partial config won't create a database entry unless user AND database are present
expect(result.databases).toHaveLength(0);
vi.unstubAllEnvs();
});
it('should print help and exit when --help flag is used', () => {
const result = parseArgs(['--help']);
expect(result.shouldExit).toBe(true);
expect(mockConsoleError).toHaveBeenCalledWith(expect.stringContaining('Usage: mysql-mcp [options]'));
});
it('should print help and exit when -h flag is used', () => {
const result = parseArgs(['-h']);
expect(result.shouldExit).toBe(true);
expect(mockConsoleError).toHaveBeenCalledWith(expect.stringContaining('Usage: mysql-mcp [options]'));
});
it('should use TOOL_FILTER environment variable', () => {
vi.stubEnv('TOOL_FILTER', '-admin');
const result = parseArgs([]);
expect(result.config.toolFilter).toBe('-admin');
vi.unstubAllEnvs();
});
it('should prefer MYSQL_MCP_TOOL_FILTER over TOOL_FILTER', () => {
vi.stubEnv('TOOL_FILTER', '-admin');
vi.stubEnv('MYSQL_MCP_TOOL_FILTER', '-security');
const result = parseArgs([]);
expect(result.config.toolFilter).toBe('-security');
vi.unstubAllEnvs();
});
it('should load oauth config from environment variables', () => {
vi.stubEnv('OAUTH_ENABLED', 'true');
vi.stubEnv('OAUTH_ISSUER', 'https://env-issuer.com');
vi.stubEnv('OAUTH_AUDIENCE', 'env-aud');
vi.stubEnv('OAUTH_JWKS_URI', 'https://jwks');
vi.stubEnv('OAUTH_CLOCK_TOLERANCE', '120');
const result = parseArgs([]);
expect(result.oauth).toBeDefined();
expect(result.oauth?.enabled).toBe(true);
expect(result.oauth?.issuer).toBe('https://env-issuer.com');
expect(result.oauth?.audience).toBe('env-aud');
expect(result.oauth?.jwksUri).toBe('https://jwks');
expect(result.oauth?.clockTolerance).toBe(120);
vi.unstubAllEnvs();
});
it('should build database config from environment variables if no arguments provided', () => {
vi.stubEnv('MYSQL_HOST', 'env-host');
vi.stubEnv('MYSQL_USER', 'env-user');
vi.stubEnv('MYSQL_PASSWORD', 'env-pass');
vi.stubEnv('MYSQL_DATABASE', 'env-db');
vi.stubEnv('MYSQL_PORT', '3307');
vi.stubEnv('MYSQL_POOL_SIZE', '20');
const result = parseArgs([]);
expect(result.databases).toHaveLength(1);
expect(result.databases[0]).toEqual(expect.objectContaining({
host: 'env-host',
username: 'env-user',
port: 3307,
database: 'env-db'
}));
expect(result.databases[0].pool?.connectionLimit).toBe(20);
vi.unstubAllEnvs();
});
it('should parse transport flags', () => {
const result = parseArgs(['--transport', 'sse', '--port', '8080']);
expect(result.config.transport).toBe('sse');
expect(result.config.port).toBe(8080);
const resultShort = parseArgs(['-t', 'http', '-p', '9090']);
expect(resultShort.config.transport).toBe('http');
expect(resultShort.config.port).toBe(9090);
});
it('should parse pool config flags', () => {
vi.stubEnv('MYSQL_HOST', 'localhost');
vi.stubEnv('MYSQL_USER', 'user');
vi.stubEnv('MYSQL_DATABASE', 'db');
const resultEnv = parseArgs([
'--pool-size', '25',
'--pool-timeout', '6000',
'--pool-queue-limit', '150'
]);
expect(resultEnv.databases[0].pool?.connectionLimit).toBe(25);
expect(resultEnv.databases[0].pool?.acquireTimeout).toBe(6000);
expect(resultEnv.databases[0].pool?.queueLimit).toBe(150);
vi.unstubAllEnvs();
});
it('should parse OAuth flags', () => {
const result = parseArgs([
'--oauth-enabled',
'--oauth-issuer', 'https://auth.com',
'--oauth-audience', 'api',
'--oauth-jwks-uri', 'https://jwks',
'--oauth-clock-tolerance', '30'
]);
expect(result.oauth).toBeDefined();
expect(result.oauth?.enabled).toBe(true);
expect(result.oauth?.issuer).toBe('https://auth.com');
expect(result.oauth?.audience).toBe('api');
expect(result.oauth?.jwksUri).toBe('https://jwks');
expect(result.oauth?.clockTolerance).toBe(30);
});
it('should exit error if value argument looks like a flag', () => {
// Case: --mysql-user -flag
expect(() => parseArgs(['--mysql-user', '-flag'])).toThrow('process.exit(1)');
});
it('should NOT add database if required fields are missing despite some being present', () => {
// Case: Host provided, but User missing in both CLI and Env
vi.stubEnv('MYSQL_DATABASE', 'env-db');
// No MYSQL_USER
const result = parseArgs(['--mysql-host', 'cli-host']);
expect(result.databases).toHaveLength(0);
vi.unstubAllEnvs();
});
it('should fallback to localhost if MYSQL_HOST is missing but others are present', () => {
vi.stubEnv('MYSQL_USER', 'env-user');
vi.stubEnv('MYSQL_DATABASE', 'env-db');
// No MYSQL_HOST
const result = parseArgs(['--mysql-user', 'cli-user']);
expect(result.databases[0].host).toBe('localhost');
expect(result.databases[0].username).toBe('cli-user');
vi.unstubAllEnvs();
});
it('should parse --version flag', () => {
const result = parseArgs(['--version']);
expect(result.shouldExit).toBe(true);
expect(mockConsoleError).toHaveBeenCalledWith(expect.stringContaining('mysql-mcp version'));
});
it('should print help when --help flag is used', () => {
const result = parseArgs(['--help']);
expect(result.shouldExit).toBe(true);
expect(mockConsoleError).toHaveBeenCalledWith(expect.stringContaining('Usage: mysql-mcp [options]'));
});
});
});
