106 lines
3.2 KiB
TypeScript
106 lines
3.2 KiB
TypeScript
import { RouterContext, BlitzRouter, BlitzProvider } from "blitz"
|
|
import { render as defaultRender } from "@testing-library/react"
|
|
import { renderHook as defaultRenderHook } from "@testing-library/react-hooks"
|
|
|
|
export * from "@testing-library/react"
|
|
|
|
// --------------------------------------------------------------------------------
|
|
// This file customizes the render() and renderHook() test functions provided
|
|
// by React testing library. It adds a router context wrapper with a mocked router.
|
|
//
|
|
// You should always import `render` and `renderHook` from this file
|
|
//
|
|
// This is the place to add any other context providers you need while testing.
|
|
// --------------------------------------------------------------------------------
|
|
|
|
// --------------------------------------------------
|
|
// render()
|
|
// --------------------------------------------------
|
|
// Override the default test render with our own
|
|
//
|
|
// You can override the router mock like this:
|
|
//
|
|
// const { baseElement } = render(<MyComponent />, {
|
|
// router: { pathname: '/my-custom-pathname' },
|
|
// });
|
|
// --------------------------------------------------
|
|
export function render(
|
|
ui: RenderUI,
|
|
{ wrapper, router, dehydratedState, ...options }: RenderOptions = {}
|
|
) {
|
|
if (!wrapper) {
|
|
// Add a default context wrapper if one isn't supplied from the test
|
|
wrapper = ({ children }) => (
|
|
<BlitzProvider dehydratedState={dehydratedState}>
|
|
<RouterContext.Provider value={{ ...mockRouter, ...router }}>
|
|
{children}
|
|
</RouterContext.Provider>
|
|
</BlitzProvider>
|
|
)
|
|
}
|
|
return defaultRender(ui, { wrapper, ...options })
|
|
}
|
|
|
|
// --------------------------------------------------
|
|
// renderHook()
|
|
// --------------------------------------------------
|
|
// Override the default test renderHook with our own
|
|
//
|
|
// You can override the router mock like this:
|
|
//
|
|
// const result = renderHook(() => myHook(), {
|
|
// router: { pathname: '/my-custom-pathname' },
|
|
// });
|
|
// --------------------------------------------------
|
|
export function renderHook(
|
|
hook: RenderHook,
|
|
{ wrapper, router, dehydratedState, ...options }: RenderHookOptions = {}
|
|
) {
|
|
if (!wrapper) {
|
|
// Add a default context wrapper if one isn't supplied from the test
|
|
wrapper = ({ children }) => (
|
|
<BlitzProvider dehydratedState={dehydratedState}>
|
|
<RouterContext.Provider value={{ ...mockRouter, ...router }}>
|
|
{children}
|
|
</RouterContext.Provider>
|
|
</BlitzProvider>
|
|
)
|
|
}
|
|
return defaultRenderHook(hook, { wrapper, ...options })
|
|
}
|
|
|
|
export const mockRouter: BlitzRouter = {
|
|
basePath: "",
|
|
pathname: "/",
|
|
route: "/",
|
|
asPath: "/",
|
|
params: {},
|
|
query: {},
|
|
isReady: true,
|
|
isLocaleDomain: false,
|
|
isPreview: false,
|
|
push: jest.fn(),
|
|
replace: jest.fn(),
|
|
reload: jest.fn(),
|
|
back: jest.fn(),
|
|
prefetch: jest.fn(),
|
|
beforePopState: jest.fn(),
|
|
events: {
|
|
on: jest.fn(),
|
|
off: jest.fn(),
|
|
emit: jest.fn(),
|
|
},
|
|
isFallback: false,
|
|
}
|
|
|
|
type DefaultParams = Parameters<typeof defaultRender>
|
|
type RenderUI = DefaultParams[0]
|
|
type RenderOptions = DefaultParams[1] & { router?: Partial<BlitzRouter>; dehydratedState?: unknown }
|
|
|
|
type DefaultHookParams = Parameters<typeof defaultRenderHook>
|
|
type RenderHook = DefaultHookParams[0]
|
|
type RenderHookOptions = DefaultHookParams[1] & {
|
|
router?: Partial<BlitzRouter>
|
|
dehydratedState?: unknown
|
|
}
|