Question

我正在尝试使用玩笑来模拟属性tz和一个函数，但我不知道一起模拟这两种情况：

如果运行类似的内容

jest.mock('moment-timezone', () => () => ({weekday: () => 5}))

jest.mock('moment-timezone', () => {
  return {
    tz: {

    }
  }
})

我可以模拟属性tz或指令moment()。我该如何编写一个模拟程序来掩盖此代码？

const moment = require('moment-timezone')

module.exports.send = () => {
  const now = moment()
  moment.tz.setDefault('America/Sao_Paulo')
  return now.weekday()
}

谢谢

Answer 1

您可以利用jest.mock()的第二个参数，该参数可以让您指定要在测试中使用的模拟模块的自定义实现。

在此自定义实现中，您还可以定义一些便利助手来模拟期望的实现值（例如weekday()）。

// send-module.test.js

jest.mock('moment-timezone', () => {
    let weekday
    const moment = jest.fn(() => {
        return {
            weekday: jest.fn(() => weekday),
        }
    })
    moment.tz = {
        setDefault: jest.fn(),
    }
    // Helper for tests to set expected weekday value
    moment.__setWeekday = (value) => weekday = value
    return moment;
})

const sendModule = require('./send-module')

test('test', () => {
    require('moment-timezone').__setWeekday(3)
    expect(sendModule.send()).toBe(3)
})

请注意，如果要模拟的模块具有巨大的API表面，则手动为每个测试文件提供模拟可能会变得乏味且重复。要解决后一种情况，您可以考虑编写一些手动模拟以使其可重用（即使用__mocks__目录约定），并通过使用jest.genMockFromModule()进行补充。

The Jest documentation has some guidance about this.

开玩笑-从moment-timezone模拟属性和函数

1 个答案: