Merge pull request #21806 from kachick/fix-transformer-size

Return a sized Enumerator from Hash#transform_values{!}

2 files changed, 8 insertions, 4 deletions

diff --git a/activesupport/test/core_ext/hash/transform_keys_test.rb b/activesupport/test/core_ext/hash/transform_keys_test.rb
index 5a0b99e22c..99af274614 100644
--- a/activesupport/test/core_ext/hash/transform_keys_test.rb
+++ b/activesupport/test/core_ext/hash/transform_keys_test.rb
@@ -18,15 +18,17 @@ class TransformKeysTest < ActiveSupport::TestCase
     assert_same original, mapped
   end
 
-  test "transform_keys returns an Enumerator if no block is given" do
+  test "transform_keys returns a sized Enumerator if no block is given" do
     original = { a: 'a', b: 'b' }
     enumerator = original.transform_keys
+    assert_equal original.size, enumerator.size
     assert_equal Enumerator, enumerator.class
   end
 
-  test "transform_keys! returns an Enumerator if no block is given" do
+  test "transform_keys! returns a sized Enumerator if no block is given" do
     original = { a: 'a', b: 'b' }
     enumerator = original.transform_keys!
+    assert_equal original.size, enumerator.size
     assert_equal Enumerator, enumerator.class
   end
 
diff --git a/activesupport/test/core_ext/hash/transform_values_test.rb b/activesupport/test/core_ext/hash/transform_values_test.rb
index 7c33227dc0..114022fbaf 100644
--- a/activesupport/test/core_ext/hash/transform_values_test.rb
+++ b/activesupport/test/core_ext/hash/transform_values_test.rb
@@ -47,15 +47,17 @@ class TransformValuesTest < ActiveSupport::TestCase
     assert_nil mapped[:b]
   end
 
-  test "transform_values returns an Enumerator if no block is given" do
+  test "transform_values returns a sized Enumerator if no block is given" do
     original = { a: 'a', b: 'b' }
     enumerator = original.transform_values
+    assert_equal original.size, enumerator.size
     assert_equal Enumerator, enumerator.class
   end
 
-  test "transform_values! returns an Enumerator if no block is given" do
+  test "transform_values! returns a sized Enumerator if no block is given" do
     original = { a: 'a', b: 'b' }
     enumerator = original.transform_values!
+    assert_equal original.size, enumerator.size
     assert_equal Enumerator, enumerator.class
   end