reactjsjestjsmockingreact-hooks

Mocking React custom hook with Jest


I need to mock my custom hook when unit testing React component. I have read few tutorials and stackoverflow answers to this simple task, but without luck to implement it correctly.

My simplest possible setup for single test is as following:

// TestComponent.js

import React from "react";
import useTest from "./useTest";

const TestComponent = () => {
  const { state } = useTest("initial_value");

  return <div>{state}</div>;
};

export default TestComponent;
// useTest.jsx - simple custom hook

import React, { useState } from "react";

const useTest = (initialState) => {
  const [state] = useState(initialState);
  return { state };
};

export default useTest;

// TestComponent.test.jsx - my test case

import React from "react";
import { render } from "@testing-library/react";
import TestComponent from "./TestComponent";

jest.mock("./useTest", () => ({
  useTest: () => "mocked_value",
}));

test("rendertest", () => {
  const component = render(<TestComponent />);
  expect(component.container).toHaveTextContent("mocked_value");
});

So I trying to mock useTest custom hook to return "mocked_value", instead of "initial_value" from real custom hook. But above code just gives me this error:

TypeError: (0 , _useTest.default) is not a function

      3 | 
      4 | const TestComponent = () => {
    > 5 |   const { state } = useTest("initial_value");
        |                     ^
      6 | 
      7 |   return <div>{state}</div>;
      8 | };

I have also tried:

import useTest from './useTest';
// ...
jest.spyOn(useTest, "useTest").mockImplementation(() => "mocked_value");
import useTest from './useTest';
// ...
jest.spyOn(useTest, "useTest").mockReturnValue("mocked_value");

But both gives me error Cannot spy the useTest property because it is not a function; undefined given instead.

How do I implement this test?


Solution

  • I'm answering to myself. This way it's working:

    jest.mock("./useTest", () => ({
      useTest: () => ({ state: 'mocked_value' }),
    }));
    

    And if I want to use default export in custom hook:

    jest.mock("./useTest", () => ({
      __esModule: true,
      default: () => ({ state: 'mocked_value' }),
    }));
    

    Also, if I want to also use setState method in my hook and export it, I can mock it like this:

    const mockedSetState = jest.fn();
    
    jest.mock("./useTest", () => ({
      useTest: () => ({ state, setState: mockedSetState }),
    }));
    

    And now it's possible to check if setState has been called once:

    expect(mockedSetState).toHaveBeenCalledTimes(1);