VanillaJS 每个测试模拟

Question

我正在编写一些代码，我有一个我想要实现的虚拟场景，如下所示。我有一个正在测试的函数，它使用来自不同模块的函数。我想模拟导入模块的功能。我该怎么做呢？

我见过this solution，但jest.doMock 函数在commonJS 中不起作用——这是我的理解。这是一个节点后端项目。

任何帮助将不胜感激。谢谢

// helper.js

let add = (a, b) => a + b;
let mult = (a, b) => a * b;

module.exports = { add, sub, mult };

// index.js

const { add, mult } = require('./helper');

const add5thenMultiply2 = (a) => {
  const res1 = add(a, 5);
  return mult(res1, 2);
};

module.exports = { add5thenMultiply2 };

// index.test.js

const { add5thenMultiply2 } = require('./index');

describe('index', () => {
  it('returns right value', () => {
    const input = 3;
    const expected = 16;
    const result = add5thenMultiply2(input);
    expect(result).toEqual(expected);
  });

  it('works with mock', () => {
    // mock add() such that it always returns 100, but for this test only
    const input = 3;
    const expected = 200;
    const result = add5thenMultiply2(input);
    expect(result).toEqual(expected);
  });
});

Answer 1

jest.doMock(moduleName, factory, options) 与 commonJS 配合得很好。您还需要使用jest.requireActual(moduleName) 部分模拟./helper 模块，即只模拟add 函数，继续使用原来的mult 函数。

例如

helper.js:

let add = (a, b) => a + b;
let mult = (a, b) => a * b;

module.exports = { add, mult };

index.js:

const { add, mult } = require('./helper');

const add5thenMultiply2 = (a) => {
  const res1 = add(a, 5);
  return mult(res1, 2);
};

module.exports = { add5thenMultiply2 };

index.test.js:

describe('index', () => {
  beforeEach(() => {
    jest.resetModules();
  });
  it('returns right value', () => {
    const { add5thenMultiply2 } = require('./index');
    const input = 3;
    const expected = 16;
    const result = add5thenMultiply2(input);
    expect(result).toEqual(expected);
  });

  it('works with mock', () => {
    jest.doMock('./helper', () => {
      return {
        ...jest.requireActual('./helper'),
        add: jest.fn().mockReturnValue(100),
      };
    });
    const { add5thenMultiply2 } = require('./index');
    const input = 3;
    const expected = 200;
    const result = add5thenMultiply2(input);
    expect(result).toEqual(expected);
  });
});

测试结果：

 PASS  examples/67830139/index.test.js (7.751 s)
  index
    ✓ returns right value (6554 ms)
    ✓ works with mock (2 ms)

-----------|---------|----------|---------|---------|-------------------
File       | % Stmts | % Branch | % Funcs | % Lines | Uncovered Line #s 
-----------|---------|----------|---------|---------|-------------------
All files  |     100 |      100 |     100 |     100 |                   
 helper.js |     100 |      100 |     100 |     100 |                   
 index.js  |     100 |      100 |     100 |     100 |                   
-----------|---------|----------|---------|---------|-------------------
Test Suites: 1 passed, 1 total
Tests:       2 passed, 2 total
Snapshots:   0 total
Time:        8.308 s