开玩笑 - 从另一个文件中模拟导入的函数不起作用

Question

我一直在尝试模拟一个在另一个文件中导入并在类中使用的函数。这里有类似的问题，我经历了很多，但仍然无法让我的测试正常工作。这是我的代码结构：

//util.js
export const stageHelper = (key) => {
    return STAGE_NAMES[key];
};

//main.js
import {stageHelper} from './util'
class Main {
    static config = {
        value: stageHelper('abc'),
    }
    
    static getValue() {
        return this.config.value;
    }
}

//main.spec.js
import * as util from './util';
import {Configuration} from '../configuration/configuration';
jest.mock('./util');

describe('Main', () => {
    const utilSpy = jest.spyOn(util, 'stageHelper').mockImplementation(() => 'testValue');
    //util.stageHelper = jest.fn().mockImplementation(() => 'testValue'); // tried this too
    //utilSpy.mockReturnValue('ja'); // tried this too
    
    expect(Main.getValue()).toEqual('testValue'); // this test fails - the value is 'undefined'
});

从我的测试中调用Main.getValue() 时，我得到undefined。但是，我希望这会返回 testValue，因为这就是我模拟的返回值。

有人可以帮我解决这个问题吗？这将不胜感激！

Answer 1

您应该在导入 main.js 之前监视 util.stageHelper() 方法。您需要使用require() 方法而不是import。 import 将导入 util.js 的原始版本。在你的测试用例中窥探为时已晚。

例如

util.js:

const STAGE_NAMES = {
  a: '1',
};
export const stageHelper = (key) => {
  return STAGE_NAMES[key];
};

main.js:

import { stageHelper } from './util';

export class Main {
  static config = {
    value: stageHelper('abc'),
  };

  static getValue() {
    return this.config.value;
  }
}

main.spec.js:

import * as util from './util';

describe('Main', () => {
  it('should pass', () => {
    const utilSpy = jest.spyOn(util, 'stageHelper').mockReturnValueOnce('testValue');
    const { Main } = require('./main');
    expect(utilSpy).toBeCalledWith('abc');
    expect(Main.getValue()).toEqual('testValue');
  });
});

单元测试结果：

 PASS  examples/67115206/main.spec.js (12.867 s)
  Main
    ✓ should pass (44 ms)

----------|---------|----------|---------|---------|-------------------
File      | % Stmts | % Branch | % Funcs | % Lines | Uncovered Line #s 
----------|---------|----------|---------|---------|-------------------
All files |    87.5 |      100 |      50 |   85.71 |                   
 main.js  |     100 |      100 |     100 |     100 |                   
 util.js  |      75 |      100 |       0 |   66.67 | 5                 
----------|---------|----------|---------|---------|-------------------
Test Suites: 1 passed, 1 total
Tests:       1 passed, 1 total
Snapshots:   0 total
Time:        16.066 s