恢复 ts-jest 中导入的 class 函数的模拟

restore mock of imported class function in ts-jest

问题

我正在模拟一个从另一个 class 导入的函数(只有一个),该函数在主 class 中被调用。我能够模拟函数和 return 我提供的值。但是我无法为后续测试将模拟功能恢复正常。

如有任何帮助,我们将不胜感激!

框架使用:jest + ts-jest

代码

~main.ts

import {SubClass} from './subclass.ts'
export class MainClass {
 let sub: SubClass = new SubClass()

public async Car(){
  let start = await sub.key();
  return start
}

}

~sub.ts

export class SubClass{ 

public async key(){
  return "you can start the car"
}
}

main.test.ts

import {SubClass} from './subclass.ts'
import {MainClass} from './mainclass.ts'
import {mocked} from 'ts-jest/utils'

jest.mock(./subclass.ts)

let main = new MainClass()
let sub = new SubClass()
let mockedFn = mocked(sub,true)

mockedFn.key = jest.fn().mockImplementation(() => console.log('mocking succesfull'))

afterEach(()=>{
mockedFn.key.mockRestore() // tried mockClear(), mockReset()
}

it('test for main func mocked result',async ()=>{
 const result = await main.car()
 expect(result).toEqual("mocking succesfull")

}
it('test for main func result',async ()=>{
 const result = await main.car()
 expect(result).toEqual("you can start the car") // getting result as undefined
}


原始 mockedFn.key 方法在重新分配时丢失,Jest 无法恢复它:

mockedFn.key = jest.fn()

应该是:

jest.spyOn(mockedFn, 'key').mockImplementation(...)

如果一个方法或 属性 应该被恢复,它不应该通过直接分配 jest.fn() 来 spied/mocked。这尤其适用于可以交叉污染测试的模拟全局变量。