File size: 2,526 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 | import { describe, expect, it, vi } from 'vitest';
import createMultiProxy from '@/utils/proxy/multi-proxy';
const baseProxyObj = {
protocol: undefined,
host: undefined,
port: undefined,
auth: undefined,
url_regex: '.*',
strategy: 'all',
};
describe('multi-proxy', () => {
it('returns empty results when no valid proxy is provided', () => {
const result = createMultiProxy(['http://inv lid.test'], baseProxyObj);
expect(result.allProxies).toHaveLength(0);
expect(result.getNextProxy()).toBeNull();
expect(() => result.resetProxy('http://inv lid.test')).not.toThrow();
});
it('rotates proxies, marks inactive, and reactivates after health checks', () => {
vi.useFakeTimers();
try {
const result = createMultiProxy(['http://proxy1.local:8080', 'http://proxy2.local:8081'], {
...baseProxyObj,
healthCheckInterval: 20,
});
const first = result.getNextProxy();
expect(first).not.toBeNull();
const firstUri = first!.uri;
const secondUri = result.allProxies.find((proxy) => proxy.uri !== firstUri)!.uri;
result.markProxyFailed(firstUri);
result.markProxyFailed(firstUri);
result.markProxyFailed(firstUri);
const firstState = result.allProxies.find((proxy) => proxy.uri === firstUri)!;
expect(firstState.isActive).toBe(false);
result.markProxyFailed(secondUri);
result.markProxyFailed(secondUri);
result.markProxyFailed(secondUri);
expect(result.getNextProxy()).toBeNull();
vi.advanceTimersByTime(45);
expect(firstState.isActive).toBe(true);
result.resetProxy(firstUri);
expect(firstState.failureCount).toBe(0);
} finally {
vi.clearAllTimers();
vi.useRealTimers();
}
});
it('returns null when proxies become inactive during selection', () => {
const result = createMultiProxy(['http://proxy1.local:8080', 'http://proxy2.local:8081'], baseProxyObj);
for (const proxy of result.allProxies) {
let calls = 0;
Object.defineProperty(proxy, 'isActive', {
configurable: true,
get() {
calls += 1;
return calls === 1;
},
});
}
expect(result.getNextProxy()).toBeNull();
});
});
|