@@ -4,8 +4,9 @@ import Anthropic from "@anthropic-ai/sdk"
44import * as vscode from "vscode"
55import axios from "axios"
66
7- import { type ProviderSettingsEntry , type ClineMessage , ORGANIZATION_ALLOW_ALL } from "@roo-code/types"
7+ import { type ProviderSettingsEntry , type ClineMessage } from "@roo-code/types"
88import { TelemetryService } from "@roo-code/telemetry"
9+ import { ORGANIZATION_ALLOW_ALL } from "@roo-code/cloud"
910
1011import { ExtensionMessage , ExtensionState } from "../../../shared/ExtensionMessage"
1112import { defaultModeSlug } from "../../../shared/modes"
@@ -17,11 +18,9 @@ import { safeWriteJson } from "../../../utils/safeWriteJson"
1718
1819import { ClineProvider } from "../ClineProvider"
1920
20- // Mock setup must come before imports
21+ // Mock setup must come before imports.
2122vi . mock ( "../../prompts/sections/custom-instructions" )
2223
23- vi . mock ( "vscode" )
24-
2524vi . mock ( "p-wait-for" , ( ) => ( {
2625 __esModule : true ,
2726 default : vi . fn ( ) . mockResolvedValue ( undefined ) ,
@@ -95,7 +94,7 @@ vi.mock("../../../services/browser/browserDiscovery", () => ({
9594 testBrowserConnection : vi . fn ( ) ,
9695} ) )
9796
98- // Remove duplicate mock - it's already defined below
97+ // Remove duplicate mock - it's already defined below.
9998
10099const mockAddCustomInstructions = vi . fn ( ) . mockResolvedValue ( "Combined instructions" )
101100
@@ -228,7 +227,6 @@ vi.mock("../../../integrations/misc/extract-text", () => ({
228227 } ) ,
229228} ) )
230229
231- // Mock getModels for router model tests
232230vi . mock ( "../../../api/providers/fetchers/modelCache" , ( ) => ( {
233231 getModels : vi . fn ( ) . mockResolvedValue ( { } ) ,
234232 flushModels : vi . fn ( ) ,
@@ -324,6 +322,10 @@ vi.mock("@roo-code/cloud", () => ({
324322 } ,
325323 } ,
326324 getRooCodeApiUrl : vi . fn ( ) . mockReturnValue ( "https://app.roocode.com" ) ,
325+ ORGANIZATION_ALLOW_ALL : {
326+ allowAll : true ,
327+ providers : { } ,
328+ } ,
327329} ) )
328330
329331afterAll ( ( ) => {
0 commit comments