116 lines
2.9 KiB
Ruby
116 lines
2.9 KiB
Ruby
require "cases/helper"
|
|
require "models/developer"
|
|
require "models/post"
|
|
require "active_support/log_subscriber/test_helper"
|
|
|
|
class LogSubscriberTest < ActiveRecord::TestCase
|
|
include ActiveSupport::LogSubscriber::TestHelper
|
|
include ActiveSupport::BufferedLogger::Severity
|
|
|
|
fixtures :posts
|
|
|
|
def setup
|
|
@old_logger = ActiveRecord::Base.logger
|
|
@using_identity_map = ActiveRecord::IdentityMap.enabled?
|
|
ActiveRecord::IdentityMap.enabled = false
|
|
Developer.primary_key
|
|
super
|
|
ActiveRecord::LogSubscriber.attach_to(:active_record)
|
|
end
|
|
|
|
def teardown
|
|
super
|
|
ActiveRecord::LogSubscriber.log_subscribers.pop
|
|
ActiveRecord::Base.logger = @old_logger
|
|
ActiveRecord::IdentityMap.enabled = @using_identity_map
|
|
end
|
|
|
|
def set_logger(logger)
|
|
ActiveRecord::Base.logger = logger
|
|
end
|
|
|
|
def test_schema_statements_are_ignored
|
|
event = Struct.new(:duration, :payload)
|
|
|
|
logger = Class.new(ActiveRecord::LogSubscriber) {
|
|
attr_accessor :debugs
|
|
|
|
def initialize
|
|
@debugs = []
|
|
super
|
|
end
|
|
|
|
def debug message
|
|
@debugs << message
|
|
end
|
|
}.new
|
|
assert_equal 0, logger.debugs.length
|
|
|
|
logger.sql(event.new(0, { :sql => 'hi mom!' }))
|
|
assert_equal 1, logger.debugs.length
|
|
|
|
logger.sql(event.new(0, { :sql => 'hi mom!', :name => 'foo' }))
|
|
assert_equal 2, logger.debugs.length
|
|
|
|
logger.sql(event.new(0, { :sql => 'hi mom!', :name => 'SCHEMA' }))
|
|
assert_equal 2, logger.debugs.length
|
|
end
|
|
|
|
def test_basic_query_logging
|
|
Developer.all
|
|
wait
|
|
assert_equal 1, @logger.logged(:debug).size
|
|
assert_match(/Developer Load/, @logger.logged(:debug).last)
|
|
assert_match(/SELECT .*?FROM .?developers.?/i, @logger.logged(:debug).last)
|
|
end
|
|
|
|
def test_exists_query_logging
|
|
Developer.exists? 1
|
|
wait
|
|
assert_equal 1, @logger.logged(:debug).size
|
|
assert_match(/Developer Exists/, @logger.logged(:debug).last)
|
|
assert_match(/SELECT .*?FROM .?developers.?/i, @logger.logged(:debug).last)
|
|
end
|
|
|
|
def test_cached_queries
|
|
ActiveRecord::Base.cache do
|
|
Developer.all
|
|
Developer.all
|
|
end
|
|
wait
|
|
assert_equal 2, @logger.logged(:debug).size
|
|
assert_match(/CACHE/, @logger.logged(:debug).last)
|
|
assert_match(/SELECT .*?FROM .?developers.?/i, @logger.logged(:debug).last)
|
|
end
|
|
|
|
def test_basic_query_doesnt_log_when_level_is_not_debug
|
|
@logger.level = INFO
|
|
Developer.all
|
|
wait
|
|
assert_equal 0, @logger.logged(:debug).size
|
|
end
|
|
|
|
def test_cached_queries_doesnt_log_when_level_is_not_debug
|
|
@logger.level = INFO
|
|
ActiveRecord::Base.cache do
|
|
Developer.all
|
|
Developer.all
|
|
end
|
|
wait
|
|
assert_equal 0, @logger.logged(:debug).size
|
|
end
|
|
|
|
def test_initializes_runtime
|
|
Thread.new { assert_equal 0, ActiveRecord::LogSubscriber.runtime }.join
|
|
end
|
|
|
|
def test_log
|
|
ActiveRecord::IdentityMap.use do
|
|
Post.find 1
|
|
Post.find 1
|
|
end
|
|
wait
|
|
assert_match(/From Identity Map/, @logger.logged(:debug).last)
|
|
end
|
|
end
|