如何使用Jest覆盖(或模拟)类方法以测试函数?

时间:2018-06-06 09:07:44

标签: javascript node.js unit-testing jestjs

我对调用类的函数的单元测试存在问题。它似乎总是称为“官方”类实例,而不是我的模拟类。我无法强制我的函数使用我的模拟实例......

有一个文件包含我要测试的功能:

const myClass = require('./myClass');
const instance = new myClass();

module.exports.functionToTest = async function () {

    // Some stuff...

    const value = await instance.myMethod();

    // Some stuff that define a result variable (partially with value).

    return result;
}

我的班级定义有一个文件:

module.exports = class myClass {
    async myMethod() {
        const result = await someStuffWillResolveMaybeTrueOrFalse();

        console.log('We used the original myMethod... Mocking has failed.');

        return result;
    }
}

有一个spec文件:

const myFile = require('./myFile');
const myClass = require('./myClass');

describe('My test', async () => {
    it('should mock myClass.myMethod in order to return false', () => {
        const instance = new myClass();
        instance.myMethod = jest.fn().mockResolvedValue(false);

        const result = await myFile.functionToTest();

        expect(result).toBeTruthy();
    }
}

不幸的是我的测试正在传递(因为myMethod返回“true”)并记录“我们使用了原始的myMethod ......模拟失败了。”

所以我想通过模拟myMethod返回false来使我的测试总是失败。

你能帮帮我吗?谢谢你的时间。

1 个答案:

答案 0 :(得分:0)

嗯。我找到了解决方案。

请参阅。使用目标函数更改文件。

const myClass = require('./myClass');
// const instance = new myClass(); <== Not here...

module.exports.functionToTest = async function () {
    const instance = new myClass(); // <== ...but there.

    // Some stuff...

    const value = await instance.myMethod();

    // Some stuff that define a result variable (partially with value).

    return result;
} 

我的spec文件:

const myFile = require('./myFile');

// I specify to Jest that I'll mock a file
jest.mock('./myClass');
const myClass = require('./myClass');

// I prepare the mock function. In that case a promise wich resolve 'false'
const mMock = jest.fn().mockResolvedValue(false);

// I mock the method 'myMethod' in 'myClass'
myClass.mockImplementation(() => {
    return {
        myMethod: mMock
    };
});


// Then, I just take the test
describe('My test', async () => {
    it('should mock myClass.myMethod in order to return false', () => {
        const result = await myFile.functionToTest();

        expect(result).toBeFalsy();
    }
}