chore: migrate core (autocomplete) tests to vitest

core/autocomplete/context/root-path-context/test/RootPathContextService.test.ts → core/autocomplete/context/root-path-context/test/RootPathContextService.vitest.ts

@@ -1,3 +1,4 @@
+import { describe, test } from "vitest";
 import { PYTHON_TEST_CASES, TYPESCRIPT_TEST_CASES } from "./testCases";
 import { testRootPathContext } from "./testUtils";
 

core/autocomplete/context/root-path-context/test/testUtils.ts

@@ -1,6 +1,6 @@
-import { jest } from "@jest/globals";
 import fs from "fs";
 import path from "path";
+import { expect, vi } from "vitest";
 
 import Parser from "web-tree-sitter";
 import { Position } from "../../../..";
@@ -36,7 +36,7 @@ export async function testRootPathContext(
   const importDefinitionsService = new ImportDefinitionsService(ide);
   const service = new RootPathContextService(importDefinitionsService, ide);
 
-  const getSnippetsMock = jest
+  const getSnippetsMock = vi
     // @ts-ignore
     .spyOn(service, "getSnippets")
     // @ts-ignore
@@ -46,7 +46,7 @@ export async function testRootPathContext(
 
   // Copy the folder to the test directory
   const folderPath = path.join(
-    __dirname,
+    process.cwd(),
     "autocomplete",
     "context",
     "root-path-context",

core/autocomplete/filtering/streamTransforms/charStream.test.ts → core/autocomplete/filtering/streamTransforms/charStream.vitest.ts

@@ -1,3 +1,4 @@
+import { describe, expect, it } from "vitest";
 import { stopAtStartOf, stopAtStopTokens } from "./charStream";
 
 async function* createMockStream(chunks: string[]): AsyncGenerator<string> {

core/autocomplete/filtering/streamTransforms/lineStream.test.ts → core/autocomplete/filtering/streamTransforms/lineStream.vitest.ts

@@ -1,10 +1,10 @@
-import { jest } from "@jest/globals";
+import { beforeEach, describe, expect, it, Mock, vi } from "vitest";
 
 import * as lineStream from "./lineStream";
 
 // eslint-disable-next-line max-lines-per-function
 describe("lineStream", () => {
-  let mockFullStop: jest.Mock;
+  let mockFullStop: Mock;
 
   async function getLineGenerator(lines: any) {
     return (async function* () {
@@ -25,7 +25,7 @@ describe("lineStream", () => {
   }
 
   beforeEach(() => {
-    mockFullStop = jest.fn();
+    mockFullStop = vi.fn();
   });
 
   describe("noTopLevelKeywordsMidline", () => {

core/autocomplete/filtering/test/filter.test.ts → core/autocomplete/filtering/test/filter.vitest.ts

@@ -1,4 +1,9 @@
-import { setUpTestDir, tearDownTestDir } from "../../../test/testDir";
+import { afterAll, beforeAll, beforeEach, describe, it } from "vitest";
+import {
+  addToTestDir,
+  setUpTestDir,
+  tearDownTestDir,
+} from "../../../test/testDir";
 
 import { TEST_CASES_WITH_DIFF, TEST_CASES_WITHOUT_DIFF } from "./testCases";
 import {
@@ -18,6 +23,7 @@ describe("Autocomplete filtering tests", () => {
   beforeAll(async () => {
     tearDownTestDir();
     setUpTestDir();
+    addToTestDir([".continueignore"]);
   });
 
   afterAll(async () => {

core/autocomplete/filtering/test/util.ts

@@ -1,3 +1,4 @@
+import { expect } from "vitest";
 import MockLLM from "../../../llm/llms/Mock";
 import { testConfigHandler, testIde } from "../../../test/fixtures";
 import { joinPathsToUri } from "../../../util/uri";

core/autocomplete/generation/GeneratorReuseManager.test.ts → core/autocomplete/generation/GeneratorReuseManager.vitest.ts

@@ -1,4 +1,12 @@
-import { jest } from "@jest/globals";
+import {
+  afterEach,
+  beforeEach,
+  describe,
+  expect,
+  Mock,
+  test,
+  vi,
+} from "vitest";
 import { GeneratorReuseManager } from "./GeneratorReuseManager";
 
 function createMockGenerator(
@@ -14,7 +22,7 @@ function createMockGenerator(
       }
     }
   };
-  const newGenerator = jest
+  const newGenerator = vi
     .fn<() => AsyncGenerator<string>>()
     .mockReturnValue(mockGenerator());
 
@@ -23,15 +31,15 @@ function createMockGenerator(
 
 describe("GeneratorReuseManager", () => {
   let reuseManager: GeneratorReuseManager;
-  let onErrorMock: jest.Mock;
+  let onErrorMock: Mock;
 
   beforeEach(() => {
-    onErrorMock = jest.fn();
+    onErrorMock = vi.fn();
     reuseManager = new GeneratorReuseManager(onErrorMock);
   });
 
   afterEach(() => {
-    jest.clearAllMocks();
+    vi.clearAllMocks();
   });
 
   test("creates new generator when there is no current generator", async () => {
@@ -169,7 +177,7 @@ describe("GeneratorReuseManager", () => {
     const mockGenerator = async function* () {
       throw error;
     };
-    const newGenerator = jest
+    const newGenerator = vi
       .fn<() => AsyncGenerator<string>>()
       .mockReturnValue(mockGenerator());
 

core/autocomplete/generation/ListenableGenerator.test.ts → core/autocomplete/generation/ListenableGenerator.vitest.ts

@@ -1,4 +1,4 @@
-import { jest } from "@jest/globals";
+import { describe, expect, it, vi } from "vitest";
 
 import { ListenableGenerator } from "./ListenableGenerator";
 
@@ -16,7 +16,7 @@ describe("ListenableGenerator", () => {
   it("should yield values from the source generator via tee()", async () => {
     const values = [1, 2, 3];
     const source = asyncGenerator(values);
-    const onError = jest.fn();
+    const onError = vi.fn();
 
     const lg = new ListenableGenerator<number>(
       source,
@@ -36,15 +36,15 @@ describe("ListenableGenerator", () => {
   it("should allow listeners to receive values", async () => {
     const values = [1, 2, 3];
     const source = asyncGenerator(values, 10); // Introduce delay to simulate async behavior
-    const onError = jest.fn();
+    const onError = vi.fn();
 
     const lg = new ListenableGenerator<number>(
       source,
       onError,
       new AbortController(),
     );
 
-    const listener = jest.fn();
+    const listener = vi.fn();
 
     // Add listener after some delay to simulate late subscription
     setTimeout(() => {
@@ -64,23 +64,23 @@ describe("ListenableGenerator", () => {
   it("should buffer values for listeners added after some values have been yielded", async () => {
     const values = [1, 2, 3];
     const source = asyncGenerator(values, 10);
-    const onError = jest.fn();
+    const onError = vi.fn();
 
     const lg = new ListenableGenerator<number>(
       source,
       onError,
       new AbortController(),
     );
 
-    const initialListener = jest.fn();
+    const initialListener = vi.fn();
 
     lg.listen(initialListener);
 
     // Wait for the first value to be yielded
     await new Promise((resolve) => setTimeout(resolve, 15));
 
     // Add a second listener
-    const newListener = jest.fn();
+    const newListener = vi.fn();
     lg.listen(newListener);
 
     // Wait for generator to finish
@@ -98,7 +98,7 @@ describe("ListenableGenerator", () => {
   it("should handle cancellation", async () => {
     const values = [1, 2, 3, 4, 5];
     const source = asyncGenerator(values, 10);
-    const onError = jest.fn();
+    const onError = vi.fn();
 
     const lg = new ListenableGenerator<number>(
       source,
@@ -131,7 +131,7 @@ describe("ListenableGenerator", () => {
     }
 
     const source = errorGenerator();
-    const onError = jest.fn();
+    const onError = vi.fn();
 
     const lg = new ListenableGenerator<number>(
       source,
@@ -152,15 +152,15 @@ describe("ListenableGenerator", () => {
   it("should notify listeners when the generator ends", async () => {
     const values = [1, 2, 3];
     const source = asyncGenerator(values);
-    const onError = jest.fn();
+    const onError = vi.fn();
 
     const lg = new ListenableGenerator<number>(
       source,
       onError,
       new AbortController(),
     );
 
-    const listener = jest.fn();
+    const listener = vi.fn();
     lg.listen(listener);
 
     // Wait for the generator to finish

core/autocomplete/generation/utils.test.ts → core/autocomplete/generation/utils.vitest.ts

@@ -1,13 +1,13 @@
-import { jest } from "@jest/globals";
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 import { stopAfterMaxProcessingTime } from "./utils";
 
 describe("stopAfterMaxProcessingTime", () => {
   beforeEach(() => {
-    jest.useFakeTimers();
+    vi.useFakeTimers();
   });
 
   afterEach(() => {
-    jest.useRealTimers();
+    vi.useRealTimers();
   });
 
   async function* createMockStream(chunks: string[]): AsyncGenerator<string> {
@@ -28,7 +28,7 @@ describe("stopAfterMaxProcessingTime", () => {
 
   it("should yield all chunks when maxTimeMs is not reached", async () => {
     const mockStream = createMockStream(["Hello", " world", "!"]);
-    const fullStop = jest.fn();
+    const fullStop = vi.fn();
     const result = stopAfterMaxProcessingTime(mockStream, 1000, fullStop);
 
     const output = await streamToString(result);
@@ -37,11 +37,11 @@ describe("stopAfterMaxProcessingTime", () => {
     expect(fullStop).not.toHaveBeenCalled();
   });
 
-  it.only("should stop processing after max time is reached", async () => {
+  it("should stop processing after max time is reached", async () => {
     // Mock implementation of Date.now
     let currentTime = 0;
     const originalDateNow = Date.now;
-    Date.now = jest.fn(() => currentTime);
+    Date.now = vi.fn(() => currentTime);
 
     // Create a generator that we can control
     async function* controlledGenerator(): AsyncGenerator<string> {
@@ -54,7 +54,7 @@ describe("stopAfterMaxProcessingTime", () => {
       }
     }
 
-    const fullStop = jest.fn();
+    const fullStop = vi.fn();
     const maxTimeMs = 500;
 
     const transformedGenerator = stopAfterMaxProcessingTime(
@@ -84,10 +84,10 @@ describe("stopAfterMaxProcessingTime", () => {
   it("should check time only periodically based on checkInterval", async () => {
     const chunks = Array(100).fill("x");
     const mockStream = createMockStream(chunks);
-    const fullStop = jest.fn();
+    const fullStop = vi.fn();
 
     // Spy on Date.now to count how many times it's called
-    const dateSpy = jest.spyOn(Date, "now");
+    const dateSpy = vi.spyOn(Date, "now");
 
     // Stream should complete normally (not hitting the timeout)
     await streamToString(
@@ -105,7 +105,7 @@ describe("stopAfterMaxProcessingTime", () => {
 
   it("should handle empty stream gracefully", async () => {
     const mockStream = createMockStream([]);
-    const fullStop = jest.fn();
+    const fullStop = vi.fn();
     const result = stopAfterMaxProcessingTime(mockStream, 1000, fullStop);
 
     const output = await streamToString(result);
@@ -117,7 +117,7 @@ describe("stopAfterMaxProcessingTime", () => {
   it("should pass through all chunks if there's no timeout", async () => {
     const chunks = Array(100).fill("test chunk");
     const mockStream = createMockStream(chunks);
-    const fullStop = jest.fn();
+    const fullStop = vi.fn();
 
     // Use undefined as timeout to simulate no timeout
     const result = stopAfterMaxProcessingTime(

core/autocomplete/snippets/gitDiffCache.test.ts → core/autocomplete/snippets/gitDiffCache.vitest.ts

@@ -1,3 +1,4 @@
+import { beforeEach, expect, test, vi } from "vitest";
 import { GitDiffCache } from "./gitDiffCache";
 
 beforeEach(() => {
@@ -7,7 +8,7 @@ beforeEach(() => {
 
 test("GitDiffCache returns cached results within cache time", async () => {
   const mockDiff = ["file1.ts", "file2.ts"];
-  const getDiffFn = jest.fn().mockResolvedValue(mockDiff);
+  const getDiffFn = vi.fn().mockResolvedValue(mockDiff);
   const cache = GitDiffCache.getInstance(getDiffFn, 1); // 1 second cache
 
   const result1 = await cache.get();
@@ -20,7 +21,7 @@ test("GitDiffCache returns cached results within cache time", async () => {
 
 test("GitDiffCache refreshes cache after expiration", async () => {
   const mockDiff = ["file1.ts"];
-  const getDiffFn = jest.fn().mockResolvedValue(mockDiff);
+  const getDiffFn = vi.fn().mockResolvedValue(mockDiff);
   const cache = GitDiffCache.getInstance(getDiffFn, 0.1); // 100ms cache
 
   const result1 = await cache.get();
@@ -31,7 +32,7 @@ test("GitDiffCache refreshes cache after expiration", async () => {
 });
 
 test("GitDiffCache returns empty array on error", async () => {
-  const getDiffFn = jest.fn().mockRejectedValue(new Error("Git error"));
+  const getDiffFn = vi.fn().mockRejectedValue(new Error("Git error"));
   const cache = GitDiffCache.getInstance(getDiffFn);
 
   const result = await cache.get();
@@ -41,7 +42,7 @@ test("GitDiffCache returns empty array on error", async () => {
 test("GitDiffCache reuses pending request", async () => {
   const mockDiff = ["file1.ts"];
   let resolvePromise: (value: string[]) => void;
-  const getDiffFn = jest.fn().mockImplementation(() => {
+  const getDiffFn = vi.fn().mockImplementation(() => {
     return new Promise((resolve) => {
       resolvePromise = resolve;
     });
@@ -63,7 +64,7 @@ test("GitDiffCache reuses pending request", async () => {
 
 test("GitDiffCache invalidate clears cache", async () => {
   const mockDiff = ["file1.ts"];
-  const getDiffFn = jest.fn().mockResolvedValue(mockDiff);
+  const getDiffFn = vi.fn().mockResolvedValue(mockDiff);
   const cache = GitDiffCache.getInstance(getDiffFn);
 
   await cache.get();
@@ -74,8 +75,8 @@ test("GitDiffCache invalidate clears cache", async () => {
 });
 
 test("GitDiffCache maintains singleton instance", () => {
-  const getDiffFn1 = jest.fn();
-  const getDiffFn2 = jest.fn();
+  const getDiffFn1 = vi.fn();
+  const getDiffFn2 = vi.fn();
 
   const instance1 = GitDiffCache.getInstance(getDiffFn1);
   const instance2 = GitDiffCache.getInstance(getDiffFn2);

core/autocomplete/templating/__tests__/formatOpenedFilesContext.test.ts → core/autocomplete/templating/__tests__/formatOpenedFilesContext.vitest.ts

@@ -3,6 +3,7 @@
  *
  */
 
+import { describe, expect, test } from "vitest";
 import {
   AutocompleteCodeSnippet,
   AutocompleteDiffSnippet,

core/autocomplete/util/completionTestUtils.test.ts → core/autocomplete/util/completionTestUtils.vitest.ts

@@ -1,3 +1,4 @@
+import { describe, expect, it } from "vitest";
 import {
   processTestCase,
   type CompletionTestCase,

core/autocomplete/util/processSingleLineCompletion.test.ts → core/autocomplete/util/processSingleLineCompletion.vitest.ts

@@ -1,3 +1,4 @@
+import { describe, expect, it } from "vitest";
 import { processTestCase } from "./completionTestUtils";
 import { processSingleLineCompletion } from "./processSingleLineCompletion";
 

core/config/yaml/LocalPlatformClient.vitest.ts

@@ -123,7 +123,6 @@ describe("LocalPlatformClient", () => {
       expect(
         (resolvedFQSNs[0] as SecretResult & { value: unknown })?.value,
       ).toBe(secretValue);
-      console.log("debug1 resolved fqsn", resolvedFQSNs);
     });
   });