spec/spec_helper.rb in sinatra_warden-0.3.0 vs spec/spec_helper.rb in sinatra_warden-0.3.1

@@ -21,20 +21,33 @@
 
   config.before(:each) do
     DataMapper.auto_migrate!
   end
 
+  # default app
   def app
-    @app ||= Rack::Builder.app do
+    @app ||= define_app TestingLogin
+  end
+  
+  # app with auth_use_referrer enabled
+  def app_with_referrer
+    @app ||= define_app TestingLoginWithReferrer
+  end
+  
+  private 
+  
+  # :which should be a sinatra app
+  def define_app(which)
+    Rack::Builder.app do
       use Rack::Session::Cookie
       use Warden::Manager do |manager|
         manager.default_strategies :password
         manager.failure_app = TestingLogin
         manager.serialize_into_session { |user| user.id }
         manager.serialize_from_session { |id| User.get(id) }
       end
       use Rack::Flash
-      run TestingLogin
+      run which
     end
   end
 end