Add more ruby2_keywords

freerange · Aug 25, 2022 · 96e71ad · 96e71ad
1 parent c1d3b2c
commit 96e71ad
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/lib/mocha/invocation.rb b/lib/mocha/invocation.rb
@@ -18,6 +18,7 @@ def initialize(mock, method_name, *arguments, &block)
       @yields = []
       @result = nil
     end
+    ruby2_keywords(:initialize) if RUBY_V3_PLUS
 
     def call(yield_parameters = YieldParameters.new, return_values = ReturnValues.new)
       yield_parameters.next_invocation.each do |yield_args|

diff --git a/lib/mocha/stubbed_method.rb b/lib/mocha/stubbed_method.rb
@@ -59,6 +59,7 @@ def define_new_method
       stub_method_owner.send(:define_method, method_name) do |*args, &block|
         self_in_scope.mock.method_missing(method_name_in_scope, *args, &block)
       end
+      stub_method_owner.send(:ruby2_keywords, method_name) if RUBY_V3_PLUS
       retain_original_visibility(stub_method_owner)
     end