@@ -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