TESTING - Shallow renderer with generic Typescript

renderShallow.tsx

import * as React from 'react'
import configureMockStore from 'redux-mock-store'
import { Provider } from 'react-redux'
import { shallow, ShallowWrapper } from 'enzyme'

export type StringObject = { [key: string]: string }
export type StoreData = { [key: string]: string | StringObject }

export interface RenderShallowWithReduxOptions<T> {
  children?: React.ReactNode
  props?: Partial<T>
  storeData?: StoreData
}

const renderShallowWithRedux = <T extends {}>(
  Component: React.ComponentType,
  { children = null, props = {}, storeData = {} }: RenderShallowWithReduxOptions<T> = {
    props: {},
    storeData: {},
  }
): ShallowWrapper => {
  const mockStore = configureMockStore([])
  const store = mockStore(storeData)
  store.dispatch = jest.fn()

  if (children) {
    return shallow(
      <Provider store={store}>
        <Component {...props}>{children}</Component>
      </Provider>
    )
  }
  return shallow(
    <Provider store={store}>
      <Component {...props} />
    </Provider>
  )
}

export default renderShallowWithRedux