Files
RSSHub/lib/utils/proxy/multi-proxy.test.ts
T
DIYgod 81ff849337 feat(tests): add comprehensive unit tests for various utility functions and views (#20779)
* feat(tests): add comprehensive unit tests for various utility functions and views

- Implement tests for cache utility to ensure proper behavior with no cache and TTL keys.
- Add tests for common utilities including string manipulation and path handling.
- Introduce tests for directory import functionality to validate file imports and pattern matching.
- Create tests for git hash retrieval to handle fallback scenarios.
- Develop tests for deprecated got utility to verify response handling and retry logic.
- Enhance got utility tests to include request hooks and search parameter handling.
- Mock header generator tests to validate user agent handling.
- Expand helpers tests to cover current path retrieval and duration parsing.
- Implement tests for ofetch utility to ensure proper proxy handling and logging.
- Add OpenTelemetry metric tests to validate metric serialization.
- Create proxy tests to verify multi-proxy selection and failure handling.
- Introduce request rewriter tests to validate fetch and get wrapper functionality.
- Add timezone utility tests to handle various input types.
- Implement view tests for Atom and RSS rendering to ensure correct output.
- Create index view tests to validate debug information display based on configuration.

* refactor: remove deprecated got implementation and associated tests

* test: expand coverage for api, middleware, and utils
2025-12-29 17:34:48 +08:00

77 lines
2.5 KiB
TypeScript

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();
});
});