Ticket #10079: enhanced_rescue_from.diff

actionpack/test/controller/rescue_test.rb

@@ -5 +5 @@
-class RescueController < ActionController::Base
-  class NotAuthorized < StandardError
-  end
+  class NotAuthorizedToRescueAsString < StandardError
+  end
-
-  class RecordInvalid < StandardError
-  end
+  class RecordInvalidToRescueAsString < StandardError
+  end
-
-  class NotAllowed < StandardError
-  end
-
+
+
+
-  class InvalidRequest < StandardError
-  end
+  class InvalidRequestToRescueAsString < StandardError
+  end
-
-  class BadGateway < StandardError
-  end
+  class BadGatewayToRescueAsString < StandardError
+  end
-
-  class ResourceUnavailable < StandardError
-  end
+  class ResourceUnavailableToRescueAsString < StandardError
+  end
-
+  # We use a fully-qualified name in some strings, and a relative constant
+  # name in some other to test correct handling of both cases.
+  
-  rescue_from NotAuthorized, :with => :deny_access
+  rescue_from 'RescueController::NotAuthorizedToRescueAsString', :with => :deny_access
+  
-  rescue_from RecordInvalid, :with => :show_errors
+  rescue_from 'RescueController::RecordInvalidToRescueAsString', :with => :show_errors
-
-  rescue_from NotAllowed, :with => proc { head :forbidden }
+  rescue_from 'RescueController::NotAllowedToRescueAsString', :with => proc { head :forbidden }
+  
-  rescue_from InvalidRequest, :with => proc { |exception| render :text => exception.message }
+  rescue_from 'InvalidRequestToRescueAsString', :with => proc { |exception| render :text => exception.message }
-
-  rescue_from BadGateway do
-    head :status => 502
-  end
+  rescue_from 'BadGatewayToRescueAsString' do
+    head :status => 502
+  end
-
-  rescue_from ResourceUnavailable do |exception|
-    render :text => exception.message
-  end
+  rescue_from 'ResourceUnavailableToRescueAsString' do |exception|
+    render :text => exception.message
+  end
-
-  def raises
-    render :text => 'already rendered'
@@ -51 +78 @@
-  def not_authorized
-    raise NotAuthorized
-  end
-
+
+
+
+
-  def not_allowed
-    raise NotAllowed
-  end
+  def not_allowed_raise_as_string
+    raise NotAllowedToRescueAsString
+  end
-
-  def invalid_request
-    raise InvalidRequest
-  end
+  def invalid_request_raise_as_string
+    raise InvalidRequestToRescueAsString
+  end
-
-  def record_invalid
-    raise RecordInvalid
-  end
+  def record_invalid_raise_as_string
+    raise RecordInvalidToRescueAsString
+  end
-  
-  def bad_gateway
-    raise BadGateway
-  end
+  def bad_gateway_raise_as_string
+    raise BadGatewayToRescueAsString
+  end
-
-  def resource_unavailable
-    raise ResourceUnavailable
-  end
-
+
+
+
+
-  def missing_template
-  end
-
@@ -278 +323 @@
-    get :not_authorized
-    assert_response :forbidden
-  end
+  def test_rescue_handler_string
+    get :not_authorized_raise_as_string
+    assert_response :forbidden
+  end
-
-  def test_rescue_handler_with_argument
-    @controller.expects(:show_errors).once.with { |e| e.is_a?(Exception) }
-    get :record_invalid
-  end
+  def test_rescue_handler_with_argument_as_string
+    @controller.expects(:show_errors).once.with { |e| e.is_a?(Exception) }
+    get :record_invalid_raise_as_string
+  end
-
-  def test_proc_rescue_handler
-    get :not_allowed
-    assert_response :forbidden
-  end
+  def test_proc_rescue_handler_as_string
+    get :not_allowed_raise_as_string
+    assert_response :forbidden
+  end
-
-  def test_proc_rescue_handle_with_argument
-    get :invalid_request
-    assert_equal "RescueController::InvalidRequest", @response.body
-  end
+  def test_proc_rescue_handle_with_argument_as_string
+    get :invalid_request_raise_as_string
+    assert_equal "RescueController::InvalidRequestToRescueAsString", @response.body
+  end
-
-  def test_block_rescue_handler
-    get :bad_gateway
-    assert_response 502
-  end
+  def test_block_rescue_handler_as_string
+    get :bad_gateway_raise_as_string
+    assert_response 502
+  end
-
-  def test_block_rescue_handler_with_argument
-    get :resource_unavailable
-    assert_equal "RescueController::ResourceUnavailable", @response.body
-  end
-
+  def test_block_rescue_handler_with_argument_as_string
+    get :resource_unavailable_raise_as_string
+    assert_equal "RescueController::ResourceUnavailableToRescueAsString", @response.body
+  end
+
+  
-  protected
-    def with_all_requests_local(local = true)
-      old_local, ActionController::Base.consider_all_requests_local =
@@ -339 +410 @@
-    end
-end
-
+class ExceptionInheritanceRescueController < ActionController::Base
+
+  class ParentException < StandardError
+  end
+
+  class ChildException < ParentException
+  end
+
+  class GrandchildException < ChildException
+  end
+
+  rescue_from ChildException,      :with => lambda { head :ok }
+  rescue_from ParentException,     :with => lambda { head :created }
+  rescue_from GrandchildException, :with => lambda { head :no_content }
+  
+  def raise_parent_exception
+    raise ParentException
+  end
+  
+  def raise_child_exception
+    raise ChildException
+  end
+  
+  def raise_grandchild_exception
+    raise GrandchildException
+  end
+end
+  
+class ExceptionInheritanceRescueTest < Test::Unit::TestCase
+ 
+  def setup
+    @controller = ExceptionInheritanceRescueController.new
+    @request    = ActionController::TestRequest.new
+    @response   = ActionController::TestResponse.new
+  end
+  
+  def test_bottom_first
+    get :raise_grandchild_exception
+    assert_response :no_content
+  end
+  
+  def test_inheritance_works
+    get :raise_child_exception
+    assert_response :created
+  end
+end
+
+class ControllerInheritanceRescueController < ExceptionInheritanceRescueController
+  class FirstExceptionInChildController < StandardError
+  end
+
+  class SecondExceptionInChildController < StandardError
+  end
+  
+  rescue_from FirstExceptionInChildController, 'SecondExceptionInChildController', :with => lambda { head :gone }
+  
+  def raise_first_exception_in_child_controller
+    raise FirstExceptionInChildController
+  end
+
+  def raise_second_exception_in_child_controller
+    raise SecondExceptionInChildController
+  end
+end
+
+class ControllerInheritanceRescueControllerTest < Test::Unit::TestCase
+ 
+  def setup
+    @controller = ControllerInheritanceRescueController.new
+    @request    = ActionController::TestRequest.new
+    @response   = ActionController::TestResponse.new
+  end
+  
+  def test_first_exception_in_child_controller
+    get :raise_first_exception_in_child_controller
+    assert_response :gone
+  end
+
+  def test_second_exception_in_child_controller
+    get :raise_second_exception_in_child_controller
+    assert_response :gone
+  end
+  
+  def test_exception_in_parent_controller
+    get :raise_parent_exception
+    assert_response :created
+  end
+end
-end # uses_mocha

actionpack/lib/action_controller/rescue.rb

@@ -41 +41 @@
-      base.rescue_templates = Hash.new(DEFAULT_RESCUE_TEMPLATE)
-      base.rescue_templates.update DEFAULT_RESCUE_TEMPLATES
-
-hash
-{}
+array
+[]
-
-      base.extend(ClassMethods)
-      base.class_eval do
@@ -55 +55 @@
-        new.process(request, response, :rescue_action, exception)
-      end
-
-
-
-
+
+
+
+
+
+
+
+
+
+
+
+
+
-      # class ApplicationController < ActionController::Base
-      #   rescue_from User::NotAuthorized, :with => :deny_access # self defined exception
-      #   rescue_from ActiveRecord::RecordInvalid, :with => :show_errors
-
+
+
+
+
+
-      #   protected
-      #     def deny_access
-      #       ...
-      #     end
-
+ 
-      #     def show_errors(exception)
-      #       exception.record.new_record? ? ...
-      #     end
@@ -78 +92 @@
-        end
-
-        klasses.each do |klass|
-
+
+
+
+
+
+
+
+
+
+
+
-        end
-      end
-    end
@@ -192 +216 @@
-      end
-
-      def handler_for_rescue(exception)
-
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-        when Symbol
-          method(handler)
-        when Proc