Merge branch 'main' into fix-type

apollographql · Aug 26, 2024 · d9ab0d1 · d9ab0d1
2 parents fec8704 + a3c26e3
commit d9ab0d1
Showing 1 changed file with 115 additions and 1 deletion.
diff --git a/src/__tests__/ApolloClient.ts b/src/__tests__/ApolloClient.ts
@@ -12,7 +12,7 @@ import { Kind } from "graphql";
 import { Observable } from "../utilities";
 import { ApolloLink } from "../link/core";
 import { HttpLink } from "../link/http";
-import { InMemoryCache } from "../cache";
+import { createFragmentRegistry, InMemoryCache } from "../cache";
 import { itAsync } from "../testing";
 import { ObservableStream, spyOnConsole } from "../testing/internal";
 import { TypedDocumentNode } from "@graphql-typed-document-node/core";
@@ -2515,6 +2515,120 @@ describe("ApolloClient", () => {
         });
       }
     });
+
+    it("works with nested fragments", async () => {
+      const cache = new InMemoryCache();
+      const client = new ApolloClient({
+        cache,
+        link: ApolloLink.empty(),
+      });
+
+      const ItemNestedFragment = gql`
+        fragment ItemNestedFragment on Item {
+          complete
+        }
+      `;
+
+      const ItemFragment = gql`
+        fragment ItemFragment on Item {
+          id
+          text
+          ...ItemNestedFragment
+        }
+
+        ${ItemNestedFragment}
+      `;
+
+      cache.writeFragment({
+        fragment: ItemFragment,
+        fragmentName: "ItemFragment",
+        data: {
+          __typename: "Item",
+          id: 5,
+          text: "Item #5",
+          complete: true,
+        },
+      });
+
+      const observable = client.watchFragment({
+        fragment: ItemFragment,
+        fragmentName: "ItemFragment",
+        from: { __typename: "Item", id: 5 },
+      });
+
+      const stream = new ObservableStream(observable);
+
+      {
+        const result = await stream.takeNext();
+
+        expect(result).toEqual({
+          data: {
+            __typename: "Item",
+            id: 5,
+            text: "Item #5",
+            complete: true,
+          },
+          complete: true,
+        });
+      }
+    });
+
+    it("can use the fragment registry for nested fragments", async () => {
+      const fragments = createFragmentRegistry();
+      const cache = new InMemoryCache({ fragments });
+
+      fragments.register(gql`
+        fragment ItemNestedFragment on Item {
+          complete
+        }
+      `);
+
+      const client = new ApolloClient({
+        cache,
+        link: ApolloLink.empty(),
+      });
+
+      const ItemFragment = gql`
+        fragment ItemFragment on Item {
+          id
+          text
+          ...ItemNestedFragment
+        }
+      `;
+
+      cache.writeFragment({
+        fragment: ItemFragment,
+        fragmentName: "ItemFragment",
+        data: {
+          __typename: "Item",
+          id: 5,
+          text: "Item #5",
+          complete: true,
+        },
+      });
+
+      const observable = client.watchFragment({
+        fragment: ItemFragment,
+        fragmentName: "ItemFragment",
+        from: { __typename: "Item", id: 5 },
+      });
+
+      const stream = new ObservableStream(observable);
+
+      {
+        const result = await stream.takeNext();
+
+        expect(result).toEqual({
+          data: {
+            __typename: "Item",
+            id: 5,
+            text: "Item #5",
+            complete: true,
+          },
+          complete: true,
+        });
+      }
+    });
   });
 
   describe("defaultOptions", () => {