Skip to content

Commit 60eaf89

Browse files
committed
test: import original vscode module
1 parent 6b2c49f commit 60eaf89

File tree

8 files changed

+11
-7
lines changed

8 files changed

+11
-7
lines changed

src/api/providers/__tests__/groq.spec.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
// npx vitest run src/api/providers/__tests__/groq.spec.ts
22

33
// Mock vscode first to avoid import errors
4-
vitest.mock("vscode", () => ({}))
4+
vitest.mock("vscode", (importOriginal) => importOriginal())
55

66
import OpenAI from "openai"
77
import { Anthropic } from "@anthropic-ai/sdk"

src/api/providers/__tests__/lite-llm.spec.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import { ApiHandlerOptions } from "../../../shared/api"
66
import { litellmDefaultModelId, litellmDefaultModelInfo } from "@roo-code/types"
77

88
// Mock vscode first to avoid import errors
9-
vi.mock("vscode", () => ({}))
9+
vitest.mock("vscode", (importOriginal) => importOriginal())
1010

1111
// Mock OpenAI
1212
vi.mock("openai", () => {

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
// npx vitest run src/api/providers/__tests__/openrouter.spec.ts
22

33
// Mock vscode first to avoid import errors
4-
vitest.mock("vscode", () => ({}))
4+
vitest.mock("vscode", (importOriginal) => importOriginal())
55

66
import { Anthropic } from "@anthropic-ai/sdk"
77
import OpenAI from "openai"
@@ -65,6 +65,7 @@ describe("OpenRouterHandler", () => {
6565
"X-Title": "Roo Code",
6666
"User-Agent": `RooCode/${Package.version}`,
6767
},
68+
timeout: 600000,
6869
})
6970
})
7071

src/api/providers/__tests__/requesty.spec.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ describe("RequestyHandler", () => {
6262
"X-Title": "Roo Code",
6363
"User-Agent": `RooCode/${Package.version}`,
6464
},
65+
timeout: 600000,
6566
})
6667
})
6768

@@ -77,6 +78,7 @@ describe("RequestyHandler", () => {
7778
"X-Title": "Roo Code",
7879
"User-Agent": `RooCode/${Package.version}`,
7980
},
81+
timeout: 600000,
8082
})
8183
})
8284

src/api/providers/__tests__/sambanova.spec.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
// npx vitest run src/api/providers/__tests__/sambanova.spec.ts
22

33
// Mock vscode first to avoid import errors
4-
vitest.mock("vscode", () => ({}))
4+
vitest.mock("vscode", (importOriginal) => importOriginal())
55

66
import OpenAI from "openai"
77
import { Anthropic } from "@anthropic-ai/sdk"

src/api/providers/__tests__/vercel-ai-gateway.spec.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
// npx vitest run src/api/providers/__tests__/vercel-ai-gateway.spec.ts
22

33
// Mock vscode first to avoid import errors
4-
vitest.mock("vscode", () => ({}))
4+
vitest.mock("vscode", (importOriginal) => importOriginal())
55

66
import { Anthropic } from "@anthropic-ai/sdk"
77
import OpenAI from "openai"
@@ -102,6 +102,7 @@ describe("VercelAiGatewayHandler", () => {
102102
"X-Title": "Roo Code",
103103
"User-Agent": expect.stringContaining("RooCode/"),
104104
}),
105+
timeout: 600000,
105106
})
106107
})
107108

src/api/providers/__tests__/vertex.spec.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
// npx vitest run src/api/providers/__tests__/vertex.spec.ts
22

33
// Mock vscode first to avoid import errors
4-
vitest.mock("vscode", () => ({}))
4+
vitest.mock("vscode", (importOriginal) => importOriginal())
55

66
import { Anthropic } from "@anthropic-ai/sdk"
77

src/api/providers/__tests__/zai.spec.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
// npx vitest run src/api/providers/__tests__/zai.spec.ts
22

33
// Mock vscode first to avoid import errors
4-
vitest.mock("vscode", () => ({}))
4+
vitest.mock("vscode", (importOriginal) => importOriginal())
55

66
import OpenAI from "openai"
77
import { Anthropic } from "@anthropic-ai/sdk"

0 commit comments

Comments
 (0)