jest mock class funciton code example

Example 1: jest mock class

test("es6 class", () => {
    const SomeClass = jest.fn();
    const mMock = jest.fn();

    SomeClass.mockImplementation(() => {
      return {
        m: mMock
      };
    });

    const some = new SomeClass();
    some.m("a", "b");
    expect(mMock.mock.calls).toEqual([["a", "b"]]);
  });

Example 2: jest spy on class method

//class.js
class MyClass {
  methodOne() {
    return 1;
  }
  methodTwo() {
    return 2;
  }
}
module.exports = MyClass;

// class.test.js
test('spy using class method', () => {
  const result = new MyClass()
  const spy = jest.spyOn(result, 'methodOne')
  result.methodOne()

  // check class method is call or not
  expect(spy).toHaveBeenCalled()

  // expect old value
  expect(result.methodOne()).toBe(1)

  // expect new value
  spy.mockReturnValueOnce(12)
  expect(result.methodOne()).toBe(12)
})