Шутка - как издеваться над классом, используемым модулем?

Question

У меня есть класс:

class RequestTimeout {
    constructor(timeoutMilliseconds) {
        this.timeoutMilliseconds = timeoutMilliseconds;
        this.timeoutID = undefined;
    }

    start() {
        return new Promise((resolve, reject) => {
            this.timeoutID = setTimeout(() => reject(new Error(`Request attempt exceeded timeout of ${this.timeoutMilliseconds}`)), this.timeoutMilliseconds);
        });
    }

    clear() {
        if (this.timeoutID) clearTimeout(this.timeoutID);
    }
}

module.exports = RequestTimeout;

Этот класс используется в модуле:

const RequestTimeout = require('./request-timeout');

function Request() {
  ...

  async function withTimeout(request, ms) {
        const timeout = new RequestTimeout(ms);

        return Promise.race([
            request(),
            timeout.start(),
        ])
            .then(
                response => {
                    timeout.clear();
                    return response;
                },
                err => {
                    timeout.clear();
                    throw err;
                }
            );
    }

  ...
}

Как мне высмеивать RequestTimeout в тесте с использованием Request? Например:

it('should clear the timeout following a successful response', async () => {
  nock('http://example.com')
    .get('/')
    .reply(200, { example: true });

  const response = await request.get({ ...baseOptions });

  expect(response.example).toEqual(true);
});

Answer 1

// MOCK

let mockGetTimeOutId = jest.fn();
jest.mock('../request-timeout', () => {
    return jest.fn().mockImplementation((ms) => {
        let timeoutId = undefined;
        return {
            start: () => new Promise((resolve, reject) => {
                timeoutId = setTimeout(() => reject(), ms);
            }),
            clear: () => mockGetTimeOutId(timeoutId),
        }
    })
});

// TEST

it('should clear the timeout following a successful response', async () => {
    nock('http://example.com')
        .get('/')
        .reply(200, { example: true });

    expect(mockGetTimeOutId).toHaveBeenCalledTimes(0);

    const response = await request.get({ ...baseOptions });

    expect(mockGetTimeOutId).toHaveBeenCalledTimes(1);
    expect(response.example).toEqual(true);
});