From 286f4c3d0a66501cf4759ff3d71be9d252cae10c Mon Sep 17 00:00:00 2001
From: Jerel Miller <jerelmiller@gmail.com>
Date: Wed, 18 Dec 2024 13:19:14 -0700
Subject: [PATCH] Add test to ensure switching from non-null to null works as
 expected

---
 .../__tests__/useSuspenseFragment.test.tsx    | 61 +++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/src/react/hooks/__tests__/useSuspenseFragment.test.tsx b/src/react/hooks/__tests__/useSuspenseFragment.test.tsx
index 6268e6e7a2..d4ef67097a 100644
--- a/src/react/hooks/__tests__/useSuspenseFragment.test.tsx
+++ b/src/react/hooks/__tests__/useSuspenseFragment.test.tsx
@@ -1019,6 +1019,67 @@ test("returns cached value when `from` changes from `null` to non-null value", a
   await expect(takeSnapshot).not.toRerender();
 });
 
+test("returns null value when `from` changes from non-null value to `null`", async () => {
+  interface ItemFragment {
+    __typename: "Item";
+    id: number;
+    text: string;
+  }
+
+  const fragment: TypedDocumentNode<ItemFragment> = gql`
+    fragment ItemFragment on Item {
+      id
+      text
+    }
+  `;
+
+  const client = new ApolloClient({ cache: new InMemoryCache() });
+
+  client.writeFragment({
+    fragment,
+    data: {
+      __typename: "Item",
+      id: 1,
+      text: "Item #1",
+    },
+  });
+
+  using _disabledAct = disableActEnvironment();
+  const { takeSnapshot, rerender } = await renderHookToSnapshotStream(
+    ({ id }) =>
+      useSuspenseFragment({
+        fragment,
+        from: id === null ? null : { __typename: "Item", id },
+      }),
+    {
+      initialProps: { id: 1 as null | number },
+      wrapper: ({ children }) => (
+        <ApolloProvider client={client}>{children}</ApolloProvider>
+      ),
+    }
+  );
+
+  {
+    const { data } = await takeSnapshot();
+
+    expect(data).toEqual({
+      __typename: "Item",
+      id: 1,
+      text: "Item #1",
+    });
+  }
+
+  await rerender({ id: null });
+
+  {
+    const { data } = await takeSnapshot();
+
+    expect(data).toBeNull();
+  }
+
+  await expect(takeSnapshot).not.toRerender();
+});
+
 test("suspends until cached value is available when `from` changes from `null` to non-null value", async () => {
   interface ItemFragment {
     __typename: "Item";