First step in adding headers to the response

Author: aesterline

lib/http_client/response.rb

@@ -1,10 +1,15 @@
 module HTTP
   class Response
-    attr_reader :body, :status_code
+    attr_reader :body, :status_code, :headers
 
     def initialize(native_response)
       @status_code = native_response.status_line.status_code
       @body = EntityUtils.to_string(native_response.entity)
+
+      @headers = native_response.all_headers.inject({}) do |map, header|
+        map[header.name] = header.value
+        map
+      end
     end
   end
 

test/http_client/test_client_headers.rb

@@ -1,6 +1,6 @@
 require 'helper'
 
-class TestClientHeaders < Test::Unit::TestCase
+class ClientHeadersTest < Test::Unit::TestCase
   def test_get_headers
     get = HTTP::Get.new("/echo_header")
     get.add_headers(:test_header => "get testing")

test/http_client/test_server_headers.rb

@@ -0,0 +1,18 @@
+require 'helper'
+
+class ServerHeadersTest < Test::Unit::TestCase
+  def test_response_contains_server_headers
+    get = HTTP::Get.new("/set_header")
+    response = @client.execute(get)
+
+    assert_equal("FooBar", response.headers["Test-Header"])
+  end
+
+  def setup
+    @client = HTTP::Client.new(:default_host => "http://localhost:8080")
+  end
+
+  def teardown
+    @client.shutdown
+  end
+end

test/http_test_server.rb

@@ -11,9 +11,16 @@ def self.start_server
       SERVER.mount('/redirect', RedirectServlet)
       SERVER.mount('/set_cookie', SetCookieServlet)
       SERVER.mount('/echo_cookie', EchoCookieServlet)
+      SERVER.mount('/set_header', SetHeaderServlet)
       Thread.new { SERVER.start }
     end
 
+    class SetHeaderServlet < WEBrick::HTTPServlet::AbstractServlet
+      def do_GET(request, response)
+        response["Test-Header"] = "FooBar"
+      end
+    end
+
     class SetCookieServlet < WEBrick::HTTPServlet::AbstractServlet
       def do_GET(request, response)
         response.cookies << WEBrick::Cookie.new("test_cookie", request.query['cookie'])