Fix test imports and missing test deps after rebase

Author: ammar-agent

src/browser/api.test.ts

@@ -3,7 +3,7 @@
  * Tests the invokeIPC function to ensure it behaves consistently with Electron's ipcRenderer.invoke()
  */
 
-import { describe, test, expect } from "bun:test";
+import { describe, test, expect } from "@jest/globals";
 
 // Helper to create a mock fetch that returns a specific response
 function createMockFetch(responseData: unknown) {

src/hooks/useReviewState.test.ts

@@ -6,7 +6,7 @@
  * The hook itself is a thin wrapper around usePersistedState with manual testing.
  */
 
-import { describe, it, expect } from "bun:test";
+import { describe, it, expect } from "@jest/globals";
 import type { HunkReadState } from "@/types/review";
 import { evictOldestReviews } from "./useReviewState";
 

src/runtime/initHook.test.ts

@@ -1,4 +1,4 @@
-import { describe, it, expect } from "bun:test";
+import { describe, it, expect } from "@jest/globals";
 import { LineBuffer, createLineBufferedLoggers } from "./initHook";
 import type { InitLogger } from "./Runtime";
 

src/services/aiService.test.ts

@@ -2,7 +2,7 @@
 // These tests would need to be rewritten to work with Bun's test runner
 // For now, the commandProcessor tests demonstrate our testing approach
 
-import { describe, it, expect, beforeEach } from "bun:test";
+import { describe, it, expect, beforeEach } from "@jest/globals";
 import { AIService } from "./aiService";
 import { HistoryService } from "./historyService";
 import { PartialService } from "./partialService";

src/services/historyService.test.ts

@@ -1,4 +1,4 @@
-import { describe, it, expect, beforeEach, afterEach } from "bun:test";
+import { describe, it, expect, beforeEach, afterEach } from "@jest/globals";
 import { HistoryService } from "./historyService";
 import { Config } from "@/config";
 import { createCmuxMessage } from "@/types/message";

src/services/initStateManager.test.ts

@@ -1,7 +1,7 @@
 import * as fs from "fs/promises";
 import * as path from "path";
 import * as os from "os";
-import { describe, it, expect, beforeEach, afterEach } from "bun:test";
+import { describe, it, expect, beforeEach, afterEach } from "@jest/globals";
 import { Config } from "@/config";
 import { InitStateManager } from "./initStateManager";
 import type { WorkspaceInitEvent } from "@/types/ipc";
@@ -53,8 +53,8 @@ describe("InitStateManager", () => {
       manager.appendOutput(workspaceId, "Installing deps...", false);
       manager.appendOutput(workspaceId, "Done!", false);
       expect(manager.getInitState(workspaceId)?.lines).toEqual([
-        { line: "Installing deps...", isError: false, timestamp: expect.any(Number) as number },
-        { line: "Done!", isError: false, timestamp: expect.any(Number) as number },
+        { line: "Installing deps...", isError: false, timestamp: expect.any(Number) },
+        { line: "Done!", isError: false, timestamp: expect.any(Number) },
       ]);
 
       // End init (await to ensure event fires)
@@ -79,8 +79,8 @@ describe("InitStateManager", () => {
 
       const state = manager.getInitState(workspaceId);
       expect(state?.lines).toEqual([
-        { line: "stdout line", isError: false, timestamp: expect.any(Number) as number },
-        { line: "stderr line", isError: true, timestamp: expect.any(Number) as number },
+        { line: "stdout line", isError: false, timestamp: expect.any(Number) },
+        { line: "stderr line", isError: true, timestamp: expect.any(Number) },
       ]);
     });
 
@@ -109,8 +109,8 @@ describe("InitStateManager", () => {
       expect(diskState?.status).toBe("success");
       expect(diskState?.exitCode).toBe(0);
       expect(diskState?.lines).toEqual([
-        { line: "Line 1", isError: false, timestamp: expect.any(Number) as number },
-        { line: "Line 2", isError: true, timestamp: expect.any(Number) as number },
+        { line: "Line 1", isError: false, timestamp: expect.any(Number) },
+        { line: "Line 2", isError: true, timestamp: expect.any(Number) },
       ]);
     });
 
@@ -221,15 +221,23 @@ describe("InitStateManager", () => {
       expect(stateAfterDelete).toBeNull();
     });
 
-    it("should clear in-memory state", () => {
+    it("should clear in-memory state", async () => {
       const workspaceId = "test-workspace";
       manager.startInit(workspaceId, "/path/to/hook");
 
       expect(manager.getInitState(workspaceId)).toBeTruthy();
 
+      // Get the init promise before clearing
+      const initPromise = manager.waitForInit(workspaceId);
+
+      // Clear in-memory state (should reject pending promises)
       manager.clearInMemoryState(workspaceId);
 
+      // Verify state is cleared
       expect(manager.getInitState(workspaceId)).toBeUndefined();
+
+      // Verify the promise was rejected
+      await expect(initPromise).rejects.toThrow("Workspace test-workspace was deleted");
     });
   });
 

src/services/partialService.test.ts

@@ -1,5 +1,5 @@
 /* eslint-disable @typescript-eslint/unbound-method */
-import { describe, test, expect, beforeEach, mock } from "bun:test";
+import { describe, test, expect, beforeEach, mock } from "@jest/globals";
 import { PartialService } from "./partialService";
 import type { HistoryService } from "./historyService";
 import type { Config } from "@/config";

src/services/streamManager.test.ts

@@ -1,4 +1,4 @@
-import { describe, test, expect, beforeEach, mock } from "bun:test";
+import { describe, test, expect, beforeEach, mock } from "@jest/globals";
 import { StreamManager } from "./streamManager";
 import type { HistoryService } from "./historyService";
 import type { PartialService } from "./partialService";

src/services/systemMessage.test.ts

@@ -3,8 +3,8 @@ import * as os from "os";
 import * as path from "path";
 import { buildSystemMessage } from "./systemMessage";
 import type { WorkspaceMetadata } from "@/types/workspace";
-import { spyOn, describe, test, expect, beforeEach, afterEach } from "bun:test";
-import type { Mock } from "bun:test";
+import { spyOn, describe, test, expect, beforeEach, afterEach } from "@jest/globals";
+import type { Mock } from "@jest/globals";
 import { LocalRuntime } from "@/runtime/LocalRuntime";
 
 describe("buildSystemMessage", () => {

src/services/tools/file_edit_operation.test.ts

@@ -4,7 +4,7 @@ import { WRITE_DENIED_PREFIX } from "@/types/tools";
 import { createRuntime } from "@/runtime/runtimeFactory";
 import type { Runtime } from "@/runtime/Runtime";
 
-import { createTestToolConfig } from "./testHelpers";
+import { createTestToolConfig, getTestDeps } from "./testHelpers";
 
 const TEST_CWD = "/tmp";
 
@@ -69,6 +69,7 @@ describe("executeFileEditOperation", () => {
         cwd: testCwd,
         runtime: mockRuntime,
         runtimeTempDir: "/tmp",
+        ...getTestDeps(),
       },
       filePath: testFilePath,
       operation: () => ({ success: true, newContent: "test", metadata: {} }),