diff options
Diffstat (limited to 'packages/core/src/code_assist/oauth2.test.ts')
| -rw-r--r-- | packages/core/src/code_assist/oauth2.test.ts | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/packages/core/src/code_assist/oauth2.test.ts b/packages/core/src/code_assist/oauth2.test.ts index 821788d5..25718cb1 100644 --- a/packages/core/src/code_assist/oauth2.test.ts +++ b/packages/core/src/code_assist/oauth2.test.ts @@ -324,6 +324,70 @@ describe('oauth2', () => { }); }); + describe('credential loading order', () => { + it('should prioritize default cached credentials over GOOGLE_APPLICATION_CREDENTIALS', async () => { + // Setup default cached credentials + const defaultCreds = { refresh_token: 'default-cached-token' }; + const defaultCredsPath = path.join( + tempHomeDir, + '.gemini', + 'oauth_creds.json', + ); + await fs.promises.mkdir(path.dirname(defaultCredsPath), { + recursive: true, + }); + await fs.promises.writeFile( + defaultCredsPath, + JSON.stringify(defaultCreds), + ); + + // Setup credentials via environment variable + const envCreds = { refresh_token: 'env-var-token' }; + const envCredsPath = path.join(tempHomeDir, 'env_creds.json'); + await fs.promises.writeFile(envCredsPath, JSON.stringify(envCreds)); + vi.stubEnv('GOOGLE_APPLICATION_CREDENTIALS', envCredsPath); + + const mockClient = { + setCredentials: vi.fn(), + getAccessToken: vi.fn().mockResolvedValue({ token: 'test-token' }), + getTokenInfo: vi.fn().mockResolvedValue({}), + on: vi.fn(), + }; + (OAuth2Client as unknown as Mock).mockImplementation( + () => mockClient as unknown as OAuth2Client, + ); + + await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig); + + // Assert the correct credentials were used + expect(mockClient.setCredentials).toHaveBeenCalledWith(defaultCreds); + expect(mockClient.setCredentials).not.toHaveBeenCalledWith(envCreds); + }); + + it('should fall back to GOOGLE_APPLICATION_CREDENTIALS if default cache is missing', async () => { + // Setup credentials via environment variable + const envCreds = { refresh_token: 'env-var-token' }; + const envCredsPath = path.join(tempHomeDir, 'env_creds.json'); + await fs.promises.writeFile(envCredsPath, JSON.stringify(envCreds)); + vi.stubEnv('GOOGLE_APPLICATION_CREDENTIALS', envCredsPath); + + const mockClient = { + setCredentials: vi.fn(), + getAccessToken: vi.fn().mockResolvedValue({ token: 'test-token' }), + getTokenInfo: vi.fn().mockResolvedValue({}), + on: vi.fn(), + }; + (OAuth2Client as unknown as Mock).mockImplementation( + () => mockClient as unknown as OAuth2Client, + ); + + await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig); + + // Assert the correct credentials were used + expect(mockClient.setCredentials).toHaveBeenCalledWith(envCreds); + }); + }); + describe('with GCP environment variables', () => { it('should use GOOGLE_CLOUD_ACCESS_TOKEN when GOOGLE_GENAI_USE_GCA is true', async () => { vi.stubEnv('GOOGLE_GENAI_USE_GCA', 'true'); |
