Test conditional rendering based on state

ConditionalComponent.test.js

import * as reactModule from "react";
import { shallow } from "enzyme";
import ConditionalComponent from "./ConditionalComponent";

describe("test Conditional component", () => {
  it("should render while loading", () => {
    const loadingValue = true;
    reactModule.useState = jest.fn(initialLoadingValue => [
      loadingValue,
      () => {}
    ]);
    wrapper = shallow(<ConditionalComponent />);
    expect(wrapper).toMatchSnapshot();
  });

  it("should render after loading", () => {
    const loadingValue = false;
    reactModule.useState = jest.fn(initialLoadingValue => [
      loadingValue,
      () => {}
    ]);
    wrapper = shallow(<ConditionalComponent />);
    expect(wrapper).toMatchSnapshot();
  });
});