From cf82df9fdd31d1a7bf82e9dfa73e13f6b190f45b Mon Sep 17 00:00:00 2001
From: Jerel Miller <jerelmiller@gmail.com>
Date: Wed, 15 Jan 2025 17:23:41 -0700
Subject: [PATCH] Rewrite test with renderHookToSnapshotStream and
 toEqualQueryResult

---
 src/react/hooks/__tests__/useQuery.test.tsx | 86 ++++++++++++---------
 1 file changed, 50 insertions(+), 36 deletions(-)

diff --git a/src/react/hooks/__tests__/useQuery.test.tsx b/src/react/hooks/__tests__/useQuery.test.tsx
index b0b660da82f..871a493d380 100644
--- a/src/react/hooks/__tests__/useQuery.test.tsx
+++ b/src/react/hooks/__tests__/useQuery.test.tsx
@@ -11140,14 +11140,25 @@ describe("useQuery Hook", () => {
         cache: new InMemoryCache(),
       });
 
-      const { result } = renderHook(() => useQuery(query), {
-        wrapper: ({ children }) => (
-          <ApolloProvider client={client}>{children}</ApolloProvider>
-        ),
+      using _disabledAct = disableActEnvironment();
+      const { takeSnapshot } = await renderHookToSnapshotStream(
+        () => useQuery(query),
+        {
+          wrapper: ({ children }) => (
+            <ApolloProvider client={client}>{children}</ApolloProvider>
+          ),
+        }
+      );
+
+      await expect(takeSnapshot()).resolves.toEqualQueryResult({
+        data: undefined,
+        called: true,
+        loading: true,
+        networkStatus: NetworkStatus.loading,
+        previousData: undefined,
+        variables: {},
       });
 
-      expect(result.current.loading).toBe(true);
-      expect(result.current.data).toBe(undefined);
       setTimeout(() => {
         link.simulateResult({
           result: {
@@ -11162,19 +11173,19 @@ describe("useQuery Hook", () => {
         });
       });
 
-      await waitFor(
-        () => {
-          expect(result.current.loading).toBe(false);
-        },
-        { interval: 1 }
-      );
-      expect(result.current.data).toEqual({
-        greeting: {
-          message: "Hello world",
-          __typename: "Greeting",
+      await expect(takeSnapshot()).resolves.toEqualQueryResult({
+        data: {
+          greeting: {
+            message: "Hello world",
+            __typename: "Greeting",
+          },
         },
+        called: true,
+        loading: false,
+        networkStatus: NetworkStatus.ready,
+        previousData: undefined,
+        variables: {},
       });
-      expect(result.current).not.toContain({ hasNext: true });
 
       setTimeout(() => {
         link.simulateResult({
@@ -11196,27 +11207,30 @@ describe("useQuery Hook", () => {
         });
       });
 
-      await waitFor(
-        () => {
-          expect(result.current.loading).toBe(false);
-        },
-        { interval: 1 }
-      );
-      await waitFor(
-        () => {
-          expect(result.current.data).toEqual({
-            greeting: {
-              message: "Hello world",
-              __typename: "Greeting",
-              recipient: {
-                name: "Alice",
-                __typename: "Person",
-              },
+      await expect(takeSnapshot()).resolves.toEqualQueryResult({
+        data: {
+          greeting: {
+            message: "Hello world",
+            __typename: "Greeting",
+            recipient: {
+              name: "Alice",
+              __typename: "Person",
             },
-          });
+          },
         },
-        { interval: 1 }
-      );
+        called: true,
+        loading: false,
+        networkStatus: NetworkStatus.ready,
+        previousData: {
+          greeting: {
+            message: "Hello world",
+            __typename: "Greeting",
+          },
+        },
+        variables: {},
+      });
+
+      await expect(takeSnapshot).not.toRerender();
     });
 
     it("should handle deferred queries in lists", async () => {