fix(cli): preserve Request headers in DevTools activity logger (#26078)

This commit is contained in:
Adib234
2026-04-28 13:34:45 -04:00
committed by GitHub
parent c841070582
commit 59b2dea0e5
2 changed files with 113 additions and 9 deletions
+92 -1
View File
@@ -4,7 +4,7 @@
* SPDX-License-Identifier: Apache-2.0
*/
import { describe, it, expect, beforeEach } from 'vitest';
import { describe, it, expect, beforeEach, vi } from 'vitest';
import { ActivityLogger, type NetworkLog } from './activityLogger.js';
import type { ConsoleLogPayload } from '@google/gemini-cli-core';
@@ -132,4 +132,95 @@ describe('ActivityLogger', () => {
expect(after.console.length).toBe(0);
expect(after.network.length).toBe(0);
});
it('preserves headers and method from Request object when intercepting fetch', async () => {
const originalFetch = global.fetch;
const mockFetch = vi.fn().mockImplementation(() =>
Promise.resolve({
status: 200,
headers: new Headers(),
body: null,
clone: () => ({
body: null,
status: 200,
headers: new Headers(),
text: async () => 'ok',
json: async () => ({}),
}),
} as unknown as Response),
);
global.fetch = mockFetch;
try {
// @ts-expect-error - accessing private property for testing
logger.isInterceptionEnabled = false;
logger.enable();
const request = new Request('https://api.example.com/data', {
headers: { Authorization: 'Bearer test-token' },
method: 'POST',
});
await global.fetch(request);
expect(mockFetch).toHaveBeenCalled();
const [, calledInit] = mockFetch.mock.calls[0];
expect(calledInit?.headers).toBeDefined();
const headers = new Headers(calledInit?.headers as HeadersInit);
expect(headers.get('Authorization')).toBe('Bearer test-token');
expect(headers.has('x-activity-request-id')).toBe(true);
expect(calledInit?.method).toBe('POST');
} finally {
global.fetch = originalFetch;
// @ts-expect-error - reset private property
logger.isInterceptionEnabled = false;
}
});
it('replaces Request headers with init headers (Fetch spec compliance)', async () => {
const originalFetch = global.fetch;
const mockFetch = vi.fn().mockImplementation(() =>
Promise.resolve({
status: 200,
headers: new Headers(),
body: null,
clone: () => ({
body: null,
status: 200,
headers: new Headers(),
text: async () => 'ok',
}),
} as unknown as Response),
);
global.fetch = mockFetch;
try {
// @ts-expect-error - accessing private property for testing
logger.isInterceptionEnabled = false;
logger.enable();
const request = new Request('https://api.example.com/data', {
headers: { 'X-Old': 'old-value', 'X-Shared': 'old-shared' },
});
await global.fetch(request, {
headers: { 'X-New': 'new-value', 'X-Shared': 'new-shared' },
});
const [, calledInit] = mockFetch.mock.calls[0];
const headers = new Headers(calledInit?.headers as HeadersInit);
expect(headers.get('X-New')).toBe('new-value');
expect(headers.get('X-Shared')).toBe('new-shared');
expect(headers.has('X-Old')).toBe(false);
expect(headers.has('x-activity-request-id')).toBe(true);
} finally {
global.fetch = originalFetch;
// @ts-expect-error - reset private property
logger.isInterceptionEnabled = false;
}
});
});