fix directive description lost in extendSchema

graphql · Jul 21, 2017 · 68cccd3 · 68cccd3
1 parent 6938c1a
commit 68cccd3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/utilities/__tests__/extendSchema-test.js b/src/utilities/__tests__/extendSchema-test.js
@@ -836,6 +836,17 @@ describe('extendSchema', () => {
     expect(newDirective.locations).to.contain('QUERY');
   });
 
+  it('sets correct description when extends with a new directive', () => {
+    const ast = parse(`
+      # new directive
+      directive @new on QUERY
+    `);
+
+    const extendedSchema = extendSchema(testSchema, ast);
+    const newDirective = extendedSchema.getDirective('new');
+    expect(newDirective.description).to.equal('new directive');
+  });
+
   it('may extend directives with new complex directive', () => {
     const ast = parse(`
       directive @profile(enable: Boolean! tag: String) on QUERY | FIELD

diff --git a/src/utilities/extendSchema.js b/src/utilities/extendSchema.js
@@ -539,6 +539,7 @@ export function extendSchema(
   ): GraphQLDirective {
     return new GraphQLDirective({
       name: directiveNode.name.value,
+      description: getDescription(directiveNode),
       locations: directiveNode.locations.map(
         node => ((node.value: any): DirectiveLocationEnum)
       ),