tests

Author: spion

src/plugins/typed-document-node/index.spec.ts

@@ -21,6 +21,31 @@ schema{
 `;
     const tree = Parser.parse(schema);
     const tdnResult = pluginTypedDocumentNode({ tree });
-    expect(tdnResult.ts).toContain(`TODO`);
+    expect(tdnResult.ts).toContain(`
+import type { TypedDocumentNode } from '@graphql-typed-document-node/core';
+import gql from 'graphql-tag';
+import { GraphQLTypes, InputType, ValueTypes, Zeus } from './index';
+`);
+
+    expect(tdnResult.ts).toContain(`export function tq<Z extends ValueTypes["Query"]>(
+  query: Z | ValueTypes["Query"],
+):  TypedDocumentNode<InputType<GraphQLTypes["Query"], Z>, {}> {
+  return gql(Zeus("query", query));
+}`);
+
+    expect(tdnResult.ts).toContain(`
+export function tm<Z extends ValueTypes["Mutation"]>(
+  mutation: Z | ValueTypes["Mutation"],
+):  TypedDocumentNode<InputType<GraphQLTypes["Mutation"], Z>, {}> {
+  return gql(Zeus("mutation", mutation));
+}
+`);
+    expect(tdnResult.ts).toContain(`
+export function ts<Z extends ValueTypes["Subscription"]>(
+  subscription: Z | ValueTypes["Subscription"],
+):  TypedDocumentNode<InputType<GraphQLTypes["Subscription"], Z>, {}> {
+  return gql(Zeus("subscription", subscription));
+}
+`);
   });
 });