Class ActiveRecord::RecordFetchWarningTest < ActiveRecord::TestCase

# File activerecord/test/cases/relation/record_fetch_warning_test.rb, line 9
def setup
  @original_logger = ActiveRecord::Base.logger
  @original_warn_on_records_fetched_greater_than = ActiveRecord::Base.warn_on_records_fetched_greater_than
  @log = StringIO.new
end

# File activerecord/test/cases/relation/record_fetch_warning_test.rb, line 15
def teardown
  ActiveRecord::Base.logger = @original_logger
  ActiveRecord::Base.warn_on_records_fetched_greater_than = @original_warn_on_records_fetched_greater_than
end

# File activerecord/test/cases/relation/record_fetch_warning_test.rb, line 30
def test_does_not_warn_on_records_fetched_less_than_allowed_limit
  ActiveRecord::Base.logger = ActiveSupport::Logger.new(@log)
  ActiveRecord::Base.logger.level = Logger::WARN
  ActiveRecord::Base.warn_on_records_fetched_greater_than = 100

  Post.all.to_a

  assert_no_match(/Query fetched/, @log.string)
end

# File activerecord/test/cases/relation/record_fetch_warning_test.rb, line 20
def test_warn_on_records_fetched_greater_than_allowed_limit
  ActiveRecord::Base.logger = ActiveSupport::Logger.new(@log)
  ActiveRecord::Base.logger.level = Logger::WARN
  ActiveRecord::Base.warn_on_records_fetched_greater_than = 1

  Post.all.to_a

  assert_match(/Query fetched/, @log.string)
end