Merge pull request jaswope#4 from waterlink/ekosz/fix/dont-drop-port

[FIX] Don't drop the port of the request

Author: waterlink

lib/rack/reverse_proxy.rb

@@ -89,6 +89,7 @@ def proxy(env, source_request, matcher)
       if response_headers['location'] && options[:replace_response_host]
         response_location = URI(response_headers['location'][0])
         response_location.host = source_request.host
+        response_location.port = source_request.port
         response_headers['location'] = response_location.to_s
       end
 

spec/rack/reverse_proxy_spec.rb

@@ -149,6 +149,12 @@ def app
         # puts last_response.headers.inspect
         last_response.headers['location'].should == "http://example.com/bar"
       end
+
+      it "should keep the port of the location" do
+        stub_request(:get, "http://example.com/test/stuff").to_return(:headers => {"location" => "http://test.com/bar"})
+        get 'http://example.com:3000/test/stuff'
+        last_response.headers['location'].should == "http://example.com:3000/bar"
+      end
     end
 
     describe "with ambiguous routes and all matching" do