Adds Context#scoped_set

Author: lancelafontaine

lib/graphql/query/context.rb

@@ -215,6 +215,10 @@ def scoped_merge!(hash)
         @scoped_context = @scoped_context.merge(hash)
       end
 
+      def scoped_set!(key, value)
+        scoped_merge!(key => value)
+      end
+
       class FieldResolutionContext
         include SharedMethods
         include Tracing::Traceable

spec/graphql/query/context_spec.rb

@@ -506,5 +506,16 @@ class ContextSchema < GraphQL::Schema
       assert_equal(expected_value, context.fetch(expected_key))
       assert_equal(expected_value, context.dig(expected_key)) if RUBY_VERSION >= '2.3.0'
     end
+
+    it "sets a value using #scoped_set!" do
+      expected_key = :a
+      expected_value = :test
+
+      context = GraphQL::Query::Context.new(query: OpenStruct.new(schema: schema), values: nil, object: nil)
+      assert_nil(context[expected_key])
+
+      context.scoped_set!(expected_key, expected_value)
+      assert_equal(expected_value, context[expected_key])
+    end
   end
 end