Merge pull request #560 from lifeomic/fix-failing-async-tests

Fix Flaky Async Tests

Author: sternetj

src/hooks/Circles/useJoinCircles.test.tsx

@@ -17,7 +17,7 @@ const queryClient = new QueryClient({
   },
 });
 
-const renderHookInContext = async () => {
+const renderHookInContext = () => {
   return renderHook(() => useJoinCircles(), {
     wrapper: ({ children }) => (
       <QueryClientProvider client={queryClient}>{children}</QueryClientProvider>
@@ -92,7 +92,7 @@ test('joins circles listed in app-config', async () => {
   axiosMock
     .onPatch('/v1/life-research/projects/projectId/app-config/circles')
     .reply(200, {});
-  const { result } = await renderHookInContext();
+  const { result } = renderHookInContext();
   await waitFor(() => expect(result.current.isSuccess).toBeDefined());
   expect(axiosMock.history.patch[0].url).toBe(
     '/v1/life-research/projects/projectId/app-config/circles',
@@ -117,7 +117,7 @@ test('does nothing if all circles are joined', async () => {
     },
   });
   axiosMock.reset();
-  const { result } = await renderHookInContext();
+  const { result } = renderHookInContext();
   await waitFor(() => expect(result.current.isSuccess).toBeDefined());
   await waitFor(() => expect(axiosMock.history.patch.length).toEqual(0));
 });

src/hooks/Circles/usePrivatePost.test.tsx

@@ -111,44 +111,52 @@ describe('useCreatePrivatePostAttachmentMutation', () => {
   });
 
   test('throws on missing asset type', async () => {
-    expect.assertions(1);
     const { result } = renderHookWithInjectedClient();
     const { mutateAsync } = result.current;
 
-    await act(async () => {
-      try {
-        await mutateAsync({
-          asset: {
-            id: 'assetId',
-            type: undefined, // purpose of the test
-            uri: 'https://some-image-url',
-          },
-          userIds: ['user1', 'user2'],
-        });
-      } catch (e) {
-        expect(e).toEqual(new Error("Unknown asset type, can't upload"));
-      }
+    let exception: Error | undefined;
+
+    act(() => {
+      mutateAsync({
+        asset: {
+          id: 'assetId',
+          type: undefined, // purpose of the test
+          uri: 'https://some-image-url',
+        },
+        userIds: ['user1', 'user2'],
+      }).catch((e) => {
+        exception = e;
+      });
+    });
+
+    await waitFor(() => {
+      expect(exception).toEqual(new Error("Unknown asset type, can't upload"));
     });
   });
 
   test('throws on unknown asset type', async () => {
-    expect.assertions(1);
     const { result } = renderHookWithInjectedClient();
     const { mutateAsync } = result.current;
 
-    await act(async () => {
-      try {
-        await mutateAsync({
-          asset: {
-            id: 'assetId',
-            type: 'application/pdf', // purpose of the test
-            uri: 'https://some-image-url',
-          },
-          userIds: ['user1', 'user2'],
-        });
-      } catch (e) {
-        expect(e).toEqual(new Error('Unsupported file type: application/pdf'));
-      }
+    let exception: Error | undefined;
+
+    act(() => {
+      mutateAsync({
+        asset: {
+          id: 'assetId',
+          type: 'application/pdf', // purpose of the test
+          uri: 'https://some-image-url',
+        },
+        userIds: ['user1', 'user2'],
+      }).catch((e) => {
+        exception = e;
+      });
+    });
+
+    await waitFor(() => {
+      expect(exception).toEqual(
+        new Error('Unsupported file type: application/pdf'),
+      );
     });
   });
 });

src/hooks/useFhirClient.test.tsx

@@ -384,10 +384,13 @@ describe('useCreateResourceMutation', () => {
         value: 73.5,
       },
     };
-    await act(async () => {
-      await result.current.mutateAsync(observation);
+    act(() => {
+      result.current.mutateAsync(observation);
     });
 
+    await waitFor(() => {
+      expect(result.current.data).toEqual(resultBundle);
+    });
     expect(axiosMock.history.post[0].url).toBe('/v1/fhir/dstu3/Observation');
     expect(JSON.parse(axiosMock.history.post[0].data)).toEqual({
       ...observation,
@@ -403,9 +406,6 @@ describe('useCreateResourceMutation', () => {
         ],
       },
     });
-    await waitFor(() => {
-      expect(result.current.data).toEqual(resultBundle);
-    });
   });
 });
 
@@ -462,10 +462,13 @@ describe('useUpsertMutation', () => {
         value: 1.8,
       },
     };
-    await act(async () => {
-      await result.current.mutateAsync([observation1, observation2]);
+    act(() => {
+      result.current.mutateAsync([observation1, observation2]);
     });
 
+    await waitFor(() => {
+      expect(result.current.data).toEqual(resultBundle);
+    });
     expect(axiosMock.history.post[0].url).toBe(
       'https://fhir.us.lifeomic.com/acct1/dstu3',
     );
@@ -511,9 +514,6 @@ describe('useUpsertMutation', () => {
         },
       ],
     });
-    await waitFor(() => {
-      expect(result.current.data).toEqual(resultBundle);
-    });
   });
 });
 
@@ -529,19 +529,17 @@ describe('useDeleteResourceMutation', () => {
       return useDeleteResourceMutation();
     }
     const { result } = renderHookInContext(useTestHook);
-    await act(async () => {
-      await result.current.mutateAsync({
+    act(() => {
+      result.current.mutateAsync({
         id: observationId,
         resourceType: 'Observation',
       });
     });
 
-    await act(async () => {
-      await waitFor(() => {
-        expect(axiosMock.history.delete[0].url).toBe(
-          `/v1/fhir/dstu3/Observation/${observationId}`,
-        );
-      });
+    await waitFor(() => {
+      expect(axiosMock.history.delete[0].url).toBe(
+        `/v1/fhir/dstu3/Observation/${observationId}`,
+      );
     });
   });
 });