45 lines
1.2 KiB
Ruby
45 lines
1.2 KiB
Ruby
module RspecProfilingExt
|
|
module PSQL
|
|
def establish_connection
|
|
::RspecProfiling::Collectors::PSQL::Result.establish_connection(ENV['RSPEC_PROFILING_POSTGRES_URL'])
|
|
end
|
|
end
|
|
|
|
module Git
|
|
def branch
|
|
if ENV['CI_COMMIT_REF_NAME']
|
|
"#{defined?(Gitlab::License) ? 'ee' : 'ce'}:#{ENV['CI_COMMIT_REF_NAME']}"
|
|
else
|
|
super
|
|
end
|
|
end
|
|
end
|
|
|
|
module Run
|
|
def example_finished(*args)
|
|
super
|
|
rescue => err
|
|
return if @already_logged_example_finished_error
|
|
|
|
$stderr.puts "rspec_profiling couldn't collect an example: #{err}. Further warnings suppressed."
|
|
@already_logged_example_finished_error = true
|
|
end
|
|
|
|
alias_method :example_passed, :example_finished
|
|
alias_method :example_failed, :example_finished
|
|
end
|
|
end
|
|
|
|
if Rails.env.test?
|
|
RspecProfiling.configure do |config|
|
|
if ENV['RSPEC_PROFILING_POSTGRES_URL']
|
|
RspecProfiling::Collectors::PSQL.prepend(RspecProfilingExt::PSQL)
|
|
config.collector = RspecProfiling::Collectors::PSQL
|
|
end
|
|
|
|
if ENV.key?('CI')
|
|
RspecProfiling::VCS::Git.prepend(RspecProfilingExt::Git)
|
|
RspecProfiling::Run.prepend(RspecProfilingExt::Run)
|
|
end
|
|
end
|
|
end
|