笑话：如何为文件中的某些测试撤消全局模拟

Question

我想为某些测试模拟 Math.random，并将其原始实现用于其他测试。我怎样才能做到这一点？我读过有关使用 jest.doMock 和 jest.dontMock 的信息，但在使用它们时遇到了一些问题，例如：

• 我似乎需要require 才能使用doMock 和dontMock，但是 我的项目只使用 ES6 模块来导入模块
• 这些函数在像Math 这样的全局模块中也存在问题。 尝试使用 jest.doMock("Math.random") 时出现错误， 结果Cannot find module 'Math' from 'app.test.js'

我的测试不一定需要使用doMock 和dontMock。它们似乎是我在笑话文档中能找到的最接近我想要实现的东西。但我对替代解决方案持开放态度。

我想在 app.js 中测试的函数...

export function getRandomId(max) {
    if (!Number.isInteger(max) || max <= 0) {
        throw new TypeError("Max is an invalid type");
    }
    return Math.floor(Math.random() * totalNumPeople) + 1;
}

在 app.test.js 里面...

describe("getRandomId", () => {
  const max = 10;
  Math.random = jest.fn();

  test("Minimum value for an ID is 1", () => {
      Math.mockImplementationOnce(() => 0);
      const id = app.getRandomId(max);
      expect(id).toBeGreaterThanOrEqual(1);
  });

  test("Error thrown for invalid argument", () => {
      // I want to use the original implementation of Math.random here
      expect(() => getRandomId("invalid")).toThrow();
  })
});

Answer 1

试试这个：

describe("getRandomId", () => {
  const max = 10;
  let randomMock;

  beforeEach(() => {
    randomMock = jest.spyOn(global.Math, 'random');
  });

  test("Minimum value for an ID is 1", () => {
      randomMock.mockReturnValue(0);
      const id = getRandomId(max);
      expect(id).toBeGreaterThanOrEqual(1);
  });

  test("Error thrown for invalid argument", () => {
      // I want to use the original implementation of Math.random here
      randomMock.mockRestore(); // restores the original (non-mocked) implementation
      expect(() => getRandomId("invalid")).toThrow();
  })
});