-
Notifications
You must be signed in to change notification settings - Fork 230
/
pure.tsx
43 lines (36 loc) 路 1.19 KB
/
pure.tsx
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import React, { Suspense } from 'react'
import { act, create, ReactTestRenderer } from 'react-test-renderer'
import { TestHookProps, NativeRendererOptions, NativeRendererReturn } from 'types'
import { createRenderHook, cleanup, addCleanup, removeCleanup } from 'core/index'
import TestHook from 'core/testHook'
function createNativeRenderer<TProps, TResult>(
testHookProps: Omit<TestHookProps<TProps, TResult>, 'hookProps'>,
{ wrapper: Wrapper }: NativeRendererOptions<TProps>
): NativeRendererReturn<TProps> {
let container: ReactTestRenderer
const toRender = (props?: TProps): JSX.Element => (
<Wrapper {...(props as TProps)}>
<TestHook hookProps={props} {...testHookProps} />
</Wrapper>
)
return {
render(props) {
act(() => {
container = create(<Suspense fallback={null}>{toRender(props)}</Suspense>)
})
},
rerender(props) {
act(() => {
container.update(<Suspense fallback={null}>{toRender(props)}</Suspense>)
})
},
unmount() {
act(() => {
container.unmount()
})
},
act
}
}
const renderHook = createRenderHook(createNativeRenderer)
export { renderHook, act, cleanup, addCleanup, removeCleanup }