130 lines
3.4 KiB
Ruby
130 lines
3.4 KiB
Ruby
require 'abstract_unit'
|
|
require 'rails/subscriber/test_helper'
|
|
|
|
class MySubscriber < Rails::Subscriber
|
|
attr_reader :event
|
|
|
|
def some_event(event)
|
|
@event = event
|
|
info event.name
|
|
end
|
|
|
|
def foo(event)
|
|
debug "debug"
|
|
info "info"
|
|
warn "warn"
|
|
end
|
|
|
|
def bar(event)
|
|
info "#{color("cool", :red)}, #{color("isn't it?", :blue, true)}"
|
|
end
|
|
end
|
|
|
|
module SubscriberTest
|
|
def setup
|
|
super
|
|
@subscriber = MySubscriber.new
|
|
Rails::Subscriber.instance_variable_set(:@log_tailer, nil)
|
|
end
|
|
|
|
def teardown
|
|
super
|
|
Rails::Subscriber.subscribers.clear
|
|
Rails::Subscriber.instance_variable_set(:@log_tailer, nil)
|
|
end
|
|
|
|
def instrument(*args, &block)
|
|
ActiveSupport::Notifications.instrument(*args, &block)
|
|
end
|
|
|
|
def test_proxies_method_to_rails_logger
|
|
@subscriber.foo(nil)
|
|
assert_equal %w(debug), @logger.logged(:debug)
|
|
assert_equal %w(info), @logger.logged(:info)
|
|
assert_equal %w(warn), @logger.logged(:warn)
|
|
end
|
|
|
|
def test_set_color_for_messages
|
|
Rails::Subscriber.colorize_logging = true
|
|
@subscriber.bar(nil)
|
|
assert_equal "\e[31mcool\e[0m, \e[1m\e[34misn't it?\e[0m", @logger.logged(:info).last
|
|
end
|
|
|
|
def test_does_not_set_color_if_colorize_logging_is_set_to_false
|
|
@subscriber.bar(nil)
|
|
assert_equal "cool, isn't it?", @logger.logged(:info).last
|
|
end
|
|
|
|
def test_event_is_sent_to_the_registered_class
|
|
Rails::Subscriber.add :my_subscriber, @subscriber
|
|
instrument "my_subscriber.some_event"
|
|
wait
|
|
assert_equal %w(my_subscriber.some_event), @logger.logged(:info)
|
|
end
|
|
|
|
def test_event_is_an_active_support_notifications_event
|
|
Rails::Subscriber.add :my_subscriber, @subscriber
|
|
instrument "my_subscriber.some_event"
|
|
wait
|
|
assert_kind_of ActiveSupport::Notifications::Event, @subscriber.event
|
|
end
|
|
|
|
def test_does_not_send_the_event_if_it_doesnt_match_the_class
|
|
Rails::Subscriber.add :my_subscriber, @subscriber
|
|
instrument "my_subscriber.unknown_event"
|
|
wait
|
|
# If we get here, it means that NoMethodError was raised.
|
|
end
|
|
|
|
def test_does_not_send_the_event_if_logger_is_nil
|
|
Rails.logger = nil
|
|
@subscriber.expects(:some_event).never
|
|
Rails::Subscriber.add :my_subscriber, @subscriber
|
|
instrument "my_subscriber.some_event"
|
|
wait
|
|
end
|
|
|
|
def test_flushes_loggers
|
|
Rails::Subscriber.add :my_subscriber, @subscriber
|
|
Rails::Subscriber.flush_all!
|
|
assert_equal 1, @logger.flush_count
|
|
end
|
|
|
|
def test_flushes_loggers_when_action_dispatch_callback_is_received
|
|
Rails::Subscriber.add :my_subscriber, @subscriber
|
|
instrument "action_dispatch.callback"
|
|
wait
|
|
assert_equal 1, @logger.flush_count
|
|
end
|
|
|
|
def test_flushes_the_same_logger_just_once
|
|
Rails::Subscriber.add :my_subscriber, @subscriber
|
|
Rails::Subscriber.add :another, @subscriber
|
|
instrument "action_dispatch.callback"
|
|
wait
|
|
assert_equal 1, @logger.flush_count
|
|
end
|
|
|
|
def test_tails_logs_when_action_dispatch_callback_is_received
|
|
log_tailer = mock()
|
|
log_tailer.expects(:tail!)
|
|
Rails::Subscriber.log_tailer = log_tailer
|
|
|
|
Rails::Subscriber.add :my_subscriber, @subscriber
|
|
instrument "action_dispatch.callback"
|
|
wait
|
|
ensure
|
|
Rails::Subscriber.log_tailer = nil
|
|
end
|
|
|
|
class SyncSubscriberTest < ActiveSupport::TestCase
|
|
include Rails::Subscriber::SyncTestHelper
|
|
include SubscriberTest
|
|
end
|
|
|
|
class AsyncSubscriberTest < ActiveSupport::TestCase
|
|
include Rails::Subscriber::AsyncTestHelper
|
|
include SubscriberTest
|
|
end
|
|
|
|
end |