Skip to content

Commit dbdc8e4

Browse files
committed
Fix tests
1 parent eeabc50 commit dbdc8e4

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

src/core/webview/__tests__/ClineProvider.spec.ts

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2684,6 +2684,13 @@ describe("ClineProvider - Router Models", () => {
26842684
expect(getModels).toHaveBeenCalledWith({ provider: "glama" })
26852685
expect(getModels).toHaveBeenCalledWith({ provider: "unbound", apiKey: "unbound-key" })
26862686
expect(getModels).toHaveBeenCalledWith({ provider: "vercel-ai-gateway" })
2687+
expect(getModels).toHaveBeenCalledWith({ provider: "deepinfra" })
2688+
expect(getModels).toHaveBeenCalledWith(
2689+
expect.objectContaining({
2690+
provider: "roo",
2691+
baseUrl: expect.any(String),
2692+
}),
2693+
)
26872694
expect(getModels).toHaveBeenCalledWith({
26882695
provider: "litellm",
26892696
apiKey: "litellm-key",
@@ -2699,6 +2706,7 @@ describe("ClineProvider - Router Models", () => {
26992706
requesty: mockModels,
27002707
glama: mockModels,
27012708
unbound: mockModels,
2709+
roo: mockModels,
27022710
litellm: mockModels,
27032711
ollama: {},
27042712
lmstudio: {},
@@ -2737,6 +2745,7 @@ describe("ClineProvider - Router Models", () => {
27372745
.mockRejectedValueOnce(new Error("Unbound API error")) // unbound fail
27382746
.mockResolvedValueOnce(mockModels) // vercel-ai-gateway success
27392747
.mockResolvedValueOnce(mockModels) // deepinfra success
2748+
.mockResolvedValueOnce(mockModels) // roo success
27402749
.mockRejectedValueOnce(new Error("LiteLLM connection failed")) // litellm fail
27412750

27422751
await messageHandler({ type: "requestRouterModels" })
@@ -2750,6 +2759,7 @@ describe("ClineProvider - Router Models", () => {
27502759
requesty: {},
27512760
glama: mockModels,
27522761
unbound: {},
2762+
roo: mockModels,
27532763
ollama: {},
27542764
lmstudio: {},
27552765
litellm: {},
@@ -2864,6 +2874,7 @@ describe("ClineProvider - Router Models", () => {
28642874
requesty: mockModels,
28652875
glama: mockModels,
28662876
unbound: mockModels,
2877+
roo: mockModels,
28672878
litellm: {},
28682879
ollama: {},
28692880
lmstudio: {},

0 commit comments

Comments
 (0)