From 2e8aecdac24e5cc531dd6d3b59f67e71a008bd2a Mon Sep 17 00:00:00 2001 From: Carlos Antonio da Silva Date: Thu, 10 Jul 2014 08:12:29 -0300 Subject: [PATCH] Create an anonymous class to test the failure app No need to have a named class inside the test. --- test/test_helpers_test.rb | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/test/test_helpers_test.rb b/test/test_helpers_test.rb index f2db7485..997ba8e8 100644 --- a/test/test_helpers_test.rb +++ b/test/test_helpers_test.rb @@ -146,25 +146,21 @@ class TestHelpersTest < ActionController::TestCase assert_match /User ##{second_user.id}/, @response.body end - - test "passes given headers from the failure app to the response" do - + 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 + Devise.warden_config[:failure_app] = Class.new(Devise::FailureApp) do def respond self.status = 401 self.response.headers["CUSTOMHEADER"] = 1 end end - Devise.warden_config[:failure_app] = CustomTestFailureApp - user = create_user - sign_in user + + sign_in create_user get :index assert_equal 1, @response.headers["CUSTOMHEADER"] ensure Devise.warden_config[:failure_app] = old_failure_app end end - end