maplibre-gl/src/util/test/mock_fetch.ts

BSD licensed community fork of mapbox-gl, a WebGL interactive maps library

export class RequestMock implements Partial<Request> {
    public readonly cache: RequestCache;
    public readonly headers: Headers = new Headers();
    public readonly method?: string;
    public readonly url?: string;

    public get signal(): AbortSignal {
        return null;
    }

    constructor(input: RequestInfo | URL, init?: RequestInit) {
        this.cache = typeof input === 'object' && 'cache' in input ? input.cache : init.cache;
        this.method = typeof input === 'object' && 'method' in input ? input.method : init.method;
        this.url = typeof input === 'object' && 'url' in input ? input.url : input.toString();
        this.headers = typeof input === 'object' && 'headers' in input ? new Headers(input.headers) : new Headers(init.headers || {});
    }
}

class AbortControllerMock {
    public signal: AbortSignal;

    public abort(): void {}
}

export type FetchMock = jest.Mock<Promise<Response>, [input: RequestInfo | URL, init?: RequestInit], any>;

let _AbortController: typeof AbortController;
let _Request: typeof Request;
let _fetch: typeof fetch;

export function destroyFetchMock(): void {
    global.AbortController = _AbortController ?? global.AbortController;
    global.Request = _Request ?? global.Request;
    global.fetch = _fetch ?? global.fetch;
}

export function setupFetchMock(): FetchMock {
    _AbortController = _AbortController ?? global.AbortController;
    _Request = _Request ?? global.Request;
    _fetch = _fetch ?? global.fetch;

    const fetchMock = jest.fn(async (_input: RequestInfo | URL, _init?: RequestInit): Promise<Response> => {
        return <Response>{};
    });

    global.AbortController = AbortControllerMock;
    global.Request = RequestMock as unknown as typeof Request;
    global.fetch = fetchMock as any;

    return fetchMock;
}