@vibescope/mcp-server 0.0.1 → 0.2.0

package/src/handlers/discovery.test.ts

@@ -1,35 +1,6 @@
-import { describe, it, expect, vi } from 'vitest';
-import type { SupabaseClient } from '@supabase/supabase-js';
-import type { HandlerContext, TokenUsage } from './types.js';
+import { describe, it, expect } from 'vitest';
 import { discoverTools, getToolInfo } from './discovery.js';
-
-// ============================================================================
-// Test Utilities
-// ============================================================================
-
-function createMockSupabase() {
-	return {} as unknown as SupabaseClient;
-}
-
-function createMockContext(supabase: SupabaseClient): HandlerContext {
-	const defaultTokenUsage: TokenUsage = {
-		callCount: 5,
-		totalTokens: 2500,
-		byTool: {},
-	};
-
-	return {
-		supabase,
-		auth: { userId: 'user-123', apiKeyId: 'api-key-123' },
-		session: {
-			instanceId: 'instance-abc',
-			currentSessionId: 'session-123',
-			currentPersona: 'Wave',
-			tokenUsage: defaultTokenUsage,
-		},
-		updateSession: vi.fn(),
-	};
-}
+import { createMockContext } from './__test-utils__.js';
 
 // ============================================================================
 // discoverTools Tests
@@ -37,8 +8,7 @@ function createMockContext(supabase: SupabaseClient): HandlerContext {
 
 describe('discoverTools', () => {
 	it('should return all categories when no category specified', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { categories: unknown[]; total_tools: number };
@@ -50,8 +20,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should return category info with tool counts', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { categories: { name: string; description: string; tool_count: number }[] };
@@ -63,8 +32,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should return tools for specific category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({ category: 'session' }, ctx);
 		const res = result.result as { category: string; description: string; tools: unknown[] };
@@ -76,8 +44,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should return tools with name and brief for specific category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({ category: 'tasks' }, ctx);
 		const res = result.result as { tools: { name: string; brief: string }[] };
@@ -89,8 +56,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should return error for unknown category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({ category: 'nonexistent' }, ctx);
 		const res = result.result as { error: string; available: string[] };
@@ -101,8 +67,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should include session category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { categories: { name: string }[] };
@@ -112,8 +77,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should include tasks category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { categories: { name: string }[] };
@@ -123,8 +87,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should include deployment category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { categories: { name: string }[] };
@@ -134,8 +97,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should include usage hint in response', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { usage: string };
@@ -152,8 +114,7 @@ describe('discoverTools', () => {
 
 describe('getToolInfo', () => {
 	it('should return error for missing tool_name', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({}, ctx);
 		const res = result.result as { error: string };
@@ -162,8 +123,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return error for unknown tool', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'nonexistent_tool' }, ctx);
 		const res = result.result as { error: string; hint: string };
@@ -173,8 +133,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for start_work_session', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'start_work_session' }, ctx);
 		const res = result.result as { tool: string; info: string };
@@ -185,8 +144,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for add_task', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'add_task' }, ctx);
 		const res = result.result as { tool: string; info: string };
@@ -197,8 +155,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for complete_task', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'complete_task' }, ctx);
 		const res = result.result as { tool: string; info: string };
@@ -208,8 +165,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for request_deployment', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'request_deployment' }, ctx);
 		const res = result.result as { tool: string; info: string };
@@ -220,8 +176,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info with parameters section', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'add_task' }, ctx);
 		const res = result.result as { info: string };
@@ -230,8 +185,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for get_help', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'get_help' }, ctx);
 		const res = result.result as { tool: string; info: string };
@@ -242,8 +196,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for discover_tools', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'discover_tools' }, ctx);
 		const res = result.result as { tool: string; info: string };