Merge pull request #3965 from rmosolgo/mutation-type-fix

Fix mutation type(...) config

Author: rmosolgo

lib/graphql/schema/resolver/has_payload_type.rb

@@ -20,7 +20,17 @@ def payload_type(new_payload_type = nil)
           @payload_type ||= generate_payload_type
         end
 
-        # alias :type :payload_type
+        def type(new_type = nil, null: nil)
+          if new_type
+            payload_type(new_type)
+            if !null.nil?
+              self.null(null)
+            end
+          else
+            super()
+          end
+        end
+
         alias :type_expr :payload_type
 
         def field_class(new_class = nil)

spec/graphql/schema/mutation_spec.rb

@@ -215,4 +215,15 @@ class Mutation < GraphQL::Schema::Object
     result = InterfaceMutationSchema.execute("mutation { signIn(login: \"abc\", password: \"abc\") { success } }")
     assert_equal true, result["data"]["signIn"]["success"]
   end
+
+  it "returns manually-configured return types" do
+    mutation = Class.new(GraphQL::Schema::Mutation) do
+      graphql_name "DoStuff"
+      type(String)
+    end
+
+    field = GraphQL::Schema::Field.new(name: "f", owner: nil, resolver_class: mutation)
+    assert_equal "String", field.type.graphql_name
+    assert_equal GraphQL::Types::String, field.type
+  end
 end