Merge pull request #21767 from ronakjangir47/missing_transform_values_test

Added missing tests for transform_values! which returns Enumerator 

1 files changed, 12 insertions, 0 deletions

diff --git a/activesupport/test/core_ext/hash/transform_values_test.rb b/activesupport/test/core_ext/hash/transform_values_test.rb
index 45ed11fef7..7c33227dc0 100644
--- a/activesupport/test/core_ext/hash/transform_values_test.rb
+++ b/activesupport/test/core_ext/hash/transform_values_test.rb
@@ -53,9 +53,21 @@ class TransformValuesTest < ActiveSupport::TestCase
     assert_equal Enumerator, enumerator.class
   end
 
+  test "transform_values! returns an Enumerator if no block is given" do
+    original = { a: 'a', b: 'b' }
+    enumerator = original.transform_values!
+    assert_equal Enumerator, enumerator.class
+  end
+
   test "transform_values is chainable with Enumerable methods" do
     original = { a: 'a', b: 'b' }
     mapped = original.transform_values.with_index { |v, i| [v, i].join }
     assert_equal({ a: 'a0', b: 'b1' }, mapped)
   end
+
+  test "transform_values! is chainable with Enumerable methods" do
+    original = { a: 'a', b: 'b' }
+    original.transform_values!.with_index { |v, i| [v, i].join }
+    assert_equal({ a: 'a0', b: 'b1' }, original)
+  end
 end