From 9c195af89c7e3103306913ed8c61a7af6a8c4ff6 Mon Sep 17 00:00:00 2001
From: shogo-ohta <shogo-ohta@cookpad.com>
Date: Fri, 26 Aug 2011 23:10:20 +0900
Subject: [PATCH] added spec for response header

---
 spec/integration/session_spec.rb | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/spec/integration/session_spec.rb b/spec/integration/session_spec.rb
index 57a8084..3258b4b 100644
--- a/spec/integration/session_spec.rb
+++ b/spec/integration/session_spec.rb
@@ -80,7 +80,7 @@ describe Capybara::Session do
     end
   end
 
-  context "status code" do
+  context "response headers with status code" do
     before(:all) do
       @app = lambda do |env|
         params = ::Rack::Utils.parse_query(env['QUERY_STRING'])
@@ -96,7 +96,7 @@ describe Capybara::Session do
           </html>
         HTML
         [200,
-          { 'Content-Type' => 'text/html', 'Content-Length' => body.length.to_s },
+          { 'Content-Type' => 'text/html', 'Content-Length' => body.length.to_s, 'X-Capybara' => 'WebKit'},
           [body]]
       end
     end
@@ -112,6 +112,18 @@ describe Capybara::Session do
       subject.reset!
       subject.status_code.should == 0
     end
+
+    it "should get response headers" do
+      subject.visit '/'
+      subject.response_headers['X-Capybara'].should == 'WebKit'
+    end
+
+    it "should reset response headers" do
+      subject.visit '/'
+      subject.response_headers['X-Capybara'].should == 'WebKit'
+      subject.reset!
+      subject.response_headers['X-Capybara'].should == nil
+    end
   end
 end