spec/graphql/schema/object_spec.rb in graphql-1.9.4 vs spec/graphql/schema/object_spec.rb in graphql-1.9.5

@@ -103,9 +103,80 @@
       new_method_defs = Hash[methods.zip(methods.map{|method| object_type.method(method.to_sym)})]
       assert_equal method_defs, new_method_defs
     end
   end
 
+  describe "using GraphQL::Function" do
+    new_test_func_payload = Class.new(GraphQL::Schema::Object) do
+      graphql_name "TestFuncPayload"
+      field :name, String, null: false
+    end
+
+    it "returns data on a field" do
+      new_func_class = Class.new(GraphQL::Function) do
+        argument :name, GraphQL::STRING_TYPE
+        type new_test_func_payload
+
+        def call(o, a, c)
+          { name: a[:name] }
+        end
+      end
+
+      new_object_class = Class.new(GraphQL::Schema::Object) do
+        graphql_name "GraphQL"
+        field :test, function: new_func_class.new
+      end
+
+      schema = Class.new(GraphQL::Schema) do
+        query(new_object_class)
+      end
+
+      query_str = <<-GRAPHQL
+      {
+        test(name: "graphql") {
+          name
+        }
+      }
+      GRAPHQL
+      res = schema.execute(query_str)
+      assert_equal "graphql", res["data"]["test"]["name"]
+    end
+
+    it "returns data on a connection" do
+      new_func_class = Class.new(GraphQL::Function) do
+        argument :name, GraphQL::STRING_TYPE
+        type new_test_func_payload.connection_type
+
+        def call(o, a, c)
+          [{ name: a[:name] }]
+        end
+      end
+
+      new_object_class = Class.new(GraphQL::Schema::Object) do
+        graphql_name "GraphQL"
+        field :test_conn, function: new_func_class.new
+      end
+
+      schema = Class.new(GraphQL::Schema) do
+        query(new_object_class)
+      end
+
+      query_str = <<-GRAPHQL
+      {
+        testConn(name: "graphql") {
+          edges {
+            node {
+              name
+            }
+          }
+        }
+      }
+      GRAPHQL
+      res = schema.execute(query_str)
+      assert_equal "graphql", res["data"]["testConn"]["edges"][0]["node"]["name"]
+    end
+  end
+
   describe "wrapping a Hash" do
     it "automatically looks up symbol and string keys" do
       query_str = <<-GRAPHQL
       {
         hashyEnsemble {