72 lines
1.8 KiB
Ruby
72 lines
1.8 KiB
Ruby
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
|
|
notifier.subscribe 'action_dispatch.request', subscriber
|
|
end
|
|
|
|
def teardown
|
|
notifier.unsubscribe subscriber
|
|
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 { raise }
|
|
|
|
assert_difference('subscriber.starts.length') do
|
|
assert_difference('subscriber.finishes.length') do
|
|
assert_raises(RuntimeError) do
|
|
logger.call 'REQUEST_METHOD' => 'GET'
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|