mock logger in tests

Author: charleslien

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

@@ -17,15 +17,22 @@ import * as checkTerminalCommandModule from '../check-terminal-command'
 import * as requestFilesPrompt from '../find-files/request-files-prompt'
 import * as aisdk from '../llm-apis/vercel-ai-sdk/ai-sdk'
 import { mainPrompt } from '../main-prompt'
-import { logger } from '../util/logger'
 import * as websocketAction from '../websockets/websocket-action'
 
 import type { PrintModeEvent } from '@codebuff/common/types/print-mode'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
+import type { Logger } from '@codebuff/types/logger'
 import type { WebSocket } from 'ws'
 
 // --- Shared Mocks & Helpers ---
 
+const logger: Logger = {
+  debug: () => {},
+  info: () => {},
+  warn: () => {},
+  error: () => {},
+}
+
 class MockWebSocket {
   send(msg: string) {}
   close() {}
@@ -107,13 +114,6 @@ describe.skip('mainPrompt (Integration)', () => {
   })
 
   it('should delete a specified function while preserving other code', async () => {
-    // Mock necessary non-LLM functions
-    spyOn(logger, 'debug').mockImplementation(() => {})
-    spyOn(logger, 'error').mockImplementation(() => {})
-    spyOn(logger, 'info').mockImplementation(() => {})
-    spyOn(logger, 'warn').mockImplementation(() => {})
-    spyOn(requestFilesPrompt, 'requestRelevantFiles').mockResolvedValue([])
-
     const initialContent = `import { Message } from '@codebuff/common/types/message'
 import { withCacheControl } from '@codebuff/common/util/messages'
 

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

@@ -1,7 +1,5 @@
 import { describe, it, expect, beforeEach } from 'bun:test'
 
-import { logger } from '../util/logger'
-
 import type { WebSocket } from 'ws'
 
 describe('Backend Tool Call Schema', () => {
@@ -221,7 +219,6 @@ export async function generateMockProjectStructureAnalysis(
 
     return analysis
   } catch (error) {
-    logger.error({ error }, 'Project analysis failed')
     return `Project analysis failed: ${error instanceof Error ? error.message : error}`
   }
 }
@@ -276,7 +273,6 @@ src/utils.ts:2:import { readFileSync } from 'fs'`,
 
     return analysis
   } catch (error) {
-    logger.error({ error }, 'Dependency analysis failed')
     return `Dependency analysis failed: ${error instanceof Error ? error.message : error}`
   }
 }
@@ -318,7 +314,6 @@ export async function generateMockFileContentAnalysis(
 
     return analysis
   } catch (error) {
-    logger.error({ error }, 'File content analysis failed')
     return `File content analysis failed: ${error instanceof Error ? error.message : error}`
   }
 }