f11030173b
Instead of inserting a row after each example to an external database, we save the CI profiling reports into the `rspec_profiling` directory and insert the data in the update-tests-metadata CI stage. This should make each spec run faster and also reduce the number of PostgreSQL connections needed by concurrent CI builds. `scripts/insert-rspec-profiling-data` also inserts one file at a time via the PostgreSQL COPY command for faster inserts. The one side effect is that the `created_at` and `updated_at` timestamps aren't available since they aren't generated in the CSV. Closes https://gitlab.com/gitlab-org/gitlab-ee/issues/10154
65 lines
1.8 KiB
Ruby
65 lines
1.8 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 = -> { "rspec_profiling/#{Time.now.to_i}-#{SecureRandom.hex(8)}-rspec-data.csv" }
|
|
end
|
|
end
|