fix: tests

Author: sestinj

extensions/cli/src/integration/model-persistence-e2e.test.ts

@@ -12,9 +12,9 @@ import {
   saveAuthConfig,
   updateModelName,
 } from "../auth/workos.js";
-import { persistModelName } from "../util/modelPersistence.js";
 import * as config from "../config.js";
 import { ModelService } from "../services/ModelService.js";
+import { persistModelName } from "../util/modelPersistence.js";
 
 // Mock the config module
 vi.mock("../config.js");

extensions/cli/src/integration/model-persistence-unauthenticated.test.ts

@@ -208,14 +208,21 @@ describe("Model Persistence for Unauthenticated Users", () => {
     updateModelName(null);
     expect(getPersistedModelName()).toBeNull();
 
+    // Capture original value to restore after test
+    const originalApiKey = process.env.CONTINUE_API_KEY;
     process.env.CONTINUE_API_KEY = "test-api-key";
 
     updateModelName("Claude 3.5 Sonnet");
 
     // Should not be saved to GlobalContext
     expect(getPersistedModelName()).toBeNull();
 
-    delete process.env.CONTINUE_API_KEY;
+    // Restore original value
+    if (originalApiKey !== undefined) {
+      process.env.CONTINUE_API_KEY = originalApiKey;
+    } else {
+      delete process.env.CONTINUE_API_KEY;
+    }
   });
 
   test("should work across config changes for logged-out users", async () => {

extensions/cli/src/integration/model-persistence-user-flow.test.ts

@@ -12,9 +12,8 @@ import {
   saveAuthConfig,
 } from "../auth/workos.js";
 import * as config from "../config.js";
-import { ModelService } from "../services/ModelService.js";
 import { AuthService } from "../services/AuthService.js";
-import { serviceContainer, SERVICE_NAMES } from "../services/index.js";
+import { ModelService } from "../services/ModelService.js";
 
 // Mock the config module
 vi.mock("../config.js");
@@ -101,7 +100,7 @@ describe("Model Persistence User Flow", () => {
 
     // Initialize ModelService (what happens on CLI start)
     const modelService = new ModelService();
-    let state = await modelService.initialize(mockAssistant, mockAuthConfig);
+    const state = await modelService.initialize(mockAssistant, mockAuthConfig);
     console.log("2. Initial model:", state.model?.name);
     expect(state.model?.name).toBe("GPT-4");
 

extensions/cli/src/integration/model-persistence.test.ts

@@ -143,6 +143,8 @@ describe("Model Persistence Integration", () => {
     // Clear GlobalContext first
     persistModelName(null);
 
+    // Capture original value to restore after test
+    const originalApiKey = process.env.CONTINUE_API_KEY;
     // Set environment variable
     process.env.CONTINUE_API_KEY = "test-api-key";
 
@@ -153,8 +155,12 @@ describe("Model Persistence Integration", () => {
     const loadedConfig = loadAuthConfig();
     expect(getModelName(loadedConfig)).toBeNull();
 
-    // Cleanup
-    delete process.env.CONTINUE_API_KEY;
+    // Restore original value
+    if (originalApiKey !== undefined) {
+      process.env.CONTINUE_API_KEY = originalApiKey;
+    } else {
+      delete process.env.CONTINUE_API_KEY;
+    }
   });
 
   test("should persist model name in auth.json file with correct format", () => {