68 lines
1.9 KiB
Ruby
68 lines
1.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
return unless Rails.env.test?
|
|
|
|
module RspecProfilingExt
|
|
module Collectors
|
|
class CSVWithTimestamps < ::RspecProfiling::Collectors::CSV
|
|
TIMESTAMP_FIELDS = %w(created_at updated_at).freeze
|
|
HEADERS = (::RspecProfiling::Collectors::CSV::HEADERS + TIMESTAMP_FIELDS).freeze
|
|
|
|
def insert(attributes)
|
|
output << HEADERS.map do |field|
|
|
if TIMESTAMP_FIELDS.include?(field)
|
|
Time.now
|
|
else
|
|
attributes.fetch(field.to_sym)
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def output
|
|
@output ||= ::CSV.open(path, "w").tap { |csv| csv << HEADERS }
|
|
end
|
|
end
|
|
end
|
|
|
|
module Git
|
|
def branch
|
|
if ENV['CI_COMMIT_REF_NAME']
|
|
"#{defined?(Gitlab::License) ? 'ee' : 'ce'}:#{ENV['CI_COMMIT_REF_NAME']}"
|
|
else
|
|
super&.chomp
|
|
end
|
|
end
|
|
|
|
def sha
|
|
super&.chomp
|
|
end
|
|
end
|
|
|
|
module Run
|
|
def example_finished(*args)
|
|
super
|
|
rescue => err
|
|
return if @already_logged_example_finished_error # rubocop:disable Gitlab/ModuleWithInstanceVariables
|
|
|
|
$stderr.puts "rspec_profiling couldn't collect an example: #{err}. Further warnings suppressed."
|
|
@already_logged_example_finished_error = true # rubocop:disable Gitlab/ModuleWithInstanceVariables
|
|
end
|
|
|
|
alias_method :example_passed, :example_finished
|
|
alias_method :example_failed, :example_finished
|
|
end
|
|
end
|
|
|
|
RspecProfiling.configure do |config|
|
|
if ENV.key?('CI') || ENV.key?('RSPEC_PROFILING')
|
|
RspecProfiling::VCS::Git.prepend(RspecProfilingExt::Git)
|
|
RspecProfiling::Run.prepend(RspecProfilingExt::Run)
|
|
config.collector = RspecProfilingExt::Collectors::CSVWithTimestamps
|
|
config.csv_path = -> do
|
|
prefix = "#{ENV['CI_JOB_NAME']}-".tr(' ', '-') if ENV['CI_JOB_NAME']
|
|
"rspec_profiling/#{prefix}#{Time.now.to_i}-#{SecureRandom.hex(8)}-rspec-data.csv"
|
|
end
|
|
end
|
|
end
|