File size: 2,930 Bytes
bf48b89 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 | import { getCurrentCell, setCurrentCell } from 'node-network-devtools';
import undici from 'undici';
import { afterEach, beforeEach, describe, expect, test, vi } from 'vitest';
import { useCustomHeader } from './fetch';
const getInitRequest = () =>
({
requestHeaders: {} as Record<string, string>,
id: '',
loadCallFrames: () => {},
cookies: '',
requestData: '',
responseData: '',
responseHeaders: {},
responseInfo: {},
}) satisfies NonNullable<ReturnType<typeof getCurrentCell>>['request'];
enum Env {
dev = 'dev',
production = 'production',
test = 'test',
}
describe('useCustomHeader', () => {
let originalEnv: string;
beforeEach(() => {
originalEnv = process.env.NODE_ENV || Env.test;
process.env.ENABLE_REMOTE_DEBUGGING = 'true';
});
afterEach(() => {
process.env.NODE_ENV = originalEnv;
});
test('should register request with custom headers in dev environment', () => {
process.env.NODE_ENV = Env.dev;
const headers = new Headers();
const headerText = 'authorization';
const headerValue = 'Bearer token';
headers.set(headerText, headerValue);
const req = getInitRequest();
setCurrentCell({
request: req,
pipes: [],
isAborted: false,
});
useCustomHeader(headers);
const cell = getCurrentCell();
expect(cell).toBeDefined();
let request = req;
if (cell) {
for (const { pipe } of cell.pipes) {
request = pipe(request);
}
}
expect(request.requestHeaders[headerText]).toEqual(headerValue);
});
test('should not register request in non-dev environment', () => {
process.env.NODE_ENV = Env.production;
const headers = new Headers();
const headerText = 'content-type';
const headerValue = 'application/json';
headers.set(headerText, headerValue);
const req = getInitRequest();
setCurrentCell({
request: req,
pipes: [],
isAborted: false,
});
useCustomHeader(headers);
const cell = getCurrentCell();
expect(cell).toBeDefined();
let request = req;
if (cell) {
for (const { pipe } of cell.pipes) {
request = pipe(request);
}
}
expect(req.requestHeaders[headerText]).toBeUndefined();
});
});
describe('wrappedFetch', () => {
test('throws when fetch fails without proxy retry', async () => {
const fetchSpy = vi.spyOn(undici, 'fetch').mockRejectedValueOnce(new Error('boom'));
const { default: wrappedFetch } = await import('./fetch');
await expect(wrappedFetch('http://example.com')).rejects.toThrow('boom');
fetchSpy.mockRestore();
});
});
|