Jest 怎么 Mock 掉整个模块？-JavaScript中文网-JavaScript教程资源分享门户

在前端开发中经常会遇到需要进行单元测试的情况，而 Jest 是一个流行的 JavaScript 测试框架。在进行单元测试时，常常需要 Mock 掉一些模块，以便可以有效地进行测试。在 Jest 中，我们可以使用 Jest.MockedFunction 来 Mock 掉一个函数或者对象，但是有些情况下我们需要 Mock 掉整个模块。那么，Jest 怎么 Mock 掉整个模块？本文将详细介绍如何使用 Jest 来 Mock 掉整个模块，希望能够对你有所帮助。

Mock 整个模块

在使用 Jest 进行单元测试时，Mock 掉整个模块的实现方式很简单，只需要使用 jest.mock() 方法即可。这个方法接收两个参数，第一个参数是需要 Mock 掉的模块的路径，第二个参数是模拟的实现。模拟的实现可以是一个函数，也可以是一个对象。当需要从被测试模块中导入这个模块时，Jest 会自动使用我们提供的模拟实现。下面是一个简单的示例代码。

// utils.js
export const add = (a, b) => {
  return a + b;
}

// math.js
import { add } from './utils';

export const sum = (a, b) => {
  return add(a, b);
}

// math.test.js
import { sum } from './math';

jest.mock('./utils', () => ({
  add: jest.fn((a, b) => a + b)
}));

describe('sum function', () => {
  test('adds two numbers', () => {
    expect(sum(1, 2)).toBe(3);
    expect(sum(2, 3)).toBe(5);
    expect(sum(4, 5)).toBe(9);
    expect(sum(5, 6)).toBe(11);
  });
});

在上面的示例代码中，我们 Mock 掉了 utils 模块。我们使用了 jest.fn() 来创建了一个被 Mock 掉的 add 函数。我们可以在 describe 中编写测试用例并调用 sum 函数。当我们测试 sum 函数时，Jest 会自动使用我们提供的模拟实现来替换掉被测试的 utils 模块。

模拟 Node.js 模块

当我们需要 Mock 掉的是一个 Node.js 模块时，我们可以使用 jest.mock() 和 jest.requireActual() 方法来实现。jest.mock() 方法的第二个参数可以传入一个函数，这个函数接收一个相同的参数，它会返回一个基于实际模块的模拟。jest.requireActual() 方法可以用来获取实际模块的导出值，这样我们就可以在模拟模块中使用它们。下面是一个示例代码。

// fs.js
import fs from 'fs';

export const readFile = (filePath) => {
  return fs.readFileSync(filePath, 'utf8');
};

// fs.test.js
import fs from 'fs';
import { readFile } from './fs';

jest.mock('fs', () => ({
  readFileSync: jest.fn()
}));

describe('readFile function', () => {
  test('reads a file', () => {
    fs.readFileSync.mockReturnValue('Hello World');

    expect(readFile('file.txt')).toBe('Hello World');
    expect(fs.readFileSync).toHaveBeenCalledWith('file.txt', 'utf8');
  });
});

在上面的示例代码中，我们 Mock 掉了 fs 模块。我们使用了 jest.fn() 来创建了一个被 Mock 掉的 readFileSync 函数。我们使用 jest.mock() 方法来模拟 fs 模块。然后我们在测试用例中，调用 readFile 函数，并验证它的返回值是否正确。我们还使用了 fs.readFileSync.mockReturnValue() 方法来设置被 Mock 掉的函数的返回值，并且使用 fs.readFileSync.mock.calls 来验证这个函数有没有被正确调用。

总结

Jest 是一个很强大的单元测试框架，在进行单元测试时可以使用 Jest 来 Mock 掉一些模块，以便可以有效地进行测试。在 Jest 中，我们可以使用 jest.mock() 方法来 Mock 掉一个模块，还可以使用 jest.requireActual() 方法来获取实际模块的导出值。希望这篇文章可以帮助你更加深入地掌握如何使用 Jest 来 Mock 掉整个模块。

来源：JavaScript中文网，转载请注明来源本文地址：https://www.javascriptcn.com/post/65b0e64fadd4f0e0ffa3eb71