RSSHub/lib/utils/ofetch.test.ts

import http from 'node:http';
import type { AddressInfo } from 'node:net';

import { http as mswHttp, HttpResponse } from 'msw';
import { afterEach, describe, expect, it, vi } from 'vitest';

const isAddressInfo = (address: string | AddressInfo | null): address is AddressInfo => typeof address === 'object' && address !== null;

const loadOfetchWithLogger = async () => {
    vi.resetModules();
    const { default: logger } = await import('@/utils/logger');
    const { default: ofetch } = await import('@/utils/ofetch');
    return { logger, ofetch };
};

afterEach(() => {
    vi.restoreAllMocks();
    vi.unstubAllGlobals();
});

describe('ofetch', () => {
    it('marks prefer-proxy header on retryable responses', async () => {
        const { default: server } = await import('@/setup.test');
        server.use(mswHttp.get('http://rsshub.test/fail-500', () => HttpResponse.text('fail', { status: 500 })));

        const { logger, ofetch } = await loadOfetchWithLogger();
        const warnSpy = vi.spyOn(logger, 'warn').mockImplementation(() => logger);

        await expect(
            ofetch('http://rsshub.test/fail-500', {
                retry: 1,
                retryDelay: 0,
                onResponse({ options }) {
                    (options as { headers: Headers | null }).headers = null;
                },
            })
        ).rejects.toBeDefined();

        expect(warnSpy).toHaveBeenCalled();
    });

    it('surfaces the root cause in fetch error messages', async () => {
        const { logger, ofetch } = await loadOfetchWithLogger();
        vi.spyOn(logger, 'error').mockImplementation(() => logger);
        const networkError = new TypeError('fetch failed', { cause: new Error('getaddrinfo ENOTFOUND t.me') });
        vi.stubGlobal('fetch', vi.fn().mockRejectedValue(networkError));

        await expect(ofetch('https://t.me/s/telegram', { retry: 0 })).rejects.toThrow('fetch failed (getaddrinfo ENOTFOUND t.me)');
    });

    it('logs redirected responses', async () => {
        const { logger, ofetch } = await loadOfetchWithLogger();
        const httpSpy = vi.spyOn(logger, 'http').mockImplementation(() => logger);

        const server = http.createServer((req, res) => {
            if (req.url === '/redirect') {
                res.statusCode = 302;
                res.setHeader('Location', '/target');
                res.end();
                return;
            }
            res.statusCode = 200;
            res.end('ok');
        });

        await new Promise<void>((resolve) => server.listen(0, resolve));
        const address = server.address();
        if (!isAddressInfo(address)) {
            throw new TypeError('expected the test server to listen on a TCP port');
        }
        const { port } = address;

        try {
            await ofetch(`http://127.0.0.1:${port}/redirect`);
        } finally {
            server.close();
        }

        expect(httpSpy).toHaveBeenCalled();
    });
});