Tests for X-Forwarded-Host code

Author: jjthrash

lib/rack/reverse_proxy.rb

@@ -25,7 +25,7 @@ def call(env)
       }
       headers['HOST'] = uri.host if all_opts[:preserve_host]
       headers['X-Forwarded-Host'] = rackreq.host if all_opts[:x_forwarded_host]
- 
+
       session = Net::HTTP.new(uri.host, uri.port)
       session.read_timeout=all_opts[:timeout] if all_opts[:timeout]
 

spec/rack/reverse_proxy_spec.rb

@@ -71,6 +71,21 @@ def app
       end
     end
 
+    describe "with x_forwarded_host turned on" do
+      def app
+        Rack::ReverseProxy.new(dummy_app) do
+          reverse_proxy_options :x_forwarded_host => true
+          reverse_proxy '/test', 'http://example.com/'
+        end
+      end
+
+      it "should optionally set the X-Forwarded-Host header to the proxying host" do
+        stub_request(:any, 'example.com/test/stuff')
+        get '/test/stuff'
+        a_request(:get, 'http://example.com/test/stuff').with(:headers => {'X-Forwarded-Host' => 'example.org'}).should have_been_made
+      end
+    end
+
     describe "with basic auth turned on" do
       def app
         Rack::ReverseProxy.new(dummy_app) do