@vibescope/mcp-server 0.2.5 → 0.2.6

package/src/handlers/milestones.test.ts

@@ -1,475 +1,475 @@
-import { describe, it, expect, beforeEach } from 'vitest';
-import {
-	addMilestone,
-	updateMilestone,
-	completeMilestone,
-	deleteMilestone,
-	getMilestones,
-} from './milestones.js';
-import { ValidationError } from '../validators.js';
-import { createMockContext } from './__test-utils__.js';
-import { mockApiClient } from './__test-setup__.js';
-
-// ============================================================================
-// addMilestone Tests
-// ============================================================================
-
-describe('addMilestone', () => {
-	beforeEach(() => mockApiClient.addMilestone.mockReset());
-
-	it('should throw error for missing task_id', async () => {
-		const ctx = createMockContext();
-
-		await expect(
-			addMilestone({ title: 'Milestone 1' }, ctx)
-		).rejects.toThrow(ValidationError);
-	});
-
-	it('should throw error for invalid task_id UUID', async () => {
-		const ctx = createMockContext();
-
-		await expect(
-			addMilestone({ task_id: 'invalid', title: 'Milestone 1' }, ctx)
-		).rejects.toThrow(ValidationError);
-	});
-
-	it('should throw error for missing title', async () => {
-		const ctx = createMockContext();
-
-		await expect(
-			addMilestone({ task_id: '123e4567-e89b-12d3-a456-426614174000' }, ctx)
-		).rejects.toThrow(ValidationError);
-	});
-
-	it('should add milestone successfully', async () => {
-		mockApiClient.addMilestone.mockResolvedValue({
-			ok: true,
-			data: { milestone_id: 'milestone-1' },
-		});
-		const ctx = createMockContext();
-
-		const result = await addMilestone(
-			{
-				task_id: '123e4567-e89b-12d3-a456-426614174000',
-				title: 'First Milestone',
-			},
-			ctx
-		);
-
-		expect(result.result).toMatchObject({
-			success: true,
-			milestone_id: 'milestone-1',
-		});
-	});
-
-	it('should call API client with correct parameters', async () => {
-		mockApiClient.addMilestone.mockResolvedValue({
-			ok: true,
-			data: { milestone_id: 'milestone-1' },
-		});
-		const ctx = createMockContext();
-
-		await addMilestone(
-			{
-				task_id: '123e4567-e89b-12d3-a456-426614174000',
-				title: 'Test Milestone',
-				description: 'A test milestone',
-				order_index: 5,
-			},
-			ctx
-		);
-
-		expect(mockApiClient.addMilestone).toHaveBeenCalledWith(
-			'123e4567-e89b-12d3-a456-426614174000',
-			{
-				title: 'Test Milestone',
-				description: 'A test milestone',
-				order_index: 5,
-			},
-			'session-123'
-		);
-	});
-
-	it('should return error when API call fails', async () => {
-		mockApiClient.addMilestone.mockResolvedValue({
-			ok: false,
-			error: 'Task not found',
-		});
-		const ctx = createMockContext();
-
-		const result = await addMilestone({
-			task_id: '123e4567-e89b-12d3-a456-426614174000',
-			title: 'Milestone 1',
-		}, ctx);
-
-		expect(result.isError).toBe(true);
-		expect(result.result).toMatchObject({ error: 'Task not found' });
-	});
-});
-
-// ============================================================================
-// updateMilestone Tests
-// ============================================================================
-
-describe('updateMilestone', () => {
-	beforeEach(() => mockApiClient.updateMilestone.mockReset());
-
-	it('should throw error for missing milestone_id', async () => {
-		const ctx = createMockContext();
-
-		await expect(updateMilestone({}, ctx)).rejects.toThrow(ValidationError);
-	});
-
-	it('should throw error for invalid milestone_id UUID', async () => {
-		const ctx = createMockContext();
-
-		await expect(
-			updateMilestone({ milestone_id: 'invalid', title: 'New Title' }, ctx)
-		).rejects.toThrow(ValidationError);
-	});
-
-	it('should throw error when no fields to update', async () => {
-		const ctx = createMockContext();
-
-		await expect(
-			updateMilestone({ milestone_id: '123e4567-e89b-12d3-a456-426614174000' }, ctx)
-		).rejects.toThrow(ValidationError);
-	});
-
-	it('should throw error for invalid status', async () => {
-		const ctx = createMockContext();
-
-		await expect(
-			updateMilestone({
-				milestone_id: '123e4567-e89b-12d3-a456-426614174000',
-				status: 'invalid_status',
-			}, ctx)
-		).rejects.toThrow(ValidationError);
-	});
-
-	it('should update title successfully', async () => {
-		mockApiClient.updateMilestone.mockResolvedValue({
-			ok: true,
-			data: { milestone: { id: 'milestone-1', title: 'Updated Title' } },
-		});
-		const ctx = createMockContext();
-
-		const result = await updateMilestone(
-			{
-				milestone_id: '123e4567-e89b-12d3-a456-426614174000',
-				title: 'Updated Title',
-			},
-			ctx
-		);
-
-		expect(result.result).toMatchObject({
-			success: true,
-		});
-	});
-
-	it('should call API client with correct parameters', async () => {
-		mockApiClient.updateMilestone.mockResolvedValue({
-			ok: true,
-			data: { milestone: { id: 'milestone-1', status: 'completed' } },
-		});
-		const ctx = createMockContext();
-
-		await updateMilestone(
-			{
-				milestone_id: '123e4567-e89b-12d3-a456-426614174000',
-				status: 'completed',
-			},
-			ctx
-		);
-
-		expect(mockApiClient.updateMilestone).toHaveBeenCalledWith(
-			'123e4567-e89b-12d3-a456-426614174000',
-			{
-				title: undefined,
-				description: undefined,
-				status: 'completed',
-				order_index: undefined,
-			}
-		);
-	});
-
-	it('should return error when API call fails', async () => {
-		mockApiClient.updateMilestone.mockResolvedValue({
-			ok: false,
-			error: 'Milestone not found',
-		});
-		const ctx = createMockContext();
-
-		const result = await updateMilestone({
-			milestone_id: '123e4567-e89b-12d3-a456-426614174000',
-			title: 'New Title',
-		}, ctx);
-
-		expect(result.isError).toBe(true);
-		expect(result.result).toMatchObject({ error: 'Milestone not found' });
-	});
-});
-
-// ============================================================================
-// completeMilestone Tests
-// ============================================================================
-
-describe('completeMilestone', () => {
-	beforeEach(() => mockApiClient.completeMilestone.mockReset());
-
-	it('should throw error for missing milestone_id', async () => {
-		const ctx = createMockContext();
-
-		await expect(completeMilestone({}, ctx)).rejects.toThrow(ValidationError);
-	});
-
-	it('should throw error for invalid milestone_id UUID', async () => {
-		const ctx = createMockContext();
-
-		await expect(
-			completeMilestone({ milestone_id: 'invalid' }, ctx)
-		).rejects.toThrow(ValidationError);
-	});
-
-	it('should complete milestone successfully', async () => {
-		mockApiClient.completeMilestone.mockResolvedValue({
-			ok: true,
-			data: { milestone: { id: 'milestone-1', status: 'completed' } },
-		});
-		const ctx = createMockContext();
-
-		const result = await completeMilestone(
-			{ milestone_id: '123e4567-e89b-12d3-a456-426614174000' },
-			ctx
-		);
-
-		expect(result.result).toMatchObject({
-			success: true,
-		});
-	});
-
-	it('should call API client completeMilestone', async () => {
-		mockApiClient.completeMilestone.mockResolvedValue({
-			ok: true,
-			data: { milestone: { id: 'milestone-1' } },
-		});
-		const ctx = createMockContext();
-
-		await completeMilestone(
-			{ milestone_id: '123e4567-e89b-12d3-a456-426614174000' },
-			ctx
-		);
-
-		expect(mockApiClient.completeMilestone).toHaveBeenCalledWith(
-			'123e4567-e89b-12d3-a456-426614174000'
-		);
-	});
-
-	it('should return error when API call fails', async () => {
-		mockApiClient.completeMilestone.mockResolvedValue({
-			ok: false,
-			error: 'Milestone not found',
-		});
-		const ctx = createMockContext();
-
-		const result = await completeMilestone({ milestone_id: '123e4567-e89b-12d3-a456-426614174000' }, ctx);
-
-		expect(result.isError).toBe(true);
-		expect(result.result).toMatchObject({ error: 'Milestone not found' });
-	});
-});
-
-// ============================================================================
-// deleteMilestone Tests
-// ============================================================================
-
-describe('deleteMilestone', () => {
-	beforeEach(() => mockApiClient.deleteMilestone.mockReset());
-
-	it('should throw error for missing milestone_id', async () => {
-		const ctx = createMockContext();
-
-		await expect(deleteMilestone({}, ctx)).rejects.toThrow(ValidationError);
-	});
-
-	it('should throw error for invalid milestone_id UUID', async () => {
-		const ctx = createMockContext();
-
-		await expect(
-			deleteMilestone({ milestone_id: 'invalid' }, ctx)
-		).rejects.toThrow(ValidationError);
-	});
-
-	it('should delete milestone successfully', async () => {
-		mockApiClient.deleteMilestone.mockResolvedValue({
-			ok: true,
-			data: { success: true },
-		});
-		const ctx = createMockContext();
-
-		const result = await deleteMilestone(
-			{ milestone_id: '123e4567-e89b-12d3-a456-426614174000' },
-			ctx
-		);
-
-		expect(result.result).toMatchObject({
-			success: true,
-			message: 'Milestone deleted',
-		});
-	});
-
-	it('should call API client deleteMilestone', async () => {
-		mockApiClient.deleteMilestone.mockResolvedValue({
-			ok: true,
-			data: { success: true },
-		});
-		const ctx = createMockContext();
-
-		await deleteMilestone(
-			{ milestone_id: '123e4567-e89b-12d3-a456-426614174000' },
-			ctx
-		);
-
-		expect(mockApiClient.deleteMilestone).toHaveBeenCalledWith(
-			'123e4567-e89b-12d3-a456-426614174000'
-		);
-	});
-
-	it('should return error when API call fails', async () => {
-		mockApiClient.deleteMilestone.mockResolvedValue({
-			ok: false,
-			error: 'Milestone not found',
-		});
-		const ctx = createMockContext();
-
-		const result = await deleteMilestone({ milestone_id: '123e4567-e89b-12d3-a456-426614174000' }, ctx);
-
-		expect(result.isError).toBe(true);
-		expect(result.result).toMatchObject({ error: 'Milestone not found' });
-	});
-});
-
-// ============================================================================
-// getMilestones Tests
-// ============================================================================
-
-describe('getMilestones', () => {
-	beforeEach(() => mockApiClient.getMilestones.mockReset());
-
-	it('should throw error for missing task_id', async () => {
-		const ctx = createMockContext();
-
-		await expect(getMilestones({}, ctx)).rejects.toThrow(ValidationError);
-	});
-
-	it('should throw error for invalid task_id UUID', async () => {
-		const ctx = createMockContext();
-
-		await expect(
-			getMilestones({ task_id: 'invalid' }, ctx)
-		).rejects.toThrow(ValidationError);
-	});
-
-	it('should return empty list with zero stats', async () => {
-		mockApiClient.getMilestones.mockResolvedValue({
-			ok: true,
-			data: {
-				milestones: [],
-				stats: {
-					total: 0,
-					completed: 0,
-					in_progress: 0,
-					pending: 0,
-					progress_percentage: 0,
-				},
-			},
-		});
-		const ctx = createMockContext();
-
-		const result = await getMilestones(
-			{ task_id: '123e4567-e89b-12d3-a456-426614174000' },
-			ctx
-		);
-
-		expect(result.result).toMatchObject({
-			milestones: [],
-			stats: {
-				total: 0,
-				completed: 0,
-				in_progress: 0,
-				pending: 0,
-				progress_percentage: 0,
-			},
-		});
-	});
-
-	it('should return milestones with stats from API', async () => {
-		const mockMilestones = [
-			{ id: 'm1', title: 'Step 1', status: 'completed', order_index: 0 },
-			{ id: 'm2', title: 'Step 2', status: 'in_progress', order_index: 1 },
-			{ id: 'm3', title: 'Step 3', status: 'pending', order_index: 2 },
-			{ id: 'm4', title: 'Step 4', status: 'pending', order_index: 3 },
-		];
-
-		mockApiClient.getMilestones.mockResolvedValue({
-			ok: true,
-			data: {
-				milestones: mockMilestones,
-				stats: {
-					total: 4,
-					completed: 1,
-					in_progress: 1,
-					pending: 2,
-					progress_percentage: 25,
-				},
-			},
-		});
-		const ctx = createMockContext();
-
-		const result = await getMilestones(
-			{ task_id: '123e4567-e89b-12d3-a456-426614174000' },
-			ctx
-		);
-
-		expect(result.result).toMatchObject({
-			milestones: mockMilestones,
-			stats: {
-				total: 4,
-				completed: 1,
-				in_progress: 1,
-				pending: 2,
-				progress_percentage: 25,
-			},
-		});
-	});
-
-	it('should call API client getMilestones', async () => {
-		mockApiClient.getMilestones.mockResolvedValue({
-			ok: true,
-			data: { milestones: [], stats: {} },
-		});
-		const ctx = createMockContext();
-
-		await getMilestones(
-			{ task_id: '123e4567-e89b-12d3-a456-426614174000' },
-			ctx
-		);
-
-		expect(mockApiClient.getMilestones).toHaveBeenCalledWith(
-			'123e4567-e89b-12d3-a456-426614174000'
-		);
-	});
-
-	it('should return error when API call fails', async () => {
-		mockApiClient.getMilestones.mockResolvedValue({
-			ok: false,
-			error: 'Task not found',
-		});
-		const ctx = createMockContext();
-
-		const result = await getMilestones({ task_id: '123e4567-e89b-12d3-a456-426614174000' }, ctx);
-
-		expect(result.isError).toBe(true);
-		expect(result.result).toMatchObject({ error: 'Task not found' });
-	});
-});
+import { describe, it, expect, beforeEach } from 'vitest';
+import {
+	addMilestone,
+	updateMilestone,
+	completeMilestone,
+	deleteMilestone,
+	getMilestones,
+} from './milestones.js';
+import { ValidationError } from '../validators.js';
+import { createMockContext } from './__test-utils__.js';
+import { mockApiClient } from './__test-setup__.js';
+
+// ============================================================================
+// addMilestone Tests
+// ============================================================================
+
+describe('addMilestone', () => {
+	beforeEach(() => mockApiClient.addMilestone.mockReset());
+
+	it('should throw error for missing task_id', async () => {
+		const ctx = createMockContext();
+
+		await expect(
+			addMilestone({ title: 'Milestone 1' }, ctx)
+		).rejects.toThrow(ValidationError);
+	});
+
+	it('should throw error for invalid task_id UUID', async () => {
+		const ctx = createMockContext();
+
+		await expect(
+			addMilestone({ task_id: 'invalid', title: 'Milestone 1' }, ctx)
+		).rejects.toThrow(ValidationError);
+	});
+
+	it('should throw error for missing title', async () => {
+		const ctx = createMockContext();
+
+		await expect(
+			addMilestone({ task_id: '123e4567-e89b-12d3-a456-426614174000' }, ctx)
+		).rejects.toThrow(ValidationError);
+	});
+
+	it('should add milestone successfully', async () => {
+		mockApiClient.addMilestone.mockResolvedValue({
+			ok: true,
+			data: { milestone_id: 'milestone-1' },
+		});
+		const ctx = createMockContext();
+
+		const result = await addMilestone(
+			{
+				task_id: '123e4567-e89b-12d3-a456-426614174000',
+				title: 'First Milestone',
+			},
+			ctx
+		);
+
+		expect(result.result).toMatchObject({
+			success: true,
+			milestone_id: 'milestone-1',
+		});
+	});
+
+	it('should call API client with correct parameters', async () => {
+		mockApiClient.addMilestone.mockResolvedValue({
+			ok: true,
+			data: { milestone_id: 'milestone-1' },
+		});
+		const ctx = createMockContext();
+
+		await addMilestone(
+			{
+				task_id: '123e4567-e89b-12d3-a456-426614174000',
+				title: 'Test Milestone',
+				description: 'A test milestone',
+				order_index: 5,
+			},
+			ctx
+		);
+
+		expect(mockApiClient.addMilestone).toHaveBeenCalledWith(
+			'123e4567-e89b-12d3-a456-426614174000',
+			{
+				title: 'Test Milestone',
+				description: 'A test milestone',
+				order_index: 5,
+			},
+			'session-123'
+		);
+	});
+
+	it('should return error when API call fails', async () => {
+		mockApiClient.addMilestone.mockResolvedValue({
+			ok: false,
+			error: 'Task not found',
+		});
+		const ctx = createMockContext();
+
+		const result = await addMilestone({
+			task_id: '123e4567-e89b-12d3-a456-426614174000',
+			title: 'Milestone 1',
+		}, ctx);
+
+		expect(result.isError).toBe(true);
+		expect(result.result).toMatchObject({ error: 'Task not found' });
+	});
+});
+
+// ============================================================================
+// updateMilestone Tests
+// ============================================================================
+
+describe('updateMilestone', () => {
+	beforeEach(() => mockApiClient.updateMilestone.mockReset());
+
+	it('should throw error for missing milestone_id', async () => {
+		const ctx = createMockContext();
+
+		await expect(updateMilestone({}, ctx)).rejects.toThrow(ValidationError);
+	});
+
+	it('should throw error for invalid milestone_id UUID', async () => {
+		const ctx = createMockContext();
+
+		await expect(
+			updateMilestone({ milestone_id: 'invalid', title: 'New Title' }, ctx)
+		).rejects.toThrow(ValidationError);
+	});
+
+	it('should throw error when no fields to update', async () => {
+		const ctx = createMockContext();
+
+		await expect(
+			updateMilestone({ milestone_id: '123e4567-e89b-12d3-a456-426614174000' }, ctx)
+		).rejects.toThrow(ValidationError);
+	});
+
+	it('should throw error for invalid status', async () => {
+		const ctx = createMockContext();
+
+		await expect(
+			updateMilestone({
+				milestone_id: '123e4567-e89b-12d3-a456-426614174000',
+				status: 'invalid_status',
+			}, ctx)
+		).rejects.toThrow(ValidationError);
+	});
+
+	it('should update title successfully', async () => {
+		mockApiClient.updateMilestone.mockResolvedValue({
+			ok: true,
+			data: { milestone: { id: 'milestone-1', title: 'Updated Title' } },
+		});
+		const ctx = createMockContext();
+
+		const result = await updateMilestone(
+			{
+				milestone_id: '123e4567-e89b-12d3-a456-426614174000',
+				title: 'Updated Title',
+			},
+			ctx
+		);
+
+		expect(result.result).toMatchObject({
+			success: true,
+		});
+	});
+
+	it('should call API client with correct parameters', async () => {
+		mockApiClient.updateMilestone.mockResolvedValue({
+			ok: true,
+			data: { milestone: { id: 'milestone-1', status: 'completed' } },
+		});
+		const ctx = createMockContext();
+
+		await updateMilestone(
+			{
+				milestone_id: '123e4567-e89b-12d3-a456-426614174000',
+				status: 'completed',
+			},
+			ctx
+		);
+
+		expect(mockApiClient.updateMilestone).toHaveBeenCalledWith(
+			'123e4567-e89b-12d3-a456-426614174000',
+			{
+				title: undefined,
+				description: undefined,
+				status: 'completed',
+				order_index: undefined,
+			}
+		);
+	});
+
+	it('should return error when API call fails', async () => {
+		mockApiClient.updateMilestone.mockResolvedValue({
+			ok: false,
+			error: 'Milestone not found',
+		});
+		const ctx = createMockContext();
+
+		const result = await updateMilestone({
+			milestone_id: '123e4567-e89b-12d3-a456-426614174000',
+			title: 'New Title',
+		}, ctx);
+
+		expect(result.isError).toBe(true);
+		expect(result.result).toMatchObject({ error: 'Milestone not found' });
+	});
+});
+
+// ============================================================================
+// completeMilestone Tests
+// ============================================================================
+
+describe('completeMilestone', () => {
+	beforeEach(() => mockApiClient.completeMilestone.mockReset());
+
+	it('should throw error for missing milestone_id', async () => {
+		const ctx = createMockContext();
+
+		await expect(completeMilestone({}, ctx)).rejects.toThrow(ValidationError);
+	});
+
+	it('should throw error for invalid milestone_id UUID', async () => {
+		const ctx = createMockContext();
+
+		await expect(
+			completeMilestone({ milestone_id: 'invalid' }, ctx)
+		).rejects.toThrow(ValidationError);
+	});
+
+	it('should complete milestone successfully', async () => {
+		mockApiClient.completeMilestone.mockResolvedValue({
+			ok: true,
+			data: { milestone: { id: 'milestone-1', status: 'completed' } },
+		});
+		const ctx = createMockContext();
+
+		const result = await completeMilestone(
+			{ milestone_id: '123e4567-e89b-12d3-a456-426614174000' },
+			ctx
+		);
+
+		expect(result.result).toMatchObject({
+			success: true,
+		});
+	});
+
+	it('should call API client completeMilestone', async () => {
+		mockApiClient.completeMilestone.mockResolvedValue({
+			ok: true,
+			data: { milestone: { id: 'milestone-1' } },
+		});
+		const ctx = createMockContext();
+
+		await completeMilestone(
+			{ milestone_id: '123e4567-e89b-12d3-a456-426614174000' },
+			ctx
+		);
+
+		expect(mockApiClient.completeMilestone).toHaveBeenCalledWith(
+			'123e4567-e89b-12d3-a456-426614174000'
+		);
+	});
+
+	it('should return error when API call fails', async () => {
+		mockApiClient.completeMilestone.mockResolvedValue({
+			ok: false,
+			error: 'Milestone not found',
+		});
+		const ctx = createMockContext();
+
+		const result = await completeMilestone({ milestone_id: '123e4567-e89b-12d3-a456-426614174000' }, ctx);
+
+		expect(result.isError).toBe(true);
+		expect(result.result).toMatchObject({ error: 'Milestone not found' });
+	});
+});
+
+// ============================================================================
+// deleteMilestone Tests
+// ============================================================================
+
+describe('deleteMilestone', () => {
+	beforeEach(() => mockApiClient.deleteMilestone.mockReset());
+
+	it('should throw error for missing milestone_id', async () => {
+		const ctx = createMockContext();
+
+		await expect(deleteMilestone({}, ctx)).rejects.toThrow(ValidationError);
+	});
+
+	it('should throw error for invalid milestone_id UUID', async () => {
+		const ctx = createMockContext();
+
+		await expect(
+			deleteMilestone({ milestone_id: 'invalid' }, ctx)
+		).rejects.toThrow(ValidationError);
+	});
+
+	it('should delete milestone successfully', async () => {
+		mockApiClient.deleteMilestone.mockResolvedValue({
+			ok: true,
+			data: { success: true },
+		});
+		const ctx = createMockContext();
+
+		const result = await deleteMilestone(
+			{ milestone_id: '123e4567-e89b-12d3-a456-426614174000' },
+			ctx
+		);
+
+		expect(result.result).toMatchObject({
+			success: true,
+			message: 'Milestone deleted',
+		});
+	});
+
+	it('should call API client deleteMilestone', async () => {
+		mockApiClient.deleteMilestone.mockResolvedValue({
+			ok: true,
+			data: { success: true },
+		});
+		const ctx = createMockContext();
+
+		await deleteMilestone(
+			{ milestone_id: '123e4567-e89b-12d3-a456-426614174000' },
+			ctx
+		);
+
+		expect(mockApiClient.deleteMilestone).toHaveBeenCalledWith(
+			'123e4567-e89b-12d3-a456-426614174000'
+		);
+	});
+
+	it('should return error when API call fails', async () => {
+		mockApiClient.deleteMilestone.mockResolvedValue({
+			ok: false,
+			error: 'Milestone not found',
+		});
+		const ctx = createMockContext();
+
+		const result = await deleteMilestone({ milestone_id: '123e4567-e89b-12d3-a456-426614174000' }, ctx);
+
+		expect(result.isError).toBe(true);
+		expect(result.result).toMatchObject({ error: 'Milestone not found' });
+	});
+});
+
+// ============================================================================
+// getMilestones Tests
+// ============================================================================
+
+describe('getMilestones', () => {
+	beforeEach(() => mockApiClient.getMilestones.mockReset());
+
+	it('should throw error for missing task_id', async () => {
+		const ctx = createMockContext();
+
+		await expect(getMilestones({}, ctx)).rejects.toThrow(ValidationError);
+	});
+
+	it('should throw error for invalid task_id UUID', async () => {
+		const ctx = createMockContext();
+
+		await expect(
+			getMilestones({ task_id: 'invalid' }, ctx)
+		).rejects.toThrow(ValidationError);
+	});
+
+	it('should return empty list with zero stats', async () => {
+		mockApiClient.getMilestones.mockResolvedValue({
+			ok: true,
+			data: {
+				milestones: [],
+				stats: {
+					total: 0,
+					completed: 0,
+					in_progress: 0,
+					pending: 0,
+					progress_percentage: 0,
+				},
+			},
+		});
+		const ctx = createMockContext();
+
+		const result = await getMilestones(
+			{ task_id: '123e4567-e89b-12d3-a456-426614174000' },
+			ctx
+		);
+
+		expect(result.result).toMatchObject({
+			milestones: [],
+			stats: {
+				total: 0,
+				completed: 0,
+				in_progress: 0,
+				pending: 0,
+				progress_percentage: 0,
+			},
+		});
+	});
+
+	it('should return milestones with stats from API', async () => {
+		const mockMilestones = [
+			{ id: 'm1', title: 'Step 1', status: 'completed', order_index: 0 },
+			{ id: 'm2', title: 'Step 2', status: 'in_progress', order_index: 1 },
+			{ id: 'm3', title: 'Step 3', status: 'pending', order_index: 2 },
+			{ id: 'm4', title: 'Step 4', status: 'pending', order_index: 3 },
+		];
+
+		mockApiClient.getMilestones.mockResolvedValue({
+			ok: true,
+			data: {
+				milestones: mockMilestones,
+				stats: {
+					total: 4,
+					completed: 1,
+					in_progress: 1,
+					pending: 2,
+					progress_percentage: 25,
+				},
+			},
+		});
+		const ctx = createMockContext();
+
+		const result = await getMilestones(
+			{ task_id: '123e4567-e89b-12d3-a456-426614174000' },
+			ctx
+		);
+
+		expect(result.result).toMatchObject({
+			milestones: mockMilestones,
+			stats: {
+				total: 4,
+				completed: 1,
+				in_progress: 1,
+				pending: 2,
+				progress_percentage: 25,
+			},
+		});
+	});
+
+	it('should call API client getMilestones', async () => {
+		mockApiClient.getMilestones.mockResolvedValue({
+			ok: true,
+			data: { milestones: [], stats: {} },
+		});
+		const ctx = createMockContext();
+
+		await getMilestones(
+			{ task_id: '123e4567-e89b-12d3-a456-426614174000' },
+			ctx
+		);
+
+		expect(mockApiClient.getMilestones).toHaveBeenCalledWith(
+			'123e4567-e89b-12d3-a456-426614174000'
+		);
+	});
+
+	it('should return error when API call fails', async () => {
+		mockApiClient.getMilestones.mockResolvedValue({
+			ok: false,
+			error: 'Task not found',
+		});
+		const ctx = createMockContext();
+
+		const result = await getMilestones({ task_id: '123e4567-e89b-12d3-a456-426614174000' }, ctx);
+
+		expect(result.isError).toBe(true);
+		expect(result.result).toMatchObject({ error: 'Task not found' });
+	});
+});