Maintain subclasses of Hash when calling Hash#slice.

ruby-i18n · Apr 12, 2014 · 08293a4 · 08293a4
1 parent 8fc94e7
commit 08293a4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/lib/i18n/core_ext/hash.rb b/lib/i18n/core_ext/hash.rb
@@ -1,6 +1,6 @@
 class Hash
   def slice(*keep_keys)
-    h = {}
+    h = self.class.new
     keep_keys.each { |key| h[key] = fetch(key) }
     h
   end unless Hash.method_defined?(:slice)

diff --git a/test/core_ext/hash_test.rb b/test/core_ext/hash_test.rb
@@ -14,6 +14,12 @@ class I18nCoreExtHashInterpolationTest < Test::Unit::TestCase
     assert_equal expected, hash.slice(:foo)
   end
 
+  test "#slice maintains subclasses of Hash" do
+    klass = Class.new(Hash)
+    hash = klass[:foo, 'bar', :baz, 'bar']
+    assert_instance_of klass,  hash.slice(:foo)
+  end
+
   test "#except" do
     hash = { :foo => 'bar',  :baz => 'bar' }
     expected = { :foo => 'bar' }