Skip to content

Commit cd4d52c

Browse files
MarkDaoustcopybara-github
authored andcommitted
chore: Fix nondeterministic baseUrl errors
PiperOrigin-RevId: 915119347
1 parent 1cc001c commit cd4d52c

3 files changed

Lines changed: 30 additions & 0 deletions

File tree

test/unit/node/base_url_test.ts

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,11 +8,28 @@ import {setDefaultBaseUrls} from '../../../src/_base_url.js';
88
import {GoogleGenAI} from '../../../src/node/node_client.js';
99

1010
describe('setDefaultBaseUrls', () => {
11+
beforeEach(() => {
12+
delete process.env['GOOGLE_API_KEY'];
13+
delete process.env['GEMINI_API_KEY'];
14+
delete process.env['GOOGLE_GEMINI_BASE_URL'];
15+
delete process.env['GOOGLE_VERTEX_BASE_URL'];
16+
delete process.env['GOOGLE_GENAI_USE_ENTERPRISE'];
17+
delete process.env['GOOGLE_GENAI_USE_VERTEXAI'];
18+
delete process.env['GOOGLE_CLOUD_PROJECT'];
19+
delete process.env['GOOGLE_CLOUD_LOCATION'];
20+
21+
setDefaultBaseUrls({});
22+
});
23+
1124
afterEach(() => {
25+
delete process.env['GOOGLE_API_KEY'];
26+
delete process.env['GEMINI_API_KEY'];
1227
delete process.env['GOOGLE_GEMINI_BASE_URL'];
1328
delete process.env['GOOGLE_VERTEX_BASE_URL'];
1429
delete process.env['GOOGLE_GENAI_USE_ENTERPRISE'];
1530
delete process.env['GOOGLE_GENAI_USE_VERTEXAI'];
31+
delete process.env['GOOGLE_CLOUD_PROJECT'];
32+
delete process.env['GOOGLE_CLOUD_LOCATION'];
1633

1734
setDefaultBaseUrls({});
1835
});

test/unit/node/client_test.ts

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,15 @@ describe('Client', () => {
2525
setDefaultBaseUrls({});
2626
});
2727
afterEach(() => {
28+
delete process.env['GOOGLE_API_KEY'];
29+
delete process.env['GEMINI_API_KEY'];
30+
delete process.env['GOOGLE_GENAI_USE_ENTERPRISE'];
31+
delete process.env['GOOGLE_GENAI_USE_VERTEXAI'];
32+
delete process.env['GOOGLE_CLOUD_PROJECT'];
33+
delete process.env['GOOGLE_CLOUD_LOCATION'];
34+
delete process.env['GOOGLE_GEMINI_BASE_URL'];
35+
delete process.env['GOOGLE_VERTEX_BASE_URL'];
36+
2837
setDefaultBaseUrls({});
2938
});
3039

test/unit/web/base_url_test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,10 @@ import {setDefaultBaseUrls} from '../../../src/_base_url.js';
88
import {GoogleGenAI} from '../../../src/web/web_client.js';
99

1010
describe('setDefaultBaseUrls', () => {
11+
beforeEach(() => {
12+
setDefaultBaseUrls({});
13+
});
14+
1115
afterEach(() => {
1216
setDefaultBaseUrls({});
1317
});

0 commit comments

Comments
 (0)