scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-08-28 05:27:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #1397 from RooVetGit/fix_context_proxy_tests

Browse source 
fix test
This commit is contained in:
Matt Rubens 2025-03-05 12:22:57 -05:00 • committed by GitHub
commit f3b3058615
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 22 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
src/core/webview/__tests__/ClineProvider.test.ts
View file

@ -37,6 +37,16 @@ jest.mock("../../contextProxy", () => {
saveChanges: jest.fn().mockResolvedValue(undefined),
dispose: jest.fn().mockResolvedValue(undefined),
hasPendingChanges: jest.fn().mockReturnValue(false),
setValue: jest.fn().mockImplementation((key, value) => {
if (key.startsWith("apiKey") || key.startsWith("openAiApiKey")) {
return context.secrets.store(key, value)
}
return context.globalState.update(key, value)
}),
setValues: jest.fn().mockImplementation((values) => {
const promises = Object.entries(values).map(([key, value]) => context.globalState.update(key, value))
return Promise.all(promises)
}),
})),
}
})
@ -267,6 +277,8 @@ describe("ClineProvider", () => {
let mockContextProxy: {
updateGlobalState: jest.Mock
getGlobalState: jest.Mock
setValue: jest.Mock
setValues: jest.Mock
storeSecret: jest.Mock
dispose: jest.Mock
}
@ -1536,6 +1548,7 @@ describe("ContextProxy integration", () => {
let mockContext: vscode.ExtensionContext
let mockOutputChannel: vscode.OutputChannel
let mockContextProxy: any
let mockGlobalStateUpdate: jest.Mock
beforeEach(() => {
// Reset mocks
@ -1543,7 +1556,11 @@ describe("ContextProxy integration", () => {
// Setup basic mocks
mockContext = {
globalState: { get: jest.fn(), update: jest.fn(), keys: jest.fn().mockReturnValue([]) },
globalState: {
get: jest.fn(),
update: jest.fn(),
keys: jest.fn().mockReturnValue([]),
},
secrets: { get: jest.fn(), store: jest.fn(), delete: jest.fn() },
extensionUri: {} as vscode.Uri,
globalStorageUri: { fsPath: "/test/path" },
@ -1555,6 +1572,8 @@ describe("ContextProxy integration", () => {
// @ts-ignore - accessing private property for testing
mockContextProxy = provider.contextProxy
mockGlobalStateUpdate = mockContext.globalState.update as jest.Mock
})
test("updateGlobalState uses contextProxy", async () => {
@ -1579,5 +1598,7 @@ describe("ContextProxy integration", () => {
expect(mockContextProxy.getGlobalState).toBeDefined()
expect(mockContextProxy.updateGlobalState).toBeDefined()
expect(mockContextProxy.storeSecret).toBeDefined()
expect(mockContextProxy.setValue).toBeDefined()
expect(mockContextProxy.setValues).toBeDefined()
})
})