re-add onSubscriptionData mock to test

apollographql · Jan 25, 2022 · a565fd5 · a565fd5
1 parent f0cd321
commit a565fd5
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/react/hooks/__tests__/useSubscription.test.tsx b/src/react/hooks/__tests__/useSubscription.test.tsx
@@ -135,10 +135,12 @@ describe('useSubscription Hook', () => {
       cache: new Cache({ addTypename: false })
     });
 
+    const onSubscriptionData = jest.fn();
     const { result, unmount, waitForNextUpdate, rerender } = renderHook(
       ({ variables }) => useSubscription(subscription, {
         variables,
-        skip: true
+        skip: true,
+        onSubscriptionData,
       }),
       {
         initialProps: {
@@ -159,10 +161,11 @@ describe('useSubscription Hook', () => {
     expect(result.current.data).toBe(undefined);
 
     rerender({ variables: { foo: 'bar2' }});
-    expect(onSetup).toHaveBeenCalledTimes(0);
-
     await expect(waitForNextUpdate({ timeout: 20 }))
       .rejects.toThrow('Timed out');
+
+    expect(onSetup).toHaveBeenCalledTimes(0);
+    expect(onSubscriptionData).toHaveBeenCalledTimes(0);
     unmount();
   });