如何在每次测试之前开玩笑地重置 state？

Question

我有以下类和文件

//A.ts 

export abstract class A{
    
    protected abstract method1(arg){
        
    }
}

// C.ts
export interface C{
    public methodC1(){
    };
}

//CImpl.ts
export class CImpl implements C{
    public methodC1(){
        // do something
    };
}

// B.ts
    import appConfig from 'config';
    export class B extends A{
        private _enable : boolean appConfig.get('IsEnabledC');
        private init(argC:C | undefined)
        {
            if(this._enable){
                argC = new CImpl();
            }
            argC.methodC1();
            return argC;
        }
    private func(argC){
        let resource = getResource();
        switch(resource.status){
            case 1:{
                throw(some exception);
            }
            case 2:{
                throw(another exception);
            }
            default:{
                // do something
            }
    }
       
    private cleanup(){
        //do something;
    }
     protected async method1(arg){
            let argC: C | undefined;
            try{
                argC = init(argC);
                await this.func(argC);
            }
            catch(exception){
                this.cleanup();
            }
        }
    }


// B.spec.ts
describe("B test", () => {
    it("check if cleanup is called on resource status 1", () => {
        const varB = new B();
        const spied = jest.spyOn(B.prototype as any,"init").mockReturnValue(new CImpl(argC));
        const spied2 = jest.spyOn(B.prototype as any,"cleanup").mockImplementation(()=>{console.log("cleanup got called")});
        await varB["method1"](arg);
        expect(spied).toBeCalledTimes(1);
        expect(spied2).toBeCalledTimes(1);
    });

    it("check if cleanup is called on resource status 2", () => {
        const varB = new B();
        const spied = jest.spyOn(B.prototype as any,"init").mockReturnValue(new CImpl(argC));
        const spied2 = jest.spyOn(B.prototype as any,"cleanup").mockImplementation(()=>{console.log("cleanup got called")});
        await varB["method1"](arg);
        expect(spied).toBeCalledTimes(1);
        expect(spied2).toBeCalledTimes(1);
    });
})

我想测试两个状态，即。 1 和 2 用于 func 方法中的资源。 我还分别在测试中模拟了 getResource() 方法和设置状态 1 和 2。 我希望在每个测试中只调用一次清理方法。 但最终在第二次测试中， spied 和 spied2 都被调用了 2 次。 当我分别运行每个测试时，它们工作正常，但是当它们一起运行时，第一个测试通过并且第二个测试在期望断言中失败，消息 jest.fn() 被调用了 2 次。

如何在每次测试之后/之前重置 state？

Answer 1

您可以使用这些方法： beforeAll, afterAll, beforeEach, afterEach 。 https://jestjs.io/docs/setup-teardown

要清除模拟，您可以使用.mockClear() https://jestjs.io/docs/mock-function-api#mockfnmockclear

upd：您只能在模拟函数上使用它。 它可以在测试中的任何地方调用，但毕竟你可能需要它。 它看起来像这样：

import { yourMockedFunction } from '...'

afterAll(() => {
  yourMockedFunction.clearMock();
});

// your tests here