1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/railties/test/rack_logger_test.rb

76 lines
1.9 KiB
Ruby
Raw Normal View History

require 'abstract_unit'
require 'active_support/testing/autorun'
require 'active_support/test_case'
require 'rails/rack/logger'
require 'logger'
module Rails
module Rack
class LoggerTest < ActiveSupport::TestCase
class TestLogger < Rails::Rack::Logger
NULL = ::Logger.new File::NULL
attr_reader :logger
def initialize(logger = NULL, taggers = nil, &block)
super(->(_) { block.call; [200, {}, []] }, taggers)
@logger = logger
end
def development?; false; end
end
class Subscriber < Struct.new(:starts, :finishes)
def initialize(starts = [], finishes = [])
super
end
def start(name, id, payload)
starts << [name, id, payload]
end
def finish(name, id, payload)
finishes << [name, id, payload]
end
end
attr_reader :subscriber, :notifier
def setup
@subscriber = Subscriber.new
@notifier = ActiveSupport::Notifications.notifier
2014-07-11 04:38:26 -04:00
@subscription = notifier.subscribe 'request.action_dispatch', subscriber
end
def teardown
2014-07-11 04:38:26 -04:00
notifier.unsubscribe @subscription
end
def test_notification
logger = TestLogger.new { }
assert_difference('subscriber.starts.length') do
assert_difference('subscriber.finishes.length') do
logger.call('REQUEST_METHOD' => 'GET').last.close
end
end
end
def test_notification_on_raise
logger = TestLogger.new do
# using an exception class that is not a StandardError subclass on purpose
raise NotImplementedError
end
assert_difference('subscriber.starts.length') do
assert_difference('subscriber.finishes.length') do
assert_raises(NotImplementedError) do
logger.call 'REQUEST_METHOD' => 'GET'
end
end
end
end
end
end
end