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

@@ -17,13 +17,14 @@
 require 'rack/iframe'
 
 ENV['RACK_ENV'] = 'test'
 
 class CachedApp < Sinatra::Base
-
   use Rack::Cache, :verbose => true, :meta_store => 'heap:/', :entitystore => 'heap:/'
 
+  enable :sessions
+
   get '/' do
     headers['Content-Type'] = 'text/plain'
     ""
   end
 
@@ -35,9 +36,18 @@
 
   get '/last_modified' do
     headers['Last-Modified'] = Chronic.parse('0 minutes ago')
     headers['Content-Type'] = 'text/plain'
     ""
+  end
+end
+
+class SessionIframeApp < CachedApp
+  use Rack::Iframe
+
+  get '/test_iframe_session' do
+    headers['Content-Type'] = 'text/plain'
+    "#{session[:iframe_session]}"
   end
 end
 
 def mock_app(headers = {}, env = {})
   default_headers = headers.merge({