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/subscriber_test.rb

130 lines
No EOL
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.after_dispatch"
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.after_dispatch"
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.after_dispatch"
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