fixing tests

andimarek · andimarek · commit e60fb5af3609 · 2020-01-24T12:28:39.000+11:00
diff --git a/src/test/groovy/graphql/schema/idl/SchemaPrinterTest.groovy b/src/test/groovy/graphql/schema/idl/SchemaPrinterTest.groovy
@@ -801,6 +801,23 @@ type Query {
 
     def idlWithDirectives() {
         return """
+            directive @interfaceFieldDirective on FIELD_DEFINITION
+            directive @unionTypeDirective on UNION
+            directive @query1 on OBJECT
+            directive @query2(arg1: String) on OBJECT
+            directive @fieldDirective1 on FIELD_DEFINITION
+            directive @fieldDirective2(argStr: String, argInt: Int, argFloat: Float, argBool: Boolean) on FIELD_DEFINITION
+            directive @argDirective on ARGUMENT_DEFINITION
+            directive @interfaceImplementingTypeDirective on OBJECT
+            directive @enumTypeDirective on ENUM
+            directive @single on OBJECT
+            directive @singleField on FIELD_DEFINITION
+            directive @interfaceImplementingFieldDirective on FIELD_DEFINITION
+            directive @enumValueDirective on ENUM_VALUE
+            directive @inputTypeDirective on INPUT_OBJECT
+            directive @inputFieldDirective on INPUT_FIELD_DEFINITION
+            directive @interfaceTypeDirective on INTERFACE
+            directive @scalarDirective on SCALAR
             
             interface SomeInterface @interfaceTypeDirective {
                 fieldA : String @interfaceFieldDirective
@@ -866,6 +883,40 @@ directive @skip(
     if: Boolean!
   ) on FIELD | FRAGMENT_SPREAD | INLINE_FRAGMENT
 
+directive @interfaceFieldDirective on FIELD_DEFINITION
+
+directive @unionTypeDirective on UNION
+
+directive @query1 on OBJECT
+
+directive @query2(arg1: String) on OBJECT
+
+directive @fieldDirective1 on FIELD_DEFINITION
+
+directive @fieldDirective2(argBool: Boolean, argFloat: Float, argInt: Int, argStr: String) on FIELD_DEFINITION
+
+directive @argDirective on ARGUMENT_DEFINITION
+
+directive @interfaceImplementingTypeDirective on OBJECT
+
+directive @enumTypeDirective on ENUM
+
+directive @single on OBJECT
+
+directive @singleField on FIELD_DEFINITION
+
+directive @interfaceImplementingFieldDirective on FIELD_DEFINITION
+
+directive @enumValueDirective on ENUM_VALUE
+
+directive @inputTypeDirective on INPUT_OBJECT
+
+directive @inputFieldDirective on INPUT_FIELD_DEFINITION
+
+directive @interfaceTypeDirective on INTERFACE
+
+directive @scalarDirective on SCALAR
+
 "Marks the field or enum value as deprecated"
 directive @deprecated(
     "The reason for the deprecation"
@@ -1088,6 +1139,7 @@ type Query {
 
     def "can print a schema as AST elements"() {
         def sdl = '''
+            directive @directive1 on SCALAR
             type Query {
                 foo : String
             }
@@ -1197,6 +1249,8 @@ directive @skip(
     if: Boolean!
   ) on FIELD | FRAGMENT_SPREAD | INLINE_FRAGMENT
 
+directive @directive1 on SCALAR
+
 "Marks the field or enum value as deprecated"
 directive @deprecated(
     "The reason for the deprecation"
diff --git a/src/test/groovy/graphql/schema/transform/FieldVisibilitySchemaTransformationTest.groovy b/src/test/groovy/graphql/schema/transform/FieldVisibilitySchemaTransformationTest.groovy
@@ -736,7 +736,7 @@ class FieldVisibilitySchemaTransformationTest extends Specification {
         given:
         GraphQLSchema schema = TestUtil.schema("""
 
-        directive @private on FIELD_DEFINITION
+        directive @private on FIELD_DEFINITION | INPUT_FIELD_DEFINITION
         
         type Query {
             foo: Foo 
@@ -774,7 +774,7 @@ class FieldVisibilitySchemaTransformationTest extends Specification {
         given:
         GraphQLSchema schema = TestUtil.schema("""
 
-        directive @private on FIELD_DEFINITION
+        directive @private on FIELD_DEFINITION | INPUT_FIELD_DEFINITION
         
         type Query {
             foo: Foo 
