require 'test_helper' class TestHelpersTest < ActionController::TestCase tests UsersController include Devise::TestHelpers class CustomFailureApp < Devise::FailureApp def redirect self.status = 306 end end test "redirects if attempting to access a page unauthenticated" do get :index assert_redirected_to new_user_session_path assert_equal "You need to sign in or sign up before continuing.", flash[:alert] end test "redirects if attempting to access a page with an unconfirmed account" do swap Devise, :allow_unconfirmed_access_for => 0.days do user = create_user assert !user.active_for_authentication? sign_in user get :index assert_redirected_to new_user_session_path end end test "returns nil if accessing current_user with an unconfirmed account" do swap Devise, :allow_unconfirmed_access_for => 0.days do user = create_user assert !user.active_for_authentication? sign_in user get :accept, :id => user assert_nil assigns(:current_user) end end test "does not redirect with valid user" do user = create_user user.confirm! sign_in user get :index assert_response :success end test "does not redirect with valid user after failed first attempt" do get :index assert_response :redirect user = create_user user.confirm! sign_in user get :index assert_response :success end test "redirects if valid user signed out" do user = create_user user.confirm! sign_in user get :index sign_out user get :index assert_redirected_to new_user_session_path end test "respects custom failure app" do begin Devise.warden_config.failure_app = CustomFailureApp get :index assert_response 306 ensure Devise.warden_config.failure_app = Devise::FailureApp end end test "returns the body of a failure app" do get :index assert_equal response.body, "You are being redirected." end test "defined Warden after_authentication callback should not be called when sign_in is called" do begin Warden::Manager.after_authentication do |user, auth, opts| flunk "callback was called while it should not" end user = create_user user.confirm! sign_in user ensure Warden::Manager._after_set_user.pop end end test "defined Warden before_logout callback should not be called when sign_out is called" do begin Warden::Manager.before_logout do |user, auth, opts| flunk "callback was called while it should not" end user = create_user user.confirm! sign_in user sign_out user ensure Warden::Manager._before_logout.pop end end test "before_failure call should work" do begin executed = false Warden::Manager.before_failure do |env,opts| executed = true end user = create_user sign_in user get :index assert executed ensure Warden::Manager._before_failure.pop end end test "allows to sign in with different users" do first_user = create_user first_user.confirm! sign_in first_user get :index assert_match /User ##{first_user.id}/, @response.body sign_out first_user second_user = create_user second_user.confirm! sign_in second_user get :index assert_match /User ##{second_user.id}/, @response.body end test "passes given headers from the failure app to the response" do begin old_failure_app = Devise.warden_config[:failure_app] class CustomTestFailureApp < Devise::FailureApp def respond self.status = 401 self.response.headers["CUSTOMHEADER"] = 1 end end Devise.warden_config[:failure_app] = CustomTestFailureApp user = create_user sign_in user get :index assert_equal 1, @response.headers["CUSTOMHEADER"] ensure Devise.warden_config[:failure_app] = old_failure_app end end end