用 Jest 模拟 new Function()

Question

我在尝试使用构造函数模拟模块时遇到问题

// code.js
const ServiceClass = require('service-library');
const serviceInstance = new ServiceClass('some param');
exports.myFunction = () => {
  serviceInstance.doSomething();
};

以及测试代码：

// code.test.js
const ServiceClass = require('service-library');
jest.mock('service-library');
const {myFunction} = require('../path/to/my/code');

test('check that the service does something', () => {
  // ????
});

它不像文档示例模拟模块，因为您需要在导入模块后实例化该模块。并且也不像模拟函数。

我如何在测试时模拟这个

doSomething()

函数？

作为参考，我试图在这里模拟

@google-cloud/*

包。我有一些项目可以利用这一点。

Answer 1

您需要首先模拟整个模块，以便返回一个玩笑模拟。然后导入到您的测试中并将模拟设置为一个函数，该函数返回一个持有间谍的对象

doSomething

。对于测试，使用

new

调用的类的模拟与使用

new

调用的函数之间没有区别。

import ServiceLibrary from 'service-library'

jest.mock('service-library', () => jest.fn())

const doSomething = jest.fn()
ServiceLibrary.mockImplementation(() => ({doSomething}))

Answer 2

按照 @andreas-köberle 解决方案，我能够像这样模拟

@google-cloud/bigquery

：

// mock bigquery library
const BigQuery = require('@google-cloud/bigquery');
jest.mock('@google-cloud/bigquery', () => jest.fn());
const load = jest.fn(() => ({'@type': 'bigquery#load_job'}));
const table = jest.fn(() => ({load}));
const dataset = jest.fn(() => ({table}));
BigQuery.mockImplementation(() => ({dataset}));

// mock cloud storage library
const {Storage} = require('@google-cloud/storage');
jest.mock('@google-cloud/storage');
const file = jest.fn(name => ({'@type': 'storage#file', name}));
const bucket = jest.fn(() => ({file}));
Storage.mockImplementation(() => ({bucket}));

我将其留在这里作为参考，以防其他人用谷歌搜索类似的内容。但要明确的是，这只是 @andreas-köberle 答案

的具体说明