首页 > 解决方案 > 在节点中使用 JEST 模拟类

问题描述

所以,这实际上是我多次发生的事情,我从来不知道如何解决它。

我有这样的课

//myClass.js
const { get } = require('../models/myModel');

class MyClass {
    constructor(id) {
        this.id = id;
    }

    async getByID() {
        return get(this.id);
    }
}

和一个控制器

//controller.js
module.exports = MyClass;

const MyClass = require('./MyClass.js');

module.exports = {
    getController: (req, res, next) => {
        try {
            const MyObject = new MyClass(1);
            const info = await MyObject.getByID();
            res.send(info)
        }
        catch (e) {
            next(e);
        }
    }
}

当我想做一个 E2E 测试时,我需要模拟它getByID以查看它解决和拒绝时会发生什么。

那么,如何模拟构造函数呢?我做了这样的事情

// myclass.spec.js


const MyClass = require('./MyClass.js');

jest.mock('./MyClass', () => jest.fn().mockImplementation(() => ({
  getByID: Promise.resolve({id: 1, name: 'John'}),
})));


describe('Testing MyClass', () => {
    it('Should return info', () => {
        const res = httpMocks.createResponse();
        const mReq = httpMocks.createRequest();
        const mNext = jest.fn();
        const mRes = mockResponse();
        await pipelineController(mReq, mRes, mNext);
        expect(mRes.send).toHaveBeenCalledWith(1);
        done();
    })
})

我知道这个测试目前有效,但现在我无法更改该getByID模拟值以查看它拒绝承诺时会发生什么。

如果我尝试将它包含在测试(it)中......它不会模拟任何东西......我想要类似的东西


const MyClass = require('./MyClass.js');
const {pipelineController} = require('./controllers.js')

jest.mock('./MyClass', () => jest.fn().mockImplementation(() => ({
  getInfo: jest.fn(),
})));


describe('Testing MyClass', () => {
    it('Should return info', () => {
        ProcessService.getInfo.mockImplementation(() => Promise.resolve('the value i want'))
        const res = httpMocks.createResponse();
        const mReq = httpMocks.createRequest();
        const mNext = jest.fn();
        const mRes = mockResponse();
        await pipelineController(mReq, mRes, mNext);
        expect(mRes.send).toHaveBeenCalledWith(1);
        done();
    })
    it('Should return info', () => {
        ProcessService.getInfo.mockImplementation(() => Promise.reject('the reason i want'))
        const res = httpMocks.createResponse();
        const mReq = httpMocks.createRequest();
        const mNext = jest.fn();
        const mRes = mockResponse();
        await pipelineController(mReq, mRes, mNext);
        expect(mRes.send).toHaveBeenCalledWith(1);
        done();
    })

})

标签: javascriptnode.jsunit-testingjestjs

解决方案


由于getByID在实例化后立即调用,因此应该在模拟类之外提供一个间谍以更改实现。

这可以通过以模糊和未指定方式工作的类自动模拟来完成,但可能适用于这种特定情况:

jest.mock('./MyClass.js')
const MyClass = require('./MyClass.js');
...
MyClass.prototype.getByID.mockResolvedValue({id: 1, name: 'John'});
// instantiate MyClass and call getByID

这可以通过在课堂外暴露一个间谍来完成:

const mockGetByID = jest.fn();
jest.mock('../../../services/ProcessService', () => jest.fn(() => (
  { getByID: mockGetByID }
)));
const MyClass = require('./MyClass.js');
...
mockGetByID.mockResolvedValue({id: 1, name: 'John'});
// instantiate MyClass and call getByID

推荐阅读