diff options
Diffstat (limited to 'packages/core/src/code_assist/oauth2.test.ts')
| -rw-r--r-- | packages/core/src/code_assist/oauth2.test.ts | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/packages/core/src/code_assist/oauth2.test.ts b/packages/core/src/code_assist/oauth2.test.ts index 77a8fb90..821788d5 100644 --- a/packages/core/src/code_assist/oauth2.test.ts +++ b/packages/core/src/code_assist/oauth2.test.ts @@ -57,9 +57,7 @@ describe('oauth2', () => { fs.rmSync(tempHomeDir, { recursive: true, force: true }); vi.clearAllMocks(); resetOauthClientForTesting(); - delete process.env.CLOUD_SHELL; - delete process.env.GOOGLE_GENAI_USE_GCA; - delete process.env.GOOGLE_CLOUD_ACCESS_TOKEN; + vi.unstubAllEnvs(); }); it('should perform a web login', async () => { @@ -328,8 +326,8 @@ describe('oauth2', () => { describe('with GCP environment variables', () => { it('should use GOOGLE_CLOUD_ACCESS_TOKEN when GOOGLE_GENAI_USE_GCA is true', async () => { - process.env.GOOGLE_GENAI_USE_GCA = 'true'; - process.env.GOOGLE_CLOUD_ACCESS_TOKEN = 'gcp-access-token'; + vi.stubEnv('GOOGLE_GENAI_USE_GCA', 'true'); + vi.stubEnv('GOOGLE_CLOUD_ACCESS_TOKEN', 'gcp-access-token'); const mockSetCredentials = vi.fn(); const mockGetAccessToken = vi @@ -387,7 +385,7 @@ describe('oauth2', () => { }); it('should not use GCP token if GOOGLE_CLOUD_ACCESS_TOKEN is not set', async () => { - process.env.GOOGLE_GENAI_USE_GCA = 'true'; + vi.stubEnv('GOOGLE_GENAI_USE_GCA', 'true'); const mockSetCredentials = vi.fn(); const mockGetAccessToken = vi @@ -418,7 +416,7 @@ describe('oauth2', () => { }); it('should not use GCP token if GOOGLE_GENAI_USE_GCA is not set', async () => { - process.env.GOOGLE_CLOUD_ACCESS_TOKEN = 'gcp-access-token'; + vi.stubEnv('GOOGLE_CLOUD_ACCESS_TOKEN', 'gcp-access-token'); const mockSetCredentials = vi.fn(); const mockGetAccessToken = vi |