1 files changed, 69 insertions, 2 deletions
diff --git a/qa/spec/specs/helpers/quarantine_spec.rb b/qa/spec/specs/helpers/quarantine_spec.rb
index 9686a9771c4..41bc3eadff4 100644
--- a/qa/spec/specs/helpers/quarantine_spec.rb
+++ b/qa/spec/specs/helpers/quarantine_spec.rb
@@ -36,7 +36,9 @@ RSpec.configure do |c|
   end
 end
 
-describe QA::Specs::Helpers::Quarantine do
+RSpec.describe QA::Specs::Helpers::Quarantine do
+  include Helpers::StubENV
+
   describe '.skip_or_run_quarantined_contexts' do
     context 'with no tag focused' do
       before do
@@ -312,7 +314,7 @@ describe QA::Specs::Helpers::Quarantine do
     end
   end
 
-  describe 'running against specific environments' do
+  describe 'running against specific environments or pipelines' do
     before do
       QA::Runtime::Scenario.define(:gitlab_address, 'https://staging.gitlab.com')
       described_class.configure_rspec
@@ -400,5 +402,70 @@ describe QA::Specs::Helpers::Quarantine do
 
       expect(group.examples.first.execution_result.pending_message).to match(/[Tt]est.*not compatible.*environment/)
     end
+
+    context 'with pipeline constraints' do
+      context 'without CI_PROJECT_NAME set' do
+        before do
+          stub_env('CI_PROJECT_NAME', nil)
+          described_class.configure_rspec
+        end
+
+        it 'runs on any pipeline' do
+          group = describe_successfully do
+            it('runs given a single named pipeline', only: { pipeline: :nightly } ) {}
+            it('runs given an array of pipelines', only: { pipeline: [:canary, :not_nightly] }) {}
+          end
+
+          aggregate_failures do
+            expect(group.examples[0].execution_result.status).to eq(:passed)
+            expect(group.examples[1].execution_result.status).to eq(:passed)
+          end
+        end
+      end
+
+      context 'when a pipeline triggered from master runs in gitlab-qa' do
+        before do
+          stub_env('CI_PROJECT_NAME', 'gitlab-qa')
+          described_class.configure_rspec
+        end
+
+        it 'runs on master pipelines' do
+          group = describe_successfully do
+            it('runs on master pipeline given a single pipeline', only: { pipeline: :master } ) {}
+            it('runs in master given an array of pipelines', only: { pipeline: [:canary, :master] }) {}
+            it('does not run in non-master pipelines', only: { pipeline: [:nightly, :not_nightly, :not_master] } ) {}
+          end
+
+          aggregate_failures do
+            expect(group.examples[0].execution_result.status).to eq(:passed)
+            expect(group.examples[1].execution_result.status).to eq(:passed)
+            expect(group.examples[2].execution_result.status).to eq(:pending)
+          end
+        end
+      end
+
+      context 'with CI_PROJECT_NAME set' do
+        before do
+          stub_env('CI_PROJECT_NAME', 'NIGHTLY')
+          described_class.configure_rspec
+        end
+
+        it 'runs on designated pipeline' do
+          group = describe_successfully do
+            it('runs on nightly', only: { pipeline: :nightly } ) {}
+            it('does not run in not_nightly', only: { pipeline: :not_nightly } ) {}
+            it('runs on nightly given an array', only: { pipeline: [:canary, :nightly] }) {}
+            it('does not run in not_nightly given an array', only: { pipeline: [:not_nightly, :canary] }) {}
+          end
+
+          aggregate_failures do
+            expect(group.examples[0].execution_result.status).to eq(:passed)
+            expect(group.examples[1].execution_result.status).to eq(:pending)
+            expect(group.examples[2].execution_result.status).to eq(:passed)
+            expect(group.examples[3].execution_result.status).to eq(:pending)
+          end
+        end
+      end
+    end
   end
 end