-
Notifications
You must be signed in to change notification settings - Fork 0
/
makeMockChainFn.ts
44 lines (41 loc) · 1.06 KB
/
makeMockChainFn.ts
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
44
import { JestMockChainFnInternalState } from './constants'
import {
clearExistingCallsHandler,
clearExistingMocksHandler,
makeProxyHandler,
} from './utils'
import {
MockChainFnInternalStateType,
MakeMockChainFnOptions,
MakeMockChainFnReturnType,
MocksMappingType,
CallsAccumulatorType,
} from './typedefs'
export const makeMockChainFn = (
options: MakeMockChainFnOptions = {}
): MakeMockChainFnReturnType => {
const mocks: MocksMappingType = {}
const calls: CallsAccumulatorType = []
const mockChainFnInternalState: MockChainFnInternalStateType = {
mocks,
calls,
}
const mockChainFnObject = {
[JestMockChainFnInternalState]: mockChainFnInternalState,
}
const mockChainFnProxyHandler = makeProxyHandler({
options,
mockChainFnObject,
})
const mockChainFnWithProxy = new Proxy(
mockChainFnObject,
mockChainFnProxyHandler
)
return {
mockChainFn: mockChainFnWithProxy,
calls,
mocks,
clearExistingCalls: clearExistingCallsHandler(calls),
clearExistingMocks: clearExistingMocksHandler(mocks),
}
}