diff options
Diffstat (limited to 'config/initializers/rspec_profiling.rb')
-rw-r--r-- | config/initializers/rspec_profiling.rb | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/config/initializers/rspec_profiling.rb b/config/initializers/rspec_profiling.rb index 5830a3d5af2..c5285dc389b 100644 --- a/config/initializers/rspec_profiling.rb +++ b/config/initializers/rspec_profiling.rb @@ -2,6 +2,8 @@ return unless Rails.env.test? +require 'gitlab_edition' + module RspecProfilingExt module Collectors class CSVWithTimestamps < ::RspecProfiling::Collectors::CSV @@ -30,7 +32,7 @@ module RspecProfilingExt module Git def branch if ENV['CI_COMMIT_REF_NAME'] - "#{defined?(Gitlab::License) ? 'ee' : 'ce'}:#{ENV['CI_COMMIT_REF_NAME']}" + "#{GitlabEdition.ee? ? 'ee' : 'ce'}:#{ENV['CI_COMMIT_REF_NAME']}" else super&.chomp end |