- Clarified some doco wrt the block on #stub.

[git-p4: depot-paths = "//src/minitest/dev/": change = 13444]
minitest · Jun 19, 2022 · c748ce9 · c748ce9
1 parent 61345d7
commit c748ce9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/minitest/mock.rb b/lib/minitest/mock.rb
@@ -267,7 +267,7 @@ class Object
   # NOTE: keyword args in callables are NOT checked for correctness
   # against the existing method. Too many edge cases to be worth it.
 
-  def stub name, val_or_callable, *block_args, **block_kwargs
+  def stub name, val_or_callable, *block_args, **block_kwargs, &block
     new_name = "__minitest_stub__#{name}"
 
     metaclass = class << self; self; end
@@ -299,7 +299,7 @@ def stub name, val_or_callable, *block_args, **block_kwargs
       end
     end
 
-    yield self
+    block[self]
   ensure
     metaclass.send :undef_method, name
     metaclass.send :alias_method, name, new_name