Я пытаюсь использоватьmount из Enzyme для тестирования моего компонента, в который вложено несколько компонентов Material UI. Я получаю эту ошибку при запуске теста:

TypeError: Cannot read property 'prepareStyles' of undefined

После некоторого копания,Я обнаружил, что тема должна быть передана в контексте, Я делаю это в тесте, но все еще получаю эту ошибку.

Мой тест:

import expect from  'expect';
import React, {PropTypes} from 'react';
import {mount} from 'enzyme';
import SearchBar from './SearchBar';
import getMuiTheme from 'material-ui/styles/getMuiTheme';

function setup() {
  const muiTheme = getMuiTheme();

  const props = {
    closeSearchBar: () => {},
    fetchSearchData: () => {},
    data: [],
    searching: false
  };

  return mount(<SearchBar {...props} />, {context: {muiTheme}});
}

describe('SearchBar Component', ()=> {

  it('Renders search toolbar properly', () => {
    const wrapper = setup();
    expect(wrapper.find('.toolbar').length).toBe(1);
    expect(wrapper.find('button').length).toBe(1);
  });
});

Мой компонент панели поиска является компонентом без сохранения состояния, поэтому я не использую его в любом контексте. Но даже когда я, я все равно получаю ту же ошибку.

Что я делаю неправильно?