Add Rack::Utils::HeaderHash#replace

Signed-off-by: Joshua Peek <josh@joshpeek.com>

2 files changed, 13 insertions, 0 deletions

diff --git a/lib/rack/utils.rb b/lib/rack/utils.rb
index 228488c1..0899c106 100644
--- a/lib/rack/utils.rb
+++ b/lib/rack/utils.rb
@@ -259,6 +259,12 @@ module Rack
         hash = dup
         hash.merge! other
       end
+
+      def replace(other)
+        clear
+        other.each { |k, v| self[k] = v }
+        self
+      end
     end
 
     # Every standard HTTP code mapped to the appropriate message.
diff --git a/test/spec_rack_utils.rb b/test/spec_rack_utils.rb
index 077ccc50..36b4e7b4 100644
--- a/test/spec_rack_utils.rb
+++ b/test/spec_rack_utils.rb
@@ -228,6 +228,13 @@ context "Rack::Utils::HeaderHash" do
     h = Rack::Utils::HeaderHash.new("foo" => ["bar", "baz"])
     h.to_hash.should.equal({ "foo" => "bar\nbaz" })
   end
+
+  specify "should replace hashes correctly" do
+    h = Rack::Utils::HeaderHash.new("Foo-Bar" => "baz")
+    j = {"foo" => "bar"}
+    h.replace(j)
+    h["foo"].should.equal "bar"
+  end
 end
 
 context "Rack::Utils::Context" do