Move suspense boundary in test to better reflect recommended usage

apollographql · Oct 19, 2023 · 40ceea6 · 40ceea6
1 parent 5e7a5d2
commit 40ceea6
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/react/hooks/__tests__/useInteractiveQuery.test.tsx b/src/react/hooks/__tests__/useInteractiveQuery.test.tsx
@@ -1634,9 +1634,7 @@ it("works with startTransition to change variables", async () => {
   function App() {
     return (
       <ApolloProvider client={client}>
-        <Suspense fallback={<SuspenseFallback />}>
-          <Parent />
-        </Suspense>
+        <Parent />
       </ApolloProvider>
     );
   }
@@ -1651,9 +1649,11 @@ it("works with startTransition to change variables", async () => {
     return (
       <div>
         <button onClick={() => loadQuery({ id: "1" })}>Load first todo</button>
-        {queryRef && (
-          <Todo queryRef={queryRef} onChange={(id) => loadQuery({ id })} />
-        )}
+        <Suspense fallback={<SuspenseFallback />}>
+          {queryRef && (
+            <Todo queryRef={queryRef} onChange={(id) => loadQuery({ id })} />
+          )}
+        </Suspense>
       </div>
     );
   }