fix test

Author: KaiyueGuan

lib/__tests__.js

@@ -3,8 +3,9 @@
 const Test = require('tape');
 const graphql = require('graphql');
 const gql = require('graphql-tag');
-const { SchemaDirectiveVisitor } = require('@graphql-tools/utils');
+const { SchemaDirectiveVisitor, MapperKind, mapSchema, getDirective} = require('@graphql-tools/utils');
 const GraphQLComponent = require('.');
+const {defaultFieldResolver} = require("graphql/index");
 
 Test('GraphQLComponent instance API (getters/setters)', (t) => {
 
@@ -1109,11 +1110,17 @@ Test(`importing root specifies 'federation: true' results in all components crea
 })
 
 Test('federated schema can include custom directive', (t) => {
-  class CustomDirective extends SchemaDirectiveVisitor {
-    // required for our dummy "custom" directive (ie. implement the SchemaDirectiveVisitor interface)
-    visitFieldDefinition() {
-      return;
-    }
+  function customDirective(directiveName) {
+    return (schema) => mapSchema(schema, {
+      [MapperKind.OBJECT_FIELD]: (fieldConfig) => {
+        const directive = getDirective(schema, fieldConfig, directiveName)?.[0]
+        if (directive) {
+          return {
+            ...fieldConfig,
+          }
+        }
+      }
+    })
   }
 
   const component = new GraphQLComponent({
@@ -1142,7 +1149,7 @@ Test('federated schema can include custom directive', (t) => {
         }
       },
     },
-    directives: { custom: CustomDirective },
+    directives: { custom: customDirective },
     federation: true
   });
 
@@ -1163,7 +1170,7 @@ Test('federated schema can include custom directive', (t) => {
     t.plan(2);
     const { schema: { _typeMap: { Extended } } } = component;
     t.equals(Extended.extensionASTNodes.length, 1, 'Extension AST Nodes is defined');
-    t.equals(Extended.extensionASTNodes[0].fields.filter((field) => field.name.value === "id" && field.directives[0].name.value === "external").length, 1, `id field marked external`);
+    t.equals(Extended.astNode.fields.filter((field) => field.name.value === "id" && field.directives[0].name.value === "external").length, 1, `id field marked external`);
   });
 });