来自Jest中的模块的模拟单功能并在其上断言

Question

我试图用Jest的模块中的单个功能模拟一个单个功能，并断言我已经使用某些参数调用了它。我有一个大致看起来像这样的文件：

export const f = () => {
  ...
}
export const g = () => {
  ...
  f(...)
}
export const h = () => {
  ...
  g(...)
}

我正在尝试测试功能 g 和 h ，而我要编写的断言是 f 在调用 g 和 h 时，请使用某些参数调用。因此，在我的测试中，我想模拟 f ，并能够断言它的所谓。但是，当我在测试中进行类似的事情时：

import f, g, h from 'module'

jest.mock('module', () => {
  const original = jest.requireActual('module')
  
  return {
    ...original,
    f: jest.fn()
  }
})
test('g calls f correctly', () => {
  g()
  // I want to assert that f was called with some parameters
})

我没有提到 f ，并且似乎当 g 在测试中调用时，实际函数 f 被称为而不是模拟。我需要在这里进行什么才能实现此功能？

Answer 1

看起来您走在正确的道路上。我建议您尝试以下示例之一：

import { f, g } from './module'

jest.mock('./module', () => {
  const actual = jest.requireActual()

  return { ...actual, f: jest.fn() }
})

test('g calls f correctly', () => {
  g()
  expect(f).toHaveBeenCalled()
})

如果您遇到错误，说 f 应该是模拟功能，则可以尝试以下操作：

import { g } from './module'

// Create a reference of the mocked function to use later in the test.
// The name must start with "mock". It's a Jest rule.
const mockedF = jest.fn()

jest.mock('./module', () => {
  const actual = jest.requireActual()

  return { ...actual, f: mockedF }
})

test('g calls f correctly', () => {
  g()
  expect(mockedF).toHaveBeenCalled()
})