spec/rack-iframe_spec.rb in rack-iframe-0.0.2 vs spec/rack-iframe_spec.rb in rack-iframe-0.0.3

@@ -201,9 +201,18 @@
 
             headers['P3P'].must_equal %(CP="ALL DSP COR CURa ADMa DEVa OUR IND COM NAV")
             status.must_equal 200 # modified
           end
         end
+
+        it 'should set session variable :iframe_session on request to /iframe_session' do
+          @user_agents.each do |user_agent|
+            browser = Rack::Test::Session.new(Rack::MockSession.new(SessionIframeApp))
+            browser.get '/iframe_session', {}, 'HTTP_USER_AGENT' => user_agent_string(user_agent)
+            browser.get '/test_iframe_session', {}, 'HTTP_USER_AGENT' => user_agent_string(user_agent)
+            browser.last_response.body.must_equal "true"
+          end
+        end
       end
     end
   end
 end