pp_sql_rails_test.rb

# frozen_string_literal: true

return if Gem::Specification.find_all_by_name('rails').empty?

require 'rails'
require 'test_helper'
require 'active_record'
require 'sqlite3'
ENV['RAILS_ENV'] = 'test'

module TeatApp
  class Application < Rails::Application
    config.eager_load = false
    config.load_defaults Gem::Version.new(Rails.version).segments.first(2).join('.').to_f
    initialize!
  end
end

module ActiveRecord
  class Base
    establish_connection(adapter: 'sqlite3', database: ':memory:')
    connection.create_table(:users) { |t| t.string :name }
    self.logger = Logger.new(::LOGGER)
  end
end

class User < ActiveRecord::Base; end

describe PpSql do
  after { clear_logs! && set_default_config! }

  it 'Rails & ActiveRecord with formatted output' do
    User.create
    assert(LOGGER.string.lines.detect { |line| line =~ /INTO\n/ })
    clear_logs!
    User.first
    assert_equal LOGGER.string.lines.count, 6
  end

  it 'Rails & ActiveRecord with default output' do
    PpSql.add_rails_logger_formatting = false
    User.create
    clear_logs!
    User.first
    assert_equal LOGGER.string.lines.count, 1
  end

  private

  def clear_logs!
    LOGGER.string.clear
  end

  def set_default_config!
    PpSql.add_rails_logger_formatting = true
  end
end