Add Rack::Utils::HeaderHash#replace

Signed-off-by: Joshua Peek <josh@joshpeek.com>
rack · Jun 15, 2009 · e0322ed · e0322ed · rtomayko · Jun 15, 2009
1 parent 5ca8f82
commit e0322ed
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/lib/rack/utils.rb b/lib/rack/utils.rb
@@ -259,6 +259,12 @@ def merge(other)
         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
@@ -228,6 +228,13 @@
     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