introspection for repeatable directive info (#2113)

dugenkui03 · web-flow · commit 43668fb09ffa · 2020-11-25T09:04:36.000+11:00
diff --git a/src/main/java/graphql/introspection/Introspection.java b/src/main/java/graphql/introspection/Introspection.java
@@ -474,6 +474,10 @@ public enum DirectiveLocation {
         });
         register(__Directive, "name", nameDataFetcher);
         register(__Directive, "description", descriptionDataFetcher);
+        register(__Directive, "isRepeatable", environment -> {
+            GraphQLDirective directive = environment.getSource();
+            return directive.isRepeatable();
+        });
     }
 
     public static final GraphQLObjectType __Schema = newObject()
diff --git a/src/main/java/graphql/introspection/IntrospectionQuery.java b/src/main/java/graphql/introspection/IntrospectionQuery.java
@@ -21,6 +21,7 @@ public interface IntrospectionQuery {
             "        args {\n" +
             "          ...InputValue\n" +
             "        }\n" +
+            "        isRepeatable\n" +
             "      }\n" +
             "    }\n" +
             "  }\n" +
diff --git a/src/test/groovy/graphql/introspection/IntrospectionTest.groovy b/src/test/groovy/graphql/introspection/IntrospectionTest.groovy
@@ -81,4 +81,26 @@ class IntrospectionTest extends Specification {
         data == [__schema: [description: "This is my schema"]]
 
     }
+
+    def "introspection for repeatable directive info"() {
+        def spec = '''
+            directive @repeatableDirective(arg: String) repeatable on FIELD
+             
+            type Query {
+               namedField: String
+            }
+        '''
+
+        when:
+        def graphQL = TestUtil.graphQL(spec).build()
+        def executionResult = graphQL.execute(IntrospectionQuery.INTROSPECTION_QUERY)
+
+        then:
+        executionResult.errors.isEmpty()
+
+        def directives = executionResult.data.getAt("__schema").getAt("directives") as List
+        def geoPolygonType = directives.find { it['name'] == 'repeatableDirective' }
+        geoPolygonType["isRepeatable"] == true
+    }
+
 }
