Add dataloader uses to context spec

Author: rmosolgo

spec/graphql/query/context_spec.rb

@@ -153,13 +153,15 @@ def value
       end
     end
 
-    class ContextKeySource < GraphQL::Dataloader::Source
-      def initialize(context)
-        @context = context
+    class PassthroughSource < GraphQL::Dataloader::Source
+      def fetch(keys)
+        keys
       end
+    end
 
+    class IntArraySource < GraphQL::Dataloader::Source
       def fetch(keys)
-        keys.map { |k| @context[k] }
+        keys.map { |k| k.times.map { |i| i } }
       end
     end
 
@@ -193,14 +195,15 @@ def set_scoped_context(key:, value:, lazy:)
           }
         else
           context.scoped_merge!(key => value)
-          self
+          context.dataloader.with(PassthroughSource).load(self)
         end
       end
 
       field :int_list, [ContextQuery], null: false
 
       def int_list
-        [1, 2, 3, 4]
+        context.scoped_set!("int_list", "assigned")
+        context.dataloader.with(IntArraySource).load(4)
       end
 
       field :set_scoped_int, ContextQuery, null: false
@@ -358,17 +361,18 @@ class ContextSchema < GraphQL::Schema
           before: getScopedContext(key: "int")
           setScopedInt {
             inside: getScopedContext(key: "int")
+            inside2: getScopedContext(key: "int_list")
           }
           after: getScopedContext(key: "int")
         }
       }
       GRAPHQL
 
       expected_data = {"intList"=>
-         [{"setScopedInt"=>{"inside"=>"1"}, "before"=>nil, "after"=>nil},
-          {"setScopedInt"=>{"inside"=>"2"}, "before"=>nil, "after"=>nil},
-          {"setScopedInt"=>{"inside"=>"3"}, "before"=>nil, "after"=>nil},
-          {"setScopedInt"=>{"inside"=>"4"}, "before"=>nil, "after"=>nil}]}
+         [{"setScopedInt"=>{"inside"=>"0", "inside2" => "assigned"}, "before"=>nil, "after"=>nil},
+          {"setScopedInt"=>{"inside"=>"1", "inside2" => "assigned"}, "before"=>nil, "after"=>nil},
+          {"setScopedInt"=>{"inside"=>"2", "inside2" => "assigned"}, "before"=>nil, "after"=>nil},
+          {"setScopedInt"=>{"inside"=>"3", "inside2" => "assigned"}, "before"=>nil, "after"=>nil}]}
       result = ContextSchema.execute(query_str)
       assert_equal(expected_data, result["data"])
     end