Also run schema test on GraphQL v15

Author: hansott

library/sources/GraphQL.schema.test.ts

@@ -0,0 +1,179 @@
+import * as t from "tap";
+import { ReportingAPIForTesting } from "../agent/api/ReportingAPIForTesting";
+import { Context, getContext, runWithContext } from "../agent/Context";
+import { startTestAgent } from "../helpers/startTestAgent";
+import { GraphQL } from "./GraphQL";
+import { Token } from "../agent/api/Token";
+
+function getTestContext(): Context {
+  return {
+    remoteAddress: "::1",
+    method: "POST",
+    url: "http://localhost:4000/graphql",
+    query: {},
+    headers: {
+      "content-type": "application/json",
+    },
+    body: {
+      query: '{ getFile(path: "/etc/bashrc") }',
+    },
+    cookies: {},
+    routeParams: {},
+    source: "express",
+    route: "/graphql",
+  };
+}
+
+export function createGraphQLTests(pkgName: string) {
+  t.test("it works", async () => {
+    const api = new ReportingAPIForTesting();
+    const agent = startTestAgent({
+      api: api,
+      token: new Token("123"),
+      wrappers: [new GraphQL()],
+      rewrite: {
+        graphql: pkgName,
+      },
+    });
+
+    const { graphql, buildSchema } = require(
+      pkgName
+    ) as typeof import("graphql");
+
+    const dsl = `
+type Mutation {
+  createBook(title: String!, authorId: ID!): Book!
+  createAuthor(name: String!): Author!
+}
+
+type Query {
+  getBook(id: ID!): Book
+  getAuthor(id: ID!): Author
+}
+
+type Book {
+  id: ID!
+  title: String!
+  author: Author!
+}
+
+type Author {
+  id: ID!
+  name: String!
+  books: [Book!]!
+}`.trimStart();
+
+    const schema = buildSchema(dsl);
+
+    const root = {
+      getBook: ({ id }: { id: string }) => {
+        return {
+          id: id,
+          title: "Book Title",
+          author: {
+            id: "1",
+            name: "Author Name",
+            books: [],
+          },
+        };
+      },
+    };
+
+    const query = async (variableValues: Record<string, unknown>) => {
+      return await graphql({
+        schema,
+        source: `
+        query getBook($id: ID!) {
+          getBook(id: $id) {
+            id
+            title
+            author {
+              id
+              name
+              books {
+                id
+                title
+              }
+            }
+          }
+        }
+      `,
+        rootValue: root,
+        variableValues: variableValues,
+      });
+    };
+
+    api.clear();
+
+    const hits = 3;
+
+    await runWithContext(getTestContext(), async () => {
+      for (let i = 0; i < hits; i++) {
+        await query({ id: "1" });
+
+        // Route is registered at the end of the request
+        agent.onRouteExecute(getTestContext());
+      }
+    });
+
+    await runWithContext(
+      {
+        remoteAddress: "::1",
+        method: "POST",
+        url: "http://localhost:4000/server-rendered-page",
+        query: {},
+        headers: {},
+        body: {
+          query: "this is not a graphql query",
+        },
+        cookies: {},
+        routeParams: {},
+        source: "express",
+        route: "/server-rendered-page",
+      },
+      async () => {
+        await query({ id: "1" });
+
+        // Route is registered at the end of the request
+        agent.onRouteExecute(getContext()!);
+      }
+    );
+
+    await agent.flushStats(1000);
+
+    t.same(api.getEvents().length, 1);
+    const [heartbeat] = api.getEvents();
+    t.match(heartbeat, {
+      type: "heartbeat",
+      routes: [
+        {
+          method: "POST",
+          path: "/graphql",
+          hits: hits,
+          graphql: {
+            type: "query",
+            name: "getBook",
+          },
+          apispec: {},
+          graphQLSchema: undefined,
+        },
+        {
+          method: "POST",
+          path: "/graphql",
+          hits: hits,
+          graphql: undefined,
+          apispec: {},
+          graphQLSchema: dsl,
+        },
+        {
+          method: "POST",
+          path: "/server-rendered-page",
+          hits: 1,
+          graphql: undefined,
+          apispec: {},
+          graphQLSchema: undefined,
+        },
+      ],
+    });
+  });
+}

library/sources/GraphQL.schema.tests.ts

@@ -0,0 +1,3 @@
+import { createGraphQLTests } from "./GraphQL.schema.tests";
+
+createGraphQLTests("graphql-v15");

library/sources/GraphQL.schema.v15.test.ts

@@ -0,0 +1,3 @@
+import { createGraphQLTests } from "./GraphQL.schema.tests";
+
+createGraphQLTests("graphql");