remove websockets from tests

Author: charleslien

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

@@ -22,7 +22,6 @@ import { handleSpawnAgents } from '../tools/handlers/tool/spawn-agents'
 
 import type { AgentState } from '@codebuff/common/types/session-state'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
-import type { WebSocket } from 'ws'
 
 const mockFileContext: ProjectFileContext = {
   projectRoot: '/test',
@@ -137,7 +136,6 @@ describe('Cost Aggregation System', () => {
       }
 
       const mockValidatedState = {
-        ws: new MockWebSocket() as unknown as WebSocket,
         fingerprintId: 'test-fingerprint',
         userId: 'test-user',
         agentTemplate: mockAgentTemplate,
@@ -212,7 +210,6 @@ describe('Cost Aggregation System', () => {
       }
 
       const mockValidatedState = {
-        ws: new MockWebSocket() as unknown as WebSocket,
         fingerprintId: 'test-fingerprint',
         userId: 'test-user',
         agentTemplate: mockAgentTemplate,
@@ -371,7 +368,6 @@ describe('Cost Aggregation System', () => {
 
       // Mock subagent spawning that adds their costs
       const mockValidatedState = {
-        ws: new MockWebSocket() as unknown as WebSocket,
         fingerprintId: 'test-fingerprint',
         userId: 'test-user',
         agentTemplate: mockAgentTemplate,

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

@@ -1,19 +1,8 @@
-import { describe, it, expect, beforeEach } from 'bun:test'
+import { describe, it, expect } from 'bun:test'
 
 import type { WebSocket } from 'ws'
 
 describe('Backend Tool Call Schema', () => {
-  let mockWs: any
-
-  beforeEach(() => {
-    // Create a simple mock WebSocket
-    mockWs = {
-      send: () => {},
-      on: () => {},
-      close: () => {},
-    }
-  })
-
   it('should validate tool call request structure', () => {
     const toolCallRequest = {
       type: 'tool-call-request' as const,
@@ -109,26 +98,23 @@ describe('Backend Tool Call Schema', () => {
   })
 
   it('should generate mock project structure analysis', async () => {
-    const analysis = await generateMockProjectStructureAnalysis(mockWs)
+    const analysis = await generateMockProjectStructureAnalysis()
 
     expect(analysis).toContain('## Project Analysis')
     expect(analysis).toContain('TypeScript/JavaScript/JSON files')
     expect(typeof analysis).toBe('string')
   })
 
   it('should generate mock dependency analysis', async () => {
-    const analysis = await generateMockDependencyAnalysis(mockWs)
+    const analysis = await generateMockDependencyAnalysis()
 
     expect(analysis).toContain('## Dependency Analysis')
     expect(analysis).toContain('Declared Dependencies')
     expect(typeof analysis).toBe('string')
   })
 
   it('should handle error scenarios in mock generators', async () => {
-    const errorAnalysis = await generateMockProjectStructureAnalysis(
-      mockWs,
-      true,
-    )
+    const errorAnalysis = await generateMockProjectStructureAnalysis(true)
 
     expect(errorAnalysis).toContain('Project analysis failed')
     expect(typeof errorAnalysis).toBe('string')
@@ -141,7 +127,6 @@ describe('Backend Tool Call Schema', () => {
  * based on the current context or user request
  */
 export async function generateMockProjectStructureAnalysis(
-  ws: WebSocket,
   simulateError: boolean = false,
 ): Promise<string> {
   try {
@@ -228,7 +213,6 @@ export async function generateMockProjectStructureAnalysis(
  * Dynamically searches for imports and analyzes dependencies
  */
 export async function generateMockDependencyAnalysis(
-  ws: WebSocket,
   searchPattern?: string,
 ): Promise<string> {
   try {