implicitly add defer directive to schema

Author: lilianammmatos

src/utilities/__tests__/buildASTSchema-test.js

@@ -15,6 +15,7 @@ import {
   assertDirective,
   GraphQLSkipDirective,
   GraphQLIncludeDirective,
+  GraphQLDeferDirective,
   GraphQLDeprecatedDirective,
 } from '../../type/directives';
 import {
@@ -182,12 +183,13 @@ describe('Schema Builder', () => {
     expect(cycleSDL(sdl, { commentDescriptions: true })).to.equal(sdl);
   });
 
-  it('Maintains @skip & @include', () => {
+  it('Maintains @skip, @include & @defer', () => {
     const schema = buildSchema('type Query');
 
-    expect(schema.getDirectives()).to.have.lengthOf(3);
+    expect(schema.getDirectives()).to.have.lengthOf(4);
     expect(schema.getDirective('skip')).to.equal(GraphQLSkipDirective);
     expect(schema.getDirective('include')).to.equal(GraphQLIncludeDirective);
+    expect(schema.getDirective('defer')).to.equal(GraphQLDeferDirective);
     expect(schema.getDirective('deprecated')).to.equal(
       GraphQLDeprecatedDirective,
     );
@@ -197,27 +199,30 @@ describe('Schema Builder', () => {
     const schema = buildSchema(`
       directive @skip on FIELD
       directive @include on FIELD
+      directive @defer on FIELD
       directive @deprecated on FIELD_DEFINITION
     `);
 
-    expect(schema.getDirectives()).to.have.lengthOf(3);
+    expect(schema.getDirectives()).to.have.lengthOf(4);
     expect(schema.getDirective('skip')).to.not.equal(GraphQLSkipDirective);
     expect(schema.getDirective('include')).to.not.equal(
       GraphQLIncludeDirective,
     );
+    expect(schema.getDirective('defer')).to.not.equal(GraphQLDeferDirective);
     expect(schema.getDirective('deprecated')).to.not.equal(
       GraphQLDeprecatedDirective,
     );
   });
 
-  it('Adding directives maintains @skip & @include', () => {
+  it('Adding directives maintains @skip, @include & @defer', () => {
     const schema = buildSchema(`
       directive @foo(arg: Int) on FIELD
     `);
 
-    expect(schema.getDirectives()).to.have.lengthOf(4);
+    expect(schema.getDirectives()).to.have.lengthOf(5);
     expect(schema.getDirective('skip')).to.not.equal(undefined);
     expect(schema.getDirective('include')).to.not.equal(undefined);
+    expect(schema.getDirective('defer')).to.not.equal(undefined);
     expect(schema.getDirective('deprecated')).to.not.equal(undefined);
   });
 

src/utilities/buildASTSchema.js

@@ -54,6 +54,7 @@ import {
   GraphQLDirective,
   GraphQLSkipDirective,
   GraphQLIncludeDirective,
+  GraphQLDeferDirective,
   GraphQLDeprecatedDirective,
 } from '../type/directives';
 import {
@@ -170,6 +171,10 @@ export function buildASTSchema(
     directives.push(GraphQLIncludeDirective);
   }
 
+  if (!directives.some(directive => directive.name === 'defer')) {
+    directives.push(GraphQLDeferDirective);
+  }
+
   if (!directives.some(directive => directive.name === 'deprecated')) {
     directives.push(GraphQLDeprecatedDirective);
   }