如何恢复使用 jest.mock() 创建的模拟？

Question

显然 mock.mockRestore() 不会恢复使用 jest.mock() 创建的模拟的原始实现

// a.js
export default class A {}


// b.js
import A from './a';
export default class B extends A {}


// test.js
import A from './a';
import B from './b';

jest.mock('./a');
jest.mock('./b');

const b = new B();

test('instanceOf', () => {
    A.mockRestore();
    B.mockRestore();
    expect(b).toBeInstanceOf(A); // fails
});

Answer 1

mockFn.mockRestore仅适用于使用jest.spyOn创建的模拟函数：

const obj = {
  func: () => 'original'
}

test('func', () => {
  const mock = jest.spyOn(obj, 'func');
  mock.mockReturnValue('mocked');

  expect(obj.func()).toBe('mocked');  // Success!

  mock.mockRestore();
  expect(obj.func()).toBe('original');  // Success!
})

jest.spyOn包装原始函数并提供mockRestore作为恢复原始函数的方法。

---

jest.mock调用工作方式有点不同。

Jest接管require系统并且jest.mock告诉Jest它应该在require返回模块mock而不是实际模块。

这意味着模块mock不会包装原始模块，它完全取代了require系统中的原始模块。 因此， mockRestore可以在模块mock中的mock函数上定义，但调用它不会恢复原始实现。

---

当您想要为整个测试模拟整个模块时，通常会使用jest.mock 。

当使用ES6样式的import语句时，它特别有用，因为babel-jest提升jest.mock调用并且它们在测试文件中的任何其他内容之前运行（包括任何import语句）：

import A from './a';  // <= A is already mocked...

jest.mock('./a');  // <= ...because this runs first

test('A', () => {
  // ...
}

---

在使用jest.mock的测试期间，没有一种简单的方法可以恢复原始模块，因为它的主要用途是模拟整个测试的模块。

如果您在同一测试期间尝试同时使用模拟和原始实现，则有几个选项：

• 使用模拟一个特定的功能jest.spyOn使用，并恢复它mockRestore
• 使用jest.doMock避免悬挂行为jest.mock ...只是注意，您还需要使用require ，使用范围之内jest.doMock而是采用顶级import
• 随时使用jest.requireAcutal来require原始模块

Answer 2

假设你不能使用 spyOn，你可以这样做：

// test.js

jest.mock('./a');
import A from './a';

A.mockImplementation(params => 'mockVal');
const actualA = jest.requireActual('./a');

test('instanceOf', () => {
    A.mockRestore(); // acts like mockReset()
    A.mockImplementation((params) => {
        return actualA(params);
    });
});