begin test

Author: enisdenjo

src/index.ts

@@ -45,6 +45,7 @@ async function getTestCases(router: ReturnType<typeof createRouter>) {
       import("./test-suites/requires-with-argument/index.js"),
       import("./test-suites/keys-mashup/index.js"),
       import("./test-suites/null-keys/index.js"),
+      import("./test-suites/i-will-name-you-later/index.js"),
     ].map((i) => i.then((e) => e.default))
   );
 

src/test-suites/i-will-name-you-later/a.subgraph.ts

@@ -0,0 +1,86 @@
+import { createSubgraph } from "../../subgraph.js";
+import { comments, posts } from "./data.js";
+
+export default createSubgraph("a", {
+  typeDefs: /* GraphQL */ `
+    extend schema
+      @link(
+        url: "https://specs.apollo.dev/federation/v2.3"
+        import: ["@key", "@external", "@requires"]
+      )
+
+    type Query {
+      feed: [Post]
+    }
+
+    type Post @key(fields: "id") {
+      id: ID!
+      comments(limit: Int!): [Comment] @external
+    }
+
+    type Comment @key(fields: "id") {
+      id: ID!
+      authorId: ID
+      body: String!
+      sameCommentOnOtherPosts: [Post] @external
+      isCommentSpam: Boolean
+        @requires(
+          fields: "sameCommentOnOtherPosts { comments(limit: 50) { id } }"
+        )
+    }
+  `,
+  resolvers: {
+    Query: {
+      feed() {
+        return posts.map((post) => ({ id: post.id }));
+      },
+    },
+    Post: {
+      __resolveReference(key: { id: string }) {
+        const post = posts.find((post) => post.id === key.id);
+
+        if (!post) {
+          return null;
+        }
+
+        return {
+          id: post.id,
+        };
+      },
+      comments(post: { id: string }, { limit }: { limit: number }) {
+        return comments
+          .filter((comment) => comment.postId === post.id)
+          .slice(0, limit)
+          .map((c) => ({
+            id: c.id,
+            authorId: c.authorId,
+            body: c.body,
+          }));
+      },
+    },
+    Comment: {
+      __resolveReference(ref: {
+        id: string;
+        sameCommentOnOtherPosts?: { comments: { id: string }[] }[];
+      }) {
+        const comment = comments.find((c) => c.id === ref.id);
+
+        if (!comment) {
+          return null;
+        }
+
+        return {
+          id: comment.id,
+          authorId: comment.authorId,
+          body: comment.body,
+          isCommentSpam: ref.sameCommentOnOtherPosts?.length
+            ? ref.sameCommentOnOtherPosts?.every((p) =>
+                // making sure that the ref contains the full "@requires"
+                p.comments.every((c) => c.id)
+              )
+            : false,
+        };
+      },
+    },
+  },
+});

src/test-suites/i-will-name-you-later/b.subgraph.ts

@@ -0,0 +1,107 @@
+import { createSubgraph } from "../../subgraph.js";
+import { authors, comments, posts } from "./data.js";
+
+export default createSubgraph("b", {
+  typeDefs: /* GraphQL */ `
+    extend schema
+      @link(
+        url: "https://specs.apollo.dev/federation/v2.3"
+        import: ["@key", "@external", "@requires"]
+      )
+
+    type Post @key(fields: "id") {
+      id: ID!
+      author: Author @requires(fields: "comments(limit: 3) { authorId }")
+      comments(limit: Int!): [Comment]
+    }
+
+    type Comment @key(fields: "id") {
+      id: ID!
+      date: String
+      authorId: ID @external
+      sameCommentOnOtherPosts: [Post]
+    }
+
+    type Author {
+      id: ID!
+      name: String
+    }
+  `,
+  resolvers: {
+    Post: {
+      __resolveReference(key: {
+        id: string;
+        comments?: Array<{ authorId: string }>;
+      }) {
+        const post = posts.find((p) => p.id === key.id);
+
+        if (!post) {
+          return null;
+        }
+
+        if (key.comments) {
+          if (key.comments.length !== 3) {
+            throw new Error("Expected 3 comments");
+          }
+
+          return {
+            id: post.id,
+            authorId: key.comments?.[2].authorId,
+          };
+        }
+
+        return {
+          id: post.id,
+        };
+      },
+      author(post: { authorId?: string }) {
+        if (!post.authorId) {
+          return null;
+        }
+
+        const author = authors.find((a) => a.id === post.authorId);
+
+        if (!author) {
+          return null;
+        }
+
+        return {
+          id: author.id,
+          name: author.name,
+        };
+      },
+      comments(post: { id: string }, { limit }: { limit: number }) {
+        const coms = comments
+          .filter((comment) => comment.postId === post.id)
+          .slice(0, limit)
+          .map((c) => ({
+            id: c.id,
+          }));
+
+        return coms;
+      },
+    },
+    Comment: {
+      __resolveReference(key: { id: string }) {
+        const comment = comments.find((c) => c.id === key.id);
+
+        if (!comment) {
+          return null;
+        }
+
+        return {
+          id: comment.id,
+        };
+      },
+      sameCommentOnOtherPosts({ id: commentId }: { id: string }) {
+        const comment = comments.find((c) => c.id === commentId);
+        if (!comment) {
+          return [];
+        }
+        return comments
+          .filter((c) => c.body === comment.body && c.postId !== comment.postId)
+          .map((c) => ({ id: c.postId }));
+      },
+    },
+  },
+});

src/test-suites/i-will-name-you-later/data.ts

@@ -0,0 +1,81 @@
+export const products = [
+  {
+    upc: "p1",
+    name: "p-name-1",
+    price: 11,
+    weight: 1,
+    category: {
+      averagePrice: 11,
+    },
+  },
+  {
+    upc: "p2",
+    name: "p-name-2",
+    price: 22,
+    weight: 2,
+    category: {
+      averagePrice: 22,
+    },
+  },
+];
+
+export const posts = [
+  {
+    id: "p1",
+    body: "p1-body",
+  },
+  {
+    id: "p2",
+    body: "p2-body",
+  },
+];
+
+export const comments = [
+  {
+    id: "c1",
+    postId: "p1",
+    authorId: "a2",
+    body: "c1-body",
+  },
+  {
+    id: "c2",
+    postId: "p1",
+    authorId: "a2",
+    body: "c2-body",
+  },
+  {
+    id: "c3",
+    postId: "p1",
+    authorId: "a2",
+    body: "c3-body",
+  },
+  {
+    id: "c4",
+    postId: "p2",
+    authorId: "a1",
+    body: "c4-body",
+  },
+  {
+    id: "c5",
+    postId: "p2",
+    authorId: "a1",
+    body: "c5-body",
+  },
+  {
+    id: "c6",
+    postId: "p2",
+    authorId: "a1",
+    body: "c6-body",
+  },
+];
+
+export const authors = [
+  {
+    id: "a1",
+    name: "a1-name",
+  },
+  {
+    id: "a2",
+    name: "a2-name",
+  },
+];

src/test-suites/i-will-name-you-later/index.ts

@@ -0,0 +1,6 @@
+import { serve } from "../../supergraph.js";
+import a from "./a.subgraph.js";
+import b from "./b.subgraph.js";
+import test from "./test.js";
+
+export default serve("i-will-name-you-later", [a, b], test);

src/test-suites/i-will-name-you-later/test.ts

@@ -0,0 +1,31 @@
+import { createTest } from "../../testkit.js";
+
+export default [
+  createTest(
+    /* GraphQL */ `
+      {
+        feed {
+          author {
+            id
+          }
+        }
+      }
+    `,
+    {
+      data: {
+        feed: [
+          {
+            author: {
+              id: "a2",
+            },
+          },
+          {
+            author: {
+              id: "a1",
+            },
+          },
+        ],
+      },
+    }
+  ),
+];