@@ -45,17 +45,7 @@ export const injectMocks = (
4545 if ( ! mocks || mocks . length === 0 ) {
4646 throw new Error ( 'Unable to instantiate mocks' ) ;
4747 }
48-
49- const restMocks = mocks . filter (
50- ( m ) => ! [ 'GRAPHQL' , 'WEBSOCKET' ] . includes ( m . method )
51- ) as HttpMock [ ] ;
52- const graphQLMocks = mocks . filter (
53- ( m ) => m . method === 'GRAPHQL'
54- ) as GraphQLMock [ ] ;
55-
56- const webSocketMocks = mocks . filter (
57- ( m ) => m . method === 'WEBSOCKET'
58- ) as WebSocketMock [ ] ;
48+ const { restMocks, graphQLMocks, webSocketMocks } = getMocksByType ( mocks ) ;
5949
6050 restMocks . forEach ( handleRestMock ) ;
6151 graphQLMocks . forEach ( handleGraphQLMock ) ;
@@ -87,16 +77,11 @@ export const reduceAllMocksForScenario = (
8777
8878 const mocks = defaultMocks . concat ( scenarioMocks ) ;
8979
90- const initialHttpMocks = mocks . filter (
91- ( m ) => ! [ 'GRAPHQL' , 'WEBSOCKET' ] . includes ( m . method )
92- ) as HttpMock [ ] ;
93- const initialGraphQlMocks = mocks . filter (
94- ( { method } ) => method === 'GRAPHQL'
95- ) as GraphQLMock [ ] ;
96-
97- const initialWebsocketMocks = mocks . filter (
98- ( m ) => m . method === 'WEBSOCKET'
99- ) as WebSocketMock [ ] ;
80+ const {
81+ restMocks : initialHttpMocks ,
82+ graphQLMocks : initialGraphQlMocks ,
83+ webSocketMocks : initialWebsocketMocks ,
84+ } = getMocksByType ( mocks ) ;
10085
10186 const websocketMocksByUrl = initialWebsocketMocks . reduce <
10287 Record < string , WebSocketMock >
@@ -313,3 +298,17 @@ const handleWebsocketMock = ({ url, server }: WebSocketMock) => {
313298 */
314299const addDelay = ( delay : number ) =>
315300 new Promise ( ( res ) => setTimeout ( res , delay ) ) ;
301+
302+ const getMocksByType = ( mocks : Mock [ ] ) => {
303+ const restMocks = mocks . filter (
304+ ( m ) => ! [ 'GRAPHQL' , 'WEBSOCKET' ] . includes ( m . method )
305+ ) as HttpMock [ ] ;
306+ const graphQLMocks = mocks . filter (
307+ ( m ) => m . method === 'GRAPHQL'
308+ ) as GraphQLMock [ ] ;
309+
310+ const webSocketMocks = mocks . filter (
311+ ( m ) => m . method === 'WEBSOCKET'
312+ ) as WebSocketMock [ ] ;
313+ return { restMocks, graphQLMocks, webSocketMocks } ;
314+ } ;
0 commit comments