Disable all network connection in jest

Author: cte

scripts/run-tests.js

@@ -2,7 +2,7 @@
 const { execSync } = require("child_process")
 
 if (process.platform === "win32") {
-	execSync("npm-run-all test:* lint:*", { stdio: "inherit" })
+	execSync("npm-run-all test:*", { stdio: "inherit" })
 } else {
-	execSync("npm-run-all -p test:* lint:*", { stdio: "inherit" })
+	execSync("npm-run-all -p test:*", { stdio: "inherit" })
 }

src/__mocks__/jest.setup.ts

@@ -1,3 +1,15 @@
+import nock from "nock"
+
+nock.disableNetConnect()
+
+export function allowNetConnect(host?: string | RegExp) {
+	if (host) {
+		nock.enableNetConnect(host)
+	} else {
+		nock.enableNetConnect()
+	}
+}
+
 // Mock the logger globally for all tests
 jest.mock("../utils/logging", () => ({
 	logger: {

src/api/providers/__tests__/glama.test.ts

@@ -5,6 +5,36 @@ import { Anthropic } from "@anthropic-ai/sdk"
 import { GlamaHandler } from "../glama"
 import { ApiHandlerOptions } from "../../../shared/api"
 
+// Mock dependencies
+jest.mock("../fetchers/cache", () => ({
+	getModels: jest.fn().mockImplementation(() => {
+		return Promise.resolve({
+			"anthropic/claude-3-7-sonnet": {
+				maxTokens: 8192,
+				contextWindow: 200000,
+				supportsImages: true,
+				supportsPromptCache: true,
+				inputPrice: 3,
+				outputPrice: 15,
+				cacheWritesPrice: 3.75,
+				cacheReadsPrice: 0.3,
+				description: "Claude 3.7 Sonnet",
+				thinking: false,
+				supportsComputerUse: true,
+			},
+			"openai/gpt-4o": {
+				maxTokens: 4096,
+				contextWindow: 128000,
+				supportsImages: true,
+				supportsPromptCache: false,
+				inputPrice: 5,
+				outputPrice: 15,
+				description: "GPT-4o",
+			},
+		})
+	}),
+}))
+
 // Mock OpenAI client
 const mockCreate = jest.fn()
 const mockWithResponse = jest.fn()

src/api/providers/__tests__/openrouter.test.ts

@@ -9,6 +9,38 @@ import { ApiHandlerOptions } from "../../../shared/api"
 // Mock dependencies
 jest.mock("openai")
 jest.mock("delay", () => jest.fn(() => Promise.resolve()))
+jest.mock("../fetchers/cache", () => ({
+	getModels: jest.fn().mockImplementation(() => {
+		return Promise.resolve({
+			"anthropic/claude-3.7-sonnet": {
+				maxTokens: 8192,
+				contextWindow: 200000,
+				supportsImages: true,
+				supportsPromptCache: true,
+				inputPrice: 3,
+				outputPrice: 15,
+				cacheWritesPrice: 3.75,
+				cacheReadsPrice: 0.3,
+				description: "Claude 3.7 Sonnet",
+				thinking: false,
+				supportsComputerUse: true,
+			},
+			"anthropic/claude-3.7-sonnet:thinking": {
+				maxTokens: 128000,
+				contextWindow: 200000,
+				supportsImages: true,
+				supportsPromptCache: true,
+				inputPrice: 3,
+				outputPrice: 15,
+				cacheWritesPrice: 3.75,
+				cacheReadsPrice: 0.3,
+				description: "Claude 3.7 Sonnet with thinking",
+				thinking: true,
+				supportsComputerUse: true,
+			},
+		})
+	}),
+}))
 
 describe("OpenRouterHandler", () => {
 	const mockOptions: ApiHandlerOptions = {

src/api/providers/__tests__/unbound.test.ts

@@ -6,6 +6,58 @@ import { ApiHandlerOptions } from "../../../shared/api"
 
 import { UnboundHandler } from "../unbound"
 
+// Mock dependencies
+jest.mock("../fetchers/cache", () => ({
+	getModels: jest.fn().mockImplementation(() => {
+		return Promise.resolve({
+			"anthropic/claude-3-5-sonnet-20241022": {
+				maxTokens: 8192,
+				contextWindow: 200000,
+				supportsImages: true,
+				supportsPromptCache: true,
+				inputPrice: 3,
+				outputPrice: 15,
+				cacheWritesPrice: 3.75,
+				cacheReadsPrice: 0.3,
+				description: "Claude 3.5 Sonnet",
+				thinking: false,
+				supportsComputerUse: true,
+			},
+			"anthropic/claude-3-7-sonnet-20250219": {
+				maxTokens: 8192,
+				contextWindow: 200000,
+				supportsImages: true,
+				supportsPromptCache: true,
+				inputPrice: 3,
+				outputPrice: 15,
+				cacheWritesPrice: 3.75,
+				cacheReadsPrice: 0.3,
+				description: "Claude 3.7 Sonnet",
+				thinking: false,
+				supportsComputerUse: true,
+			},
+			"openai/gpt-4o": {
+				maxTokens: 4096,
+				contextWindow: 128000,
+				supportsImages: true,
+				supportsPromptCache: false,
+				inputPrice: 5,
+				outputPrice: 15,
+				description: "GPT-4o",
+			},
+			"openai/o3-mini": {
+				maxTokens: 4096,
+				contextWindow: 128000,
+				supportsImages: true,
+				supportsPromptCache: false,
+				inputPrice: 1,
+				outputPrice: 3,
+				description: "O3 Mini",
+			},
+		})
+	}),
+}))
+
 // Mock OpenAI client
 const mockCreate = jest.fn()
 const mockWithResponse = jest.fn()

src/api/providers/glama.ts

@@ -19,7 +19,7 @@ export class GlamaHandler extends RouterProvider implements SingleCompletionHand
 	constructor(options: ApiHandlerOptions) {
 		super({
 			options,
-			name: "unbound",
+			name: "glama",
 			baseURL: "https://glama.ai/api/gateway/openai/v1",
 			apiKey: options.glamaApiKey,
 			modelId: options.glamaModelId,