116 lines
4.4 KiB
TypeScript
116 lines
4.4 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from 'vitest';
|
|
|
|
import { generateHeaders, PRESETS } from '@/utils/header-generator';
|
|
import ofetch from '@/utils/ofetch';
|
|
|
|
describe('header-generator', () => {
|
|
it('should has no ua', async () => {
|
|
const response = await ofetch('http://rsshub.test/headers');
|
|
expect(response['user-agent']).toBeUndefined();
|
|
});
|
|
|
|
it('should match ua configurated', async () => {
|
|
const testUa = 'Mozilla/5.0 (iPhone; CPU iPhone OS 17_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.0 Mobile/15E148 Safari/604.1';
|
|
const response = await ofetch('http://rsshub.test/headers', {
|
|
headers: {
|
|
'user-agent': testUa,
|
|
},
|
|
});
|
|
expect(response['user-agent']).toBe(testUa);
|
|
});
|
|
|
|
it('generateHeaders should include sec-ch and sec-fetch headers', () => {
|
|
const headers = generateHeaders(PRESETS.MODERN_MACOS_CHROME);
|
|
|
|
expect(headers['user-agent']).toBeDefined();
|
|
expect(headers['sec-ch-ua']).toBeDefined();
|
|
expect(headers['sec-ch-ua-mobile']).toBeDefined();
|
|
expect(headers['sec-ch-ua-platform']).toBeDefined();
|
|
expect(headers['sec-fetch-site']).toBeDefined();
|
|
expect(headers['sec-fetch-mode']).toBeDefined();
|
|
expect(headers['sec-fetch-user']).toBeDefined();
|
|
expect(headers['sec-fetch-dest']).toBeDefined();
|
|
|
|
expect(headers['sec-ch-ua-platform']).toBe('"macOS"');
|
|
expect(headers['sec-ch-ua-mobile']).toBe('?0');
|
|
});
|
|
|
|
it('generateHeaders should work with headerGeneratorOptions', () => {
|
|
const headers = generateHeaders(PRESETS.MODERN_WINDOWS_CHROME);
|
|
|
|
expect(headers['user-agent']).toBeDefined();
|
|
expect(headers['sec-ch-ua']).toBeDefined();
|
|
expect(headers['sec-ch-ua-mobile']).toBeDefined();
|
|
expect(headers['sec-ch-ua-platform']).toBeDefined();
|
|
|
|
// Platform may vary due to header-generator randomness, just check it's a quoted string
|
|
expect(headers['sec-ch-ua-platform']).toMatch(/^".*"$/);
|
|
expect(headers['sec-ch-ua-mobile']).toBe('?0');
|
|
expect(headers['user-agent']).toMatch(/Chrome/);
|
|
});
|
|
|
|
it('generateHeaders should use default preset when no preset is provided', () => {
|
|
const headers = generateHeaders();
|
|
|
|
expect(headers['user-agent']).toBeDefined();
|
|
expect(headers['sec-ch-ua']).toBeDefined();
|
|
expect(headers['sec-ch-ua-mobile']).toBeDefined();
|
|
expect(headers['sec-ch-ua-platform']).toBeDefined();
|
|
|
|
expect(headers['sec-ch-ua-platform']).toBe('"macOS"');
|
|
expect(headers['sec-ch-ua-mobile']).toBe('?0');
|
|
expect(headers['user-agent']).toMatch(/Chrome/);
|
|
});
|
|
});
|
|
|
|
describe('header-generator (mocked)', () => {
|
|
afterEach(() => {
|
|
vi.resetModules();
|
|
vi.clearAllMocks();
|
|
vi.unmock('header-generator');
|
|
});
|
|
|
|
it('retries invalid safari user agents', async () => {
|
|
const headersQueue = [{ 'user-agent': 'Mozilla/5.0 Applebot Safari' }, { 'user-agent': 'Mozilla/5.0 Safari' }];
|
|
|
|
vi.doMock('header-generator', () => ({
|
|
HeaderGenerator: class {
|
|
getHeaders() {
|
|
return headersQueue.shift() ?? { 'user-agent': 'Mozilla/5.0 Safari' };
|
|
}
|
|
},
|
|
PRESETS: {
|
|
MODERN_MACOS_CHROME: { mock: true },
|
|
},
|
|
}));
|
|
|
|
vi.resetModules();
|
|
const { generateHeaders: generateMockedHeaders } = await import('@/utils/header-generator');
|
|
const headers = generateMockedHeaders({ preset: 'safari' } as any);
|
|
|
|
expect(headers['user-agent']).toContain('Safari');
|
|
expect(headersQueue.length).toBe(0);
|
|
});
|
|
|
|
it('accepts firefox user agents', async () => {
|
|
const headersQueue = [{ 'user-agent': 'Mozilla/5.0 Firefox' }];
|
|
|
|
vi.doMock('header-generator', () => ({
|
|
HeaderGenerator: class {
|
|
getHeaders() {
|
|
return headersQueue.shift() ?? { 'user-agent': 'Mozilla/5.0 Firefox' };
|
|
}
|
|
},
|
|
PRESETS: {
|
|
MODERN_MACOS_CHROME: { mock: true },
|
|
},
|
|
}));
|
|
|
|
vi.resetModules();
|
|
const { generateHeaders: generateMockedHeaders } = await import('@/utils/header-generator');
|
|
const headers = generateMockedHeaders();
|
|
|
|
expect(headers['user-agent']).toContain('Firefox');
|
|
});
|
|
});
|