diff options
Diffstat (limited to 'spec/tooling/lib/tooling/helpers/predictive_tests_helper_spec.rb')
-rw-r--r-- | spec/tooling/lib/tooling/helpers/predictive_tests_helper_spec.rb | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/spec/tooling/lib/tooling/helpers/predictive_tests_helper_spec.rb b/spec/tooling/lib/tooling/helpers/predictive_tests_helper_spec.rb new file mode 100644 index 00000000000..48a5866ac56 --- /dev/null +++ b/spec/tooling/lib/tooling/helpers/predictive_tests_helper_spec.rb @@ -0,0 +1,51 @@ +# frozen_string_literal: true + +require 'tempfile' +require_relative '../../../../../tooling/lib/tooling/helpers/predictive_tests_helper' + +class MockClass # rubocop:disable Gitlab/NamespacedClass + include Tooling::Helpers::PredictiveTestsHelper +end + +RSpec.describe Tooling::Helpers::PredictiveTestsHelper, feature_category: :tooling do + let(:instance) { MockClass.new } + + describe '#folders_for_available_editions' do + let(:base_folder_path) { 'app/views' } + + subject { instance.folders_for_available_editions(base_folder_path) } + + context 'when FOSS' do + before do + allow(GitlabEdition).to receive(:ee?).and_return(false) + allow(GitlabEdition).to receive(:jh?).and_return(false) + end + + it 'returns the correct paths' do + expect(subject).to match_array([base_folder_path]) + end + end + + context 'when EE' do + before do + allow(GitlabEdition).to receive(:ee?).and_return(true) + allow(GitlabEdition).to receive(:jh?).and_return(false) + end + + it 'returns the correct paths' do + expect(subject).to match_array([base_folder_path, "ee/#{base_folder_path}"]) + end + end + + context 'when JiHu' do + before do + allow(GitlabEdition).to receive(:ee?).and_return(true) + allow(GitlabEdition).to receive(:jh?).and_return(true) + end + + it 'returns the correct paths' do + expect(subject).to match_array([base_folder_path, "ee/#{base_folder_path}", "jh/#{base_folder_path}"]) + end + end + end +end |