Fix unit tests

Author: charleslien

backend/src/__tests__/cost-aggregation.integration.test.ts

@@ -22,6 +22,7 @@ import { mainPrompt } from '../main-prompt'
 import * as agentRegistry from '../templates/agent-registry'
 import * as websocketAction from '../websockets/websocket-action'
 
+import type { AgentTemplate } from '../templates/types'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
 import type { WebSocket } from 'ws'
 
@@ -139,12 +140,13 @@ describe('Cost Aggregation Integration Tests', () => {
         spawnerPrompt: '',
         model: 'gpt-4o-mini',
         includeMessageHistory: true,
+        mcpServers: {},
         toolNames: ['spawn_agents'],
         spawnableAgents: ['editor'],
         systemPrompt: 'Base agent system prompt',
         instructionsPrompt: 'Base agent instructions',
         stepPrompt: 'Base agent step prompt',
-      },
+      } satisfies AgentTemplate,
       editor: {
         id: 'editor',
         displayName: 'Editor Agent',
@@ -153,12 +155,13 @@ describe('Cost Aggregation Integration Tests', () => {
         spawnerPrompt: '',
         model: 'gpt-4o-mini',
         includeMessageHistory: true,
+        mcpServers: {},
         toolNames: ['write_file'],
         spawnableAgents: [],
         systemPrompt: 'Editor agent system prompt',
         instructionsPrompt: 'Editor agent instructions',
         stepPrompt: 'Editor agent step prompt',
-      },
+      } satisfies AgentTemplate,
     }
 
     // Mock cost tracking to return 0 so only onCostCalculated contributes

backend/src/__tests__/main-prompt.test.ts

@@ -32,6 +32,7 @@ import { mainPrompt } from '../main-prompt'
 import * as processFileBlockModule from '../process-file-block'
 import * as websocketAction from '../websockets/websocket-action'
 
+import type { AgentTemplate } from '@codebuff/common/types/agent-template'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
 import type { WebSocket } from 'ws'
 
@@ -56,12 +57,13 @@ describe('mainPrompt', () => {
         spawnerPrompt: '',
         model: 'gpt-4o-mini',
         includeMessageHistory: true,
+        mcpServers: {},
         toolNames: ['write_file', 'run_terminal_command'],
         spawnableAgents: [],
         systemPrompt: '',
         instructionsPrompt: '',
         stepPrompt: '',
-      },
+      } satisfies AgentTemplate,
       [AgentTemplateTypes.base_max]: {
         id: AgentTemplateTypes.base_max,
         displayName: 'Base Max Agent',
@@ -70,12 +72,13 @@ describe('mainPrompt', () => {
         spawnerPrompt: '',
         model: 'gpt-4o',
         includeMessageHistory: true,
+        mcpServers: {},
         toolNames: ['write_file', 'run_terminal_command'],
         spawnableAgents: [],
         systemPrompt: '',
         instructionsPrompt: '',
         stepPrompt: '',
-      },
+      } satisfies AgentTemplate,
     }
   })
 

backend/src/__tests__/malformed-tool-call.test.ts

@@ -311,7 +311,7 @@ describe('malformed tool call error handling', () => {
     expect(errorMessage).toBeDefined()
     expect(
       (errorMessage?.content.output?.[0] as any)?.value?.errorMessage,
-    ).toContain('Tool not found: unknown_tool')
+    ).toContain('Tool unknown_tool not found')
   })
 
   test('should not affect valid tool calls in message history', async () => {