Skip complexity calculation by directiveEstimator when astNode is undefined

Author: hnq90

src/__tests__/QueryComplexity-test.ts

@@ -17,6 +17,7 @@ import schema from './fixtures/schema';
 import ComplexityVisitor, {getComplexity} from '../QueryComplexity';
 import {
   simpleEstimator,
+  directiveEstimator,
   fieldConfigEstimator,
 } from '../index';
 
@@ -321,4 +322,45 @@ describe('QueryComplexity analysis', () => {
       'At least one complexity estimator has to return a complexity score.'
     );
   });
+
+  it.only('should return NaN when no astNode available on field when use directiveEstimator', () => {
+    const ast = parse(`
+      query {
+        _service {
+          sdl
+        }
+      }
+    `);
+
+    const complexity = getComplexity({
+      estimators: [
+        directiveEstimator(),
+      ],
+      schema,
+      query: ast
+    });
+    expect(Number.isNaN(complexity)).to.equal(true);
+  });
+  
+  it.only('should skip complexity calculation by directiveEstimator when no astNode available on field', () => {
+    const ast = parse(`
+      query {
+        _service {
+          sdl
+        }
+      }
+    `);
+
+    const complexity = getComplexity({
+      estimators: [
+        directiveEstimator(),
+        simpleEstimator({
+          defaultComplexity: 1
+        })
+      ],
+      schema,
+      query: ast
+    });
+    expect(complexity).to.equal(2);
+  });
 });

src/__tests__/fixtures/schema.ts

@@ -83,6 +83,14 @@ const Union = new GraphQLUnionType({
   resolveType: () => Item
 });
 
+const SDLInterface = new GraphQLInterfaceType({
+  name: 'SDLInterface',
+  fields: {
+    sdl: { type: GraphQLString }
+  },
+  resolveType: () => '"SDL"'
+});
+
 const Query = new GraphQLObjectType({
   name: 'Query',
   fields: () => ({
@@ -126,7 +134,8 @@ const Query = new GraphQLObjectType({
           type: new GraphQLNonNull(GraphQLInt)
         }
       }
-    }
+    },
+    _service: {type: SDLInterface},
   }),
 });
 

src/estimators/directive/index.ts

@@ -28,6 +28,11 @@ export default function (options?: {}): ComplexityEstimator {
   });
 
   return (args: ComplexityEstimatorArgs) => {
+    // Ignore if astNode is undefined
+    if (!args.field.astNode) {
+      return;
+    }
+
     const values = getDirectiveValues(directive, args.field.astNode);
 
     // Ignore if no directive set