spec/support/sql_statement_profiling.rb

sql_statement_profiling.rb

RSpec.configure do |config|

  example_sql_counts = Hash.new(0)

  config.around(:example) do |procsy|
    sql_count = 0
    callback = ->(*args) { sql_count +=1 }
    ActiveSupport::Notifications.subscribed(callback, "sql.active_record") do
      procsy.call
    end

    example_sql_counts[procsy.example.id] += sql_count

    puts "SQL Count: #{sql_count}"
  end

  config.after(:suite) do
    example_count = example_sql_counts.size
    sum = example_sql_counts.sum(&:last)
    puts "TOTAL: #{sum}"
    average = sum.to_f / example_count
    puts "AVG: #{average}"
    most = example_sql_counts.sort_by(&:last).last
    puts "MOST: #{most}"
  end
end