mockar função dentro de chamada de modulo

com-spy.test.js

import fn from "./index";
import module from './module';

jest.mock("./module", () => {
  const mod = jest.requireActual('./module').default;

  const mocked = () => {
    const moduleInstance = mod();

    mocked.lastMockedOnClick = jest.spyOn(moduleInstance, 'onClick');
    mocked.lastMockedOnEvent = jest.spyOn(moduleInstance, 'onEvent');

    return moduleInstance;
  };

  return mocked;
});

afterEach(jest.clearAllMocks);
afterAll(jest.restoreAllMocks);

test("sem mockar eventos", () => {
  fn();
  expect(module.lastMockedOnClick).toHaveBeenCalledTimes(1);
  expect(module.lastMockedOnEvent).toHaveBeenCalledTimes(1);
});

test("mockando eventos", () => {
  module.lastMockedOnClick.mockImplementation(() => console.log('mocked onClick'));
  module.lastMockedOnEvent.mockImplementation(() => console.log('mocked onEvent'));
  fn();
  expect(module.lastMockedOnClick).toHaveBeenCalledTimes(1);
  expect(module.lastMockedOnEvent).toHaveBeenCalledTimes(1);
});

sem-spy.test.js

import fn from "./index";
import module from './module';

jest.mock("./module", () => {
  const mocked = () => ({
    onClick: () => {
      mocked.lastOnClickCalled = jest.fn(() => console.log('mocked onClick'));
      return mocked.lastOnClickCalled();
    },
    onEvent: () => {
      mocked.lastOnEventCalled = jest.fn(() => console.log('mocked onEvent'));
      return mocked.lastOnEventCalled();
    },
  });
  
  return mocked;
});

afterEach(jest.clearAllMocks);
afterAll(jest.restoreAllMocks);

test("...", () => {
  fn();
  expect(module.lastOnClickCalled).toHaveBeenCalled();
});