import { describe, it, expect, beforeEach, vi } from 'vitest';
import { ToolRegistry } from '../../src/tools/index.js';
// Mock all clients
vi.mock('../../src/clients/intervals.js', () => ({
IntervalsClient: vi.fn().mockImplementation(function() {
return {
getActivities: vi.fn().mockResolvedValue([]),
getPlannedEvents: vi.fn().mockResolvedValue([]),
getFitnessMetrics: vi.fn().mockResolvedValue([]),
getTrainingLoadTrends: vi.fn().mockResolvedValue({ data: [], summary: {} }),
getAthleteTimezone: vi.fn().mockResolvedValue('America/New_York'),
getAthleteProfile: vi.fn().mockResolvedValue({ id: 'test', sports: [] }),
getActivityIntervals: vi.fn().mockResolvedValue({ activity_id: 'test', intervals: [], groups: [] }),
getSportSettingsForSport: vi.fn().mockResolvedValue({ sport: 'cycling', settings: {} }),
getUnitPreferences: vi.fn().mockResolvedValue({ system: 'metric', weight: 'kg', temperature: 'celsius' }),
};
}),
}));
vi.mock('../../src/clients/whoop.js', () => ({
WhoopClient: vi.fn().mockImplementation(function() {
return {
getTodayRecovery: vi.fn().mockResolvedValue(null),
getStrainData: vi.fn().mockResolvedValue([]),
getRecoveries: vi.fn().mockResolvedValue([]),
getWorkouts: vi.fn().mockResolvedValue([]),
setTimezoneGetter: vi.fn(),
};
}),
}));
vi.mock('../../src/clients/trainerroad.js', () => ({
TrainerRoadClient: vi.fn().mockImplementation(function() {
return {
getTodayWorkouts: vi.fn().mockResolvedValue([]),
getPlannedWorkouts: vi.fn().mockResolvedValue([]),
getUpcomingWorkouts: vi.fn().mockResolvedValue([]),
};
}),
}));
describe('ToolRegistry', () => {
let registry: ToolRegistry;
beforeEach(() => {
vi.clearAllMocks();
registry = new ToolRegistry({
intervals: { apiKey: 'test', athleteId: 'test' },
whoop: {
accessToken: 'test',
refreshToken: 'test',
clientId: 'test',
clientSecret: 'test',
},
trainerroad: { calendarUrl: 'https://test.com' },
});
});
describe('constructor', () => {
it('should create registry with all clients', () => {
expect(registry).toBeDefined();
});
it('should create registry without Whoop client', () => {
const registryWithoutWhoop = new ToolRegistry({
intervals: { apiKey: 'test', athleteId: 'test' },
whoop: null,
trainerroad: { calendarUrl: 'https://test.com' },
});
expect(registryWithoutWhoop).toBeDefined();
});
it('should create registry without TrainerRoad client', () => {
const registryWithoutTr = new ToolRegistry({
intervals: { apiKey: 'test', athleteId: 'test' },
whoop: {
accessToken: 'test',
refreshToken: 'test',
clientId: 'test',
clientSecret: 'test',
},
trainerroad: null,
});
expect(registryWithoutTr).toBeDefined();
});
});
describe('registerTools', () => {
it('should register tools with mock server', () => {
const registeredTools: string[] = [];
const mockServer = {
registerTool: vi.fn().mockImplementation((name: string) => {
registeredTools.push(name);
}),
};
registry.registerTools(mockServer as any);
expect(registeredTools).toContain('get_todays_summary');
expect(registeredTools).toContain('get_athlete_profile');
expect(registeredTools).toContain('get_strain_history');
expect(registeredTools).toContain('get_workout_history');
expect(registeredTools).toContain('get_recovery_trends');
expect(registeredTools).toContain('get_upcoming_workouts');
// Analysis tools
expect(registeredTools).toContain('get_training_load_trends');
expect(registeredTools).toContain('get_workout_details');
expect(registeredTools).toContain('get_workout_intervals');
expect(registeredTools).toContain('get_workout_notes');
expect(registeredTools).toContain('get_workout_weather');
expect(registeredTools).toContain('get_workout_heat_zones');
// Performance curves
expect(registeredTools).toContain('get_power_curve');
expect(registeredTools).toContain('get_pace_curve');
expect(registeredTools).toContain('get_hr_curve');
// Sports settings
expect(registeredTools).toContain('get_sports_settings');
// Wellness
expect(registeredTools).toContain('get_wellness_trends');
// Activity totals
expect(registeredTools).toContain('get_activity_totals');
// Races
expect(registeredTools).toContain('get_upcoming_races');
// Workout sync tools
expect(registeredTools).toContain('create_run_workout');
expect(registeredTools).toContain('update_workout');
expect(registeredTools).toContain('delete_workout');
expect(registeredTools).toContain('sync_trainerroad_runs');
expect(registeredTools).toContain('set_workout_intervals');
// Cycling workout tools
expect(registeredTools).toContain('get_cycling_workout_syntax');
expect(registeredTools).toContain('create_cycling_workout');
expect(registeredTools.length).toBe(27);
});
it('should call server.registerTool for each tool', () => {
const mockServer = {
registerTool: vi.fn(),
};
registry.registerTools(mockServer as any);
expect(mockServer.registerTool).toHaveBeenCalledTimes(27);
});
it('should pass config object with title, description, and annotations to each tool', () => {
const mockServer = {
registerTool: vi.fn(),
};
registry.registerTools(mockServer as any);
// Check first tool call has correct structure (registerTool uses name, config, handler)
const [name, config, handler] = mockServer.registerTool.mock.calls[0];
expect(typeof name).toBe('string');
expect(typeof config).toBe('object');
expect(typeof config.title).toBe('string'); // Human-readable title
expect(typeof config.description).toBe('string');
expect(config.annotations).toBeDefined();
expect(typeof handler).toBe('function');
// Verify all tools have titles
for (const call of mockServer.registerTool.mock.calls) {
const [toolName, toolConfig] = call;
expect(toolConfig.title, `Tool ${toolName} should have a title`).toBeDefined();
expect(typeof toolConfig.title).toBe('string');
}
});
});
});
