Skip to content

Commit f0a2452

Browse files
committed
fix: add comprehensive VSCode and telemetry mocking for Gemini grounding tests
1 parent df215b1 commit f0a2452

File tree

1 file changed

+71
-0
lines changed

1 file changed

+71
-0
lines changed

src/core/task/__tests__/Task.gemini-grounding.spec.ts

Lines changed: 71 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,77 @@ import { Task } from "../Task"
55
import type { ClineProvider } from "../../webview/ClineProvider"
66
import type { ProviderSettings } from "@roo-code/types"
77

8+
// Mock TelemetryService
9+
vi.mock("@roo-code/telemetry", () => ({
10+
TelemetryService: {
11+
instance: {
12+
logEvent: vi.fn(),
13+
logError: vi.fn(),
14+
logPerformance: vi.fn(),
15+
captureTaskCreated: vi.fn(),
16+
captureTaskCompleted: vi.fn(),
17+
captureTaskAborted: vi.fn(),
18+
},
19+
},
20+
BaseTelemetryClient: vi.fn(() => ({
21+
logEvent: vi.fn(),
22+
logError: vi.fn(),
23+
logPerformance: vi.fn(),
24+
})),
25+
}))
26+
27+
// Mock VSCode API
28+
vi.mock("vscode", () => ({
29+
Uri: {
30+
file: vi.fn((path: string) => ({ fsPath: path, path })),
31+
parse: vi.fn((uri: string) => ({ fsPath: uri, path: uri })),
32+
},
33+
workspace: {
34+
getConfiguration: vi.fn(() => ({
35+
get: vi.fn(),
36+
update: vi.fn(),
37+
})),
38+
workspaceFolders: [],
39+
createFileSystemWatcher: vi.fn(() => ({
40+
onDidCreate: vi.fn(),
41+
onDidChange: vi.fn(),
42+
onDidDelete: vi.fn(),
43+
dispose: vi.fn(),
44+
})),
45+
},
46+
window: {
47+
showErrorMessage: vi.fn(),
48+
showWarningMessage: vi.fn(),
49+
showInformationMessage: vi.fn(),
50+
createTextEditorDecorationType: vi.fn(() => ({
51+
dispose: vi.fn(),
52+
})),
53+
},
54+
commands: {
55+
registerCommand: vi.fn(),
56+
},
57+
EventEmitter: vi.fn(() => ({
58+
event: vi.fn(),
59+
fire: vi.fn(),
60+
dispose: vi.fn(),
61+
})),
62+
RelativePattern: vi.fn((base: any, pattern: string) => ({
63+
base,
64+
pattern,
65+
})),
66+
Range: vi.fn(),
67+
Position: vi.fn(),
68+
Selection: vi.fn(),
69+
TextEdit: vi.fn(),
70+
WorkspaceEdit: vi.fn(),
71+
OverviewRulerLane: {
72+
Left: 1,
73+
Center: 2,
74+
Right: 4,
75+
Full: 7,
76+
},
77+
}))
78+
879
describe("Task Gemini Grounding Loop Prevention", () => {
980
let mockProvider: Partial<ClineProvider>
1081
let mockApiConfiguration: ProviderSettings

0 commit comments

Comments
 (0)