mirror of
https://github.com/orangecoding/fredy.git
synced 2026-06-16 12:31:07 +00:00
104 lines
4.2 KiB
JavaScript
104 lines
4.2 KiB
JavaScript
/*
|
|
* Copyright (c) 2026 by Christian Kellner.
|
|
* Licensed under Apache-2.0 with Commons Clause and Attribution/Naming Clause
|
|
*/
|
|
|
|
import { expect } from 'vitest';
|
|
|
|
import {
|
|
getPreLaunchConfig,
|
|
applyBotPreventionToPage,
|
|
applyLanguagePersistence,
|
|
applyPostNavigationHumanSignals,
|
|
} from '../../../lib/services/extractor/botPrevention.js';
|
|
|
|
describe('botPrevention helper', () => {
|
|
it('getPreLaunchConfig builds deterministic values when jitter disabled', () => {
|
|
const url = 'https://example.com/some/path';
|
|
const options = {
|
|
acceptLanguage: 'de-DE,de;q=0.9',
|
|
userAgent: 'TestAgent/1.0',
|
|
viewport: { width: 1200, height: 700, deviceScaleFactor: 2 },
|
|
viewportJitter: false,
|
|
referer: 'https://example.com/ref',
|
|
timezone: 'Europe/Berlin',
|
|
};
|
|
const cfg = getPreLaunchConfig(url, options);
|
|
|
|
expect(cfg.acceptLanguage).toBe('de-DE,de;q=0.9');
|
|
expect(cfg.langArg).toBe('--lang=de-DE');
|
|
expect(cfg.windowSizeArg).toBe('--window-size=1200,700');
|
|
expect(cfg.viewport).toEqual({ width: 1200, height: 700, deviceScaleFactor: 2 });
|
|
expect(cfg.userAgent).toBe('TestAgent/1.0');
|
|
expect(cfg.headers['Accept-Language']).toBe('de-DE,de;q=0.9');
|
|
expect(cfg.headers['User-Agent']).toBe('TestAgent/1.0');
|
|
expect(cfg.headers.Referer).toBe('https://example.com/ref');
|
|
expect(cfg.extraArgs).toContain('--disable-blink-features=AutomationControlled');
|
|
expect(cfg.extraArgs).toContain('--proxy-bypass-list=<-loopback>');
|
|
});
|
|
|
|
it('applyBotPreventionToPage sets UA, viewport, headers and injects patches', async () => {
|
|
const calls = [];
|
|
const page = {
|
|
setUserAgent: async (ua) => calls.push(['setUserAgent', ua]),
|
|
setViewport: async (vp) => calls.push(['setViewport', vp]),
|
|
setJavaScriptEnabled: async (on) => calls.push(['setJavaScriptEnabled', on]),
|
|
setExtraHTTPHeaders: async (h) => calls.push(['setExtraHTTPHeaders', h]),
|
|
emulateTimezone: async (tz) => calls.push(['emulateTimezone', tz]),
|
|
evaluateOnNewDocument: async (fn) => calls.push(['evaluateOnNewDocument', typeof fn]),
|
|
};
|
|
const cfg = getPreLaunchConfig('https://example.org/', {
|
|
userAgent: 'Foo/Bar',
|
|
acceptLanguage: 'en-US,en',
|
|
viewport: { width: 1000, height: 600, deviceScaleFactor: 1 },
|
|
viewportJitter: false,
|
|
timezone: 'UTC',
|
|
});
|
|
|
|
await applyBotPreventionToPage(page, cfg);
|
|
|
|
expect(calls[0]).toEqual(['setUserAgent', 'Foo/Bar']);
|
|
expect(calls.some((c) => c[0] === 'setViewport' && c[1].width === 1000 && c[1].height === 600)).toBe(true);
|
|
expect(calls.some((c) => c[0] === 'setJavaScriptEnabled' && c[1] === true)).toBe(true);
|
|
const headerCall = calls.find((c) => c[0] === 'setExtraHTTPHeaders');
|
|
expect(headerCall).toBeDefined();
|
|
expect(headerCall[1]['Accept-Language']).toBe('en-US,en');
|
|
expect(headerCall[1]['User-Agent']).toBe('Foo/Bar');
|
|
expect(calls.some((c) => c[0] === 'emulateTimezone' && c[1] === 'UTC')).toBe(true);
|
|
expect(calls.some((c) => c[0] === 'evaluateOnNewDocument' && c[1] === 'function')).toBe(true);
|
|
});
|
|
|
|
it('applyLanguagePersistence stores languages early', async () => {
|
|
const calls = [];
|
|
const page = {
|
|
evaluateOnNewDocument: async (fn, arg) => calls.push(['evaluateOnNewDocument', typeof fn, arg]),
|
|
};
|
|
const cfg = getPreLaunchConfig('https://example.org/', {
|
|
acceptLanguage: 'de-DE,de;q=0.9',
|
|
viewportJitter: false,
|
|
});
|
|
await applyLanguagePersistence(page, cfg);
|
|
const call = calls[0];
|
|
expect(call[0]).toBe('evaluateOnNewDocument');
|
|
expect(call[1]).toBe('function');
|
|
expect(call[2]).toBe('de-DE,de');
|
|
});
|
|
|
|
it('applyPostNavigationHumanSignals moves mouse and scrolls when enabled', async () => {
|
|
const mouseCalls = [];
|
|
const page = {
|
|
mouse: {
|
|
move: async (x, y, opts) => mouseCalls.push(['move', x, y, opts && typeof opts.steps === 'number']),
|
|
wheel: async (opts) => mouseCalls.push(['wheel', typeof opts.deltaY === 'number']),
|
|
},
|
|
};
|
|
const cfg = {
|
|
humanDelay: true,
|
|
viewport: { width: 1200, height: 800 },
|
|
};
|
|
await applyPostNavigationHumanSignals(page, cfg);
|
|
expect(mouseCalls.some((c) => c[0] === 'move')).toBe(true);
|
|
expect(mouseCalls.some((c) => c[0] === 'wheel')).toBe(true);
|
|
});
|
|
});
|