add build schema test

graphql · Jan 9, 2020 · fc70b69 · fc70b69
1 parent 428edc5
commit fc70b69
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/utilities/__tests__/buildASTSchema-test.js b/src/utilities/__tests__/buildASTSchema-test.js
@@ -16,6 +16,8 @@ import {
   GraphQLSkipDirective,
   GraphQLIncludeDirective,
   GraphQLDeprecatedDirective,
+  GraphQLStreamDirective,
+  GraphQLDeferDirective,
 } from '../../type/directives';
 import {
   GraphQLID,
@@ -247,6 +249,22 @@ describe('Schema Builder', () => {
     expect(schema.getDirective('deprecated')).to.not.equal(undefined);
   });
 
+  it('Adds @stream and @defer when experimental flags are passed to schema', () => {
+    const schema = buildSchema('type Query', {
+      experimentalStream: true,
+      experimentalDefer: true,
+    });
+
+    expect(schema.getDirectives()).to.have.lengthOf(5);
+    expect(schema.getDirective('stream')).to.equal(GraphQLStreamDirective);
+    expect(schema.getDirective('defer')).to.equal(GraphQLDeferDirective);
+    expect(schema.getDirective('skip')).to.equal(GraphQLSkipDirective);
+    expect(schema.getDirective('include')).to.equal(GraphQLIncludeDirective);
+    expect(schema.getDirective('deprecated')).to.equal(
+      GraphQLDeprecatedDirective,
+    );
+  });
+
   it('Type modifiers', () => {
     const sdl = dedent`
       type Query {

diff --git a/src/utilities/buildASTSchema.js b/src/utilities/buildASTSchema.js
@@ -154,5 +154,7 @@ export function buildSchema(
     commentDescriptions: (options && options.commentDescriptions) || false,
     assumeValidSDL: (options && options.assumeValidSDL) || false,
     assumeValid: (options && options.assumeValid) || false,
+    experimentalDefer: (options && options.experimentalDefer) || false,
+    experimentalStream: (options && options.experimentalStream) || false,
   });
 }