From ff891c0aec6cf9b15364894781deb3c7315fadaa Mon Sep 17 00:00:00 2001
From: Frederick Cheung <frederick.cheung@gmail.com>
Date: Thu, 11 Dec 2008 15:19:03 +0000
Subject: [PATCH] Make delete_if/reject faster and fix other mutators

---
 activesupport/lib/active_support/ordered_hash.rb |   39 ++++++++++++++--------
 activesupport/test/ordered_hash_test.rb          |   37 ++++++++++++++++++++
 2 files changed, 62 insertions(+), 14 deletions(-)

diff --git a/activesupport/lib/active_support/ordered_hash.rb b/activesupport/lib/active_support/ordered_hash.rb
index fa68db5..6179c36 100644
--- a/activesupport/lib/active_support/ordered_hash.rb
+++ b/activesupport/lib/active_support/ordered_hash.rb
@@ -18,19 +18,13 @@ module ActiveSupport
       end
 
       def delete(key)
-        array_index = has_key?(key) && index(key)
-        if array_index
-          @keys.delete_at(array_index)
+        if has_key? key
+          index = @keys.index(key)
+          @keys.delete_at index
         end
         super
       end
-
-      def delete_if
-        super
-        sync_keys!
-        self
-      end
-
+             
       def reject!
         super
         sync_keys!
@@ -41,9 +35,6 @@ module ActiveSupport
         dup.reject!(&block)
       end
 
-      alias_method :super_keys, :keys
-      private :super_keys
-
       def keys
         @keys
       end
@@ -67,11 +58,31 @@ module ActiveSupport
       def each
         keys.each {|key| yield [key, self[key]]}
       end
+      
+      alias_method :each_pair, :each
 
+      def clear
+        super
+        @keys.clear
+        self
+      end
+      
+      def shift
+        k = @keys.first
+        v = delete(k)
+        [k, v]
+      end
+      
+      def merge(other_hash)
+        result = dup
+        other_hash.each {|k,v| result[k]=v}
+        result
+      end
+      
       private
 
       def sync_keys!
-        (@keys - super_keys).each { |k| @keys.delete(k) }
+        @keys.delete_if {|k| !has_key?(k)}
       end
     end
   end
diff --git a/activesupport/test/ordered_hash_test.rb b/activesupport/test/ordered_hash_test.rb
index ca5fce6..f0b7670 100644
--- a/activesupport/test/ordered_hash_test.rb
+++ b/activesupport/test/ordered_hash_test.rb
@@ -36,9 +36,11 @@ class OrderedHashTest < Test::Unit::TestCase
 
     @ordered_hash[key] = value
     assert_equal @keys.length + 1, @ordered_hash.length
+    assert_equal @ordered_hash.keys.length, @ordered_hash.length
 
     assert_equal value, @ordered_hash.delete(key)
     assert_equal @keys.length, @ordered_hash.length
+    assert_equal @ordered_hash.keys.length, @ordered_hash.length
 
     assert_nil @ordered_hash.delete(bad_key)
   end
@@ -83,6 +85,17 @@ class OrderedHashTest < Test::Unit::TestCase
   def test_each_with_index
     @ordered_hash.each_with_index { |pair, index| assert_equal [@keys[index], @values[index]], pair}
   end
+  
+  def test_each_pair
+    values = []
+    keys = []
+    @ordered_hash.each_pair do |key, value| 
+      keys << key
+      values << value
+    end
+    assert_equal @values, values
+    assert_equal @keys, keys
+  end
 
   def test_delete_if
     (copy = @ordered_hash.dup).delete('pink')
@@ -103,4 +116,28 @@ class OrderedHashTest < Test::Unit::TestCase
     assert_equal copy, @ordered_hash
     assert !new_ordered_hash.keys.include?('pink')
   end
+  
+  def test_clear
+    @ordered_hash.clear
+    assert_equal [], @ordered_hash.keys
+  end
+  
+  def test_merge
+    other_hash =  ActiveSupport::OrderedHash.new
+    other_hash['purple'] = '800080'
+    other_hash['violet'] = 'ee82ee'
+    merged = @ordered_hash.merge other_hash
+    assert_equal merged.length, @ordered_hash.length + other_hash.length
+    assert_equal @keys + ['purple', 'violet'], merged.keys
+    
+    @ordered_hash.merge! other_hash
+    assert_equal @ordered_hash, merged
+    assert_equal @ordered_hash.keys, merged.keys
+  end
+  
+  def test_shift
+    pair = @ordered_hash.shift
+    assert_equal [@keys.first, @values.first], pair
+    assert !@ordered_hash.keys.include?(pair.first)
+  end
 end
\ No newline at end of file
-- 
1.6.0.1