fix up some hoc timings

apollographql · Aug 6, 2021 · 95179a8 · 95179a8
1 parent 644f623
commit 95179a8
Showing 1 changed file with 30 additions and 18 deletions.
diff --git a/src/react/hoc/__tests__/queries/errors.test.tsx b/src/react/hoc/__tests__/queries/errors.test.tsx
@@ -214,24 +214,36 @@ describe('[queries] errors', () => {
             componentDidUpdate() {
               const { props } = this;
               iteration += 1;
-              if (iteration === 1) {
-                // initial loading state is done, we have data
-                expect(props.data!.allPeople).toEqual(
-                  data.allPeople
-                );
-                props.setVar(2);
-              } else if (iteration === 2) {
-                // variables have changed, wee are loading again but also have data
-                expect(props.data!.loading).toBeTruthy();
-              } else if (iteration === 3) {
-                // the second request had an error!
-                expect(props.data!.error).toBeTruthy();
-                expect(props.data!.error!.networkError).toBeTruthy();
-                // // We need to set a timeout to ensure the unhandled rejection is swept up
-                setTimeout(() => {
-                  expect(unhandled.length).toEqual(0);
-                  done = true;
-                });
+              try {
+                if (iteration === 1) {
+                  // initial loading state is done, we have data
+                  expect(props.data!.allPeople).toEqual(
+                    data.allPeople
+                  );
+                  props.setVar(2);
+                } else if (iteration === 2) {
+                  expect(props.data!.allPeople).toEqual(
+                    data.allPeople
+                  );
+                } else if (iteration === 3) {
+                  // variables have changed, wee are loading again but also have data
+                  expect(props.data!.loading).toBeTruthy();
+                } else if (iteration === 4) {
+                  // the second request had an error!
+                  expect(props.data!.error).toBeTruthy();
+                  expect(props.data!.error!.networkError).toBeTruthy();
+                  // // We need to set a timeout to ensure the unhandled rejection is swept up
+                  setTimeout(() => {
+                    try {
+                      expect(unhandled.length).toEqual(0);
+                    } catch (err) {
+                      reject(err);
+                    }
+                    done = true;
+                  });
+                }
+              } catch (err) {
+                reject(err);
               }
             }
             render() {