Profile an RSpec example

profile.rb

#!/usr/bin/env ruby

$:.unshift 'spec'

require 'rspec'
require 'rails_helper'

spec = ARGV[0]

# -----------------------------------------------------------------------------
# If using StackProf
#
require 'stackprof'

interval = ENV.fetch('INTERVAL', 1000).to_i
limit = ENV.fetch('LIMIT', 20)
output_file = "tmp/#{spec.split('/').last}.dump"

StackProf.run(mode: :wall, out: output_file) do
  10.times do
    RSpec::Core::Runner.run([spec], $stderr, $stdout)
  end
end

system("stackprof #{output_file} --text --limit #{limit}")

# -----------------------------------------------------------------------------
# If using RubyProf
#
require 'ruby-prof'

result = RubyProf.profile do
  RSpec::Core::Runner.run([spec], $stderr, $stdout)
end

File.open('tmp/graphs/callstack.html', 'w') do |file|
  RubyProf::CallStackPrinter.new(result).print(file)
end