jestでexportの外で使用している関数をmockしたい

Question

したいこと

jestを使ってreactをテストしようとしているのですが、モジュールのmockがうまくいきません。
export defaultの関数の外でmock対象の関数を実行していることが原因みたいなのですが、元のコードを変えずにテストしたいです。
mockしたいモジュールはmaterial-uiのmakeStylesという関数です。

コード

テスト対象のコード

import { makeStyles } from "@material-ui/core/styles";
const useStyles = makeStyles(theme => ({
  root: {
    spacing: 8,
    flexGrow: 1
  }
}));
export default function Search(props) {
  const classes = useStyles();
  return(
    <div className={classes.root}>テスト</div>
  );
}

テストコード

import * as muiStyle from "@material-ui/core/styles";
import "jest";
import React from "react";
import { act } from "react-test-renderer";
import Search from "./Search";
jest.mock("@material-ui/core/styles");
muiStyle.makeStyles = jest.fn().mockReturnValue(() => {root:"dummy"});

test("serch", async () => {
  let component;
  await act(async () => {
    component = create(<Search />);
  });
  const tree = component.toJSON();
  expect(tree).toMatchSnapshot();
}

###出力されたエラー
TypeError: useStyles is not a functionと出ます。
useStylesにはundefindが入っているみたいです。

試したこと

テストコードの順番をこのように変えてみましたがうまくいきませんでした。

import Search from "./Search";
jest.mock("@material-ui/core/styles");
muiStyle.makeStyles = jest.fn().mockReturnValue(() => {root:"dummy"});

Search関数内でmakeStylesを呼び出すと問題なくmockされます。

何か良い解決策があれば、ご教授ください。