From 6b5098386a3c3ed1feefd5b454776280b3afabfa Mon Sep 17 00:00:00 2001 From: Michael Mac-Vicar Date: Tue, 25 Feb 2025 19:01:34 -0300 Subject: [PATCH] Allow extra domains in test socks proxy --- tests/config/proxy.ts | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/config/proxy.ts b/tests/config/proxy.ts index 42910134fc..4b9a8c2c8a 100644 --- a/tests/config/proxy.ts +++ b/tests/config/proxy.ts @@ -129,16 +129,16 @@ export class TestProxy { } export async function setupSocksForwardingServer({ - port, forwardPort, allowedTargetPort + port, forwardPort, allowedTargetPort, additionalAllowedHosts = [] }: { - port: number, forwardPort: number, allowedTargetPort: number + port: number, forwardPort: number, allowedTargetPort: number, additionalAllowedHosts?: string[] }) { const connectHosts = []; const connections = new Map(); const socksProxy = new SocksProxy(); socksProxy.setPattern('*'); socksProxy.addListener(SocksProxy.Events.SocksRequested, async (payload: SocksSocketRequestedPayload) => { - if (!['127.0.0.1', 'fake-localhost-127-0-0-1.nip.io', 'localhost'].includes(payload.host) || payload.port !== allowedTargetPort) { + if (!['127.0.0.1', 'fake-localhost-127-0-0-1.nip.io', 'localhost', ...additionalAllowedHosts].includes(payload.host) || payload.port !== allowedTargetPort) { socksProxy.sendSocketError({ uid: payload.uid, error: 'ECONNREFUSED' }); return; }