vendor/rails/actionpack/test/controller/dispatcher_test.rb

   1 require 'abstract_unit'
   2
   3 class DispatcherTest < Test::Unit::TestCase
   4   Dispatcher = ActionController::Dispatcher
   5
   6   def setup
   7     ENV['REQUEST_METHOD'] = 'GET'
   8
   9     Dispatcher.middleware = ActionController::MiddlewareStack.new do |middleware|
  10       middlewares = File.expand_path(File.join(File.dirname(__FILE__), "../../lib/action_controller/middlewares.rb"))
  11       middleware.instance_eval(File.read(middlewares))
  12     end
  13
  14     # Clear callbacks as they are redefined by Dispatcher#define_dispatcher_callbacks
  15     Dispatcher.instance_variable_set("@prepare_dispatch_callbacks", ActiveSupport::Callbacks::CallbackChain.new)
  16     Dispatcher.instance_variable_set("@before_dispatch_callbacks", ActiveSupport::Callbacks::CallbackChain.new)
  17     Dispatcher.instance_variable_set("@after_dispatch_callbacks", ActiveSupport::Callbacks::CallbackChain.new)
  18
  19     Dispatcher.stubs(:require_dependency)
  20   end
  21
  22   def teardown
  23     ENV.delete 'REQUEST_METHOD'
  24   end
  25
  26   def test_clears_dependencies_after_dispatch_if_in_loading_mode
  27     ActiveSupport::Dependencies.expects(:clear).once
  28     dispatch(false)
  29   end
  30
  31   def test_reloads_routes_before_dispatch_if_in_loading_mode
  32     ActionController::Routing::Routes.expects(:reload).once
  33     dispatch(false)
  34   end
  35
  36   def test_leaves_dependencies_after_dispatch_if_not_in_loading_mode
  37     ActionController::Routing::Routes.expects(:reload).never
  38     ActiveSupport::Dependencies.expects(:clear).never
  39
  40     dispatch
  41   end
  42
  43   # Stub out dispatch error logger
  44   class << Dispatcher
  45     def log_failsafe_exception(status, exception); end
  46   end
  47
  48   def test_failsafe_response
  49     Dispatcher.any_instance.expects(:dispatch).raises('b00m')
  50     ActionController::Failsafe.any_instance.expects(:log_failsafe_exception)
  51
  52     assert_nothing_raised do
  53       assert_equal [
  54         500,
  55         {"Content-Type" => "text/html"},
  56         "<html><body><h1>500 Internal Server Error</h1></body></html>"
  57       ], dispatch
  58     end
  59   end
  60
  61   def test_prepare_callbacks
  62     a = b = c = nil
  63     Dispatcher.to_prepare { |*args| a = b = c = 1 }
  64     Dispatcher.to_prepare { |*args| b = c = 2 }
  65     Dispatcher.to_prepare { |*args| c = 3 }
  66
  67     # Ensure to_prepare callbacks are not run when defined
  68     assert_nil a || b || c
  69
  70     # Run callbacks
  71     Dispatcher.run_prepare_callbacks
  72
  73     assert_equal 1, a
  74     assert_equal 2, b
  75     assert_equal 3, c
  76
  77     # Make sure they are only run once
  78     a = b = c = nil
  79     dispatch
  80     assert_nil a || b || c
  81   end
  82
  83   def test_to_prepare_with_identifier_replaces
  84     a = b = nil
  85     Dispatcher.to_prepare(:unique_id) { |*args| a = b = 1 }
  86     Dispatcher.to_prepare(:unique_id) { |*args| a = 2 }
  87
  88     Dispatcher.run_prepare_callbacks
  89     assert_equal 2, a
  90     assert_equal nil, b
  91   end
  92
  93   private
  94     def dispatch(cache_classes = true)
  95       ActionController::Routing::RouteSet.any_instance.stubs(:call).returns([200, {}, 'response'])
  96       Dispatcher.define_dispatcher_callbacks(cache_classes)
  97       Dispatcher.new.call({})
  98     end
  99
 100     def assert_subclasses(howmany, klass, message = klass.subclasses.inspect)
 101       assert_equal howmany, klass.subclasses.size, message
 102     end
 103 end