Add support for extend input to parser

Author: cjoudrey

lib/graphql/compatibility/schema_parser_specification.rb

@@ -533,6 +533,63 @@ def test_it_parses_enum_type_extension_with_directives
             assert_equal 0, enum_type_extension.values.length
           end
 
+          def test_it_parses_input_object_type_extension_with_fields
+            document = parse('
+              extend input UserInput {
+                login: String!
+              }
+            ')
+
+            input_object_type_extension = document.definitions.first
+            assert_equal GraphQL::Language::Nodes::InputObjectTypeExtension, input_object_type_extension.class
+            assert_equal 'UserInput', input_object_type_extension.name
+            assert_equal [2, 15], input_object_type_extension.position
+
+            assert_equal 1, input_object_type_extension.fields.length
+            assert_equal GraphQL::Language::Nodes::InputValueDefinition, input_object_type_extension.fields.first.class
+            assert_equal 'login', input_object_type_extension.fields.first.name
+
+            assert_equal 0, input_object_type_extension.directives.length
+          end
+
+          def test_it_parses_input_object_type_extension_with_directives_and_fields
+            document = parse('
+              extend input UserInput @deprecated {
+                login: String!
+              }
+            ')
+
+            input_object_type_extension = document.definitions.first
+            assert_equal GraphQL::Language::Nodes::InputObjectTypeExtension, input_object_type_extension.class
+            assert_equal 'UserInput', input_object_type_extension.name
+            assert_equal [2, 15], input_object_type_extension.position
+
+            assert_equal 1, input_object_type_extension.fields.length
+            assert_equal GraphQL::Language::Nodes::InputValueDefinition, input_object_type_extension.fields.first.class
+            assert_equal 'login', input_object_type_extension.fields.first.name
+
+            assert_equal 1, input_object_type_extension.directives.length
+            assert_equal GraphQL::Language::Nodes::Directive, input_object_type_extension.directives.first.class
+            assert_equal 'deprecated', input_object_type_extension.directives.first.name
+          end
+
+          def test_it_parses_input_object_type_extension_with_directives
+            document = parse('
+              extend input UserInput @deprecated
+            ')
+
+            input_object_type_extension = document.definitions.first
+            assert_equal GraphQL::Language::Nodes::InputObjectTypeExtension, input_object_type_extension.class
+            assert_equal 'UserInput', input_object_type_extension.name
+            assert_equal [2, 15], input_object_type_extension.position
+
+            assert_equal 0, input_object_type_extension.fields.length
+
+            assert_equal 1, input_object_type_extension.directives.length
+            assert_equal GraphQL::Language::Nodes::Directive, input_object_type_extension.directives.first.class
+            assert_equal 'deprecated', input_object_type_extension.directives.first.name
+          end
+
           def test_it_parses_whole_definition_with_descriptions
             document = parse(SCHEMA_DEFINITION_STRING)
 

lib/graphql/language/nodes.rb

@@ -571,6 +571,8 @@ def children
           fields + directives
         end
       end
+
+      class InputObjectTypeExtension < InputObjectTypeDefinition; end;
     end
   end
 end

lib/graphql/language/parser.y

@@ -311,6 +311,7 @@ rule
     | interface_type_extension
     | union_type_extension
     | enum_type_extension
+    | input_object_type_extension
 
   scalar_type_extension: EXTEND SCALAR name directives_list { return make_node(:ScalarTypeExtension, name: val[2], directives: val[3], position_source: val[0]) }
 
@@ -333,6 +334,10 @@ rule
       EXTEND ENUM name directives_list_opt LCURLY enum_value_definitions RCURLY { return make_node(:EnumTypeExtension, name: val[2], directives: val[3], values: val[5], position_source: val[0]) }
     | EXTEND ENUM name directives_list { return make_node(:EnumTypeExtension, name: val[2], directives: val[3], values: [], position_source: val[0]) }
 
+  input_object_type_extension:
+      EXTEND INPUT name directives_list_opt LCURLY input_value_definition_list RCURLY { return make_node(:InputObjectTypeExtension, name: val[2], directives: val[3], fields: val[5], position_source: val[0]) }
+    | EXTEND INPUT name directives_list { return make_node(:InputObjectTypeExtension, name: val[2], directives: val[3], fields: [], position_source: val[0]) }
+
   scalar_type_definition: SCALAR name directives_list_opt { return make_node(:ScalarTypeDefinition, name: val[1], directives: val[2], description: get_description(val[0]), position_source: val[0]) }
 
   object_type_definition: