Expose pipeline stage seeds from pipeline instance

2 files changed, 18 insertions, 5 deletions

diff --git a/app/models/ci/pipeline.rb b/app/models/ci/pipeline.rb
index d5b6da4eee6..155fefcc9c3 100644
--- a/app/models/ci/pipeline.rb
+++ b/app/models/ci/pipeline.rb
@@ -296,11 +296,13 @@ module Ci
     end
 
     def stage_seeds
-      return [] unless config_processor
+      return unless config_processor
+
+      seeds_scope = { ref: ref, tag: tag?, trigger: trigger_requests.first }
 
-      config_processor.stage_seeds(ref: ref,
-                                   tag: tag?,
-                                   trigger: trigger_requests.first)
+      config_processor.stage_seeds(seeds_scope).tap do |seeds|
+        seeds.pipeline = self
+      end
     end
 
     def has_stages?
@@ -312,7 +314,7 @@ module Ci
     end
 
     def config_processor
-      return nil unless ci_yaml_file
+      return unless ci_yaml_file
       return @config_processor if defined?(@config_processor)
 
       @config_processor ||= begin
diff --git a/spec/models/ci/pipeline_spec.rb b/spec/models/ci/pipeline_spec.rb
index c4f07c4a693..3a1fe666ff0 100644
--- a/spec/models/ci/pipeline_spec.rb
+++ b/spec/models/ci/pipeline_spec.rb
@@ -202,6 +202,17 @@ describe Ci::Pipeline, models: true do
                              status: 'success')
     end
 
+    describe '#stage_seeds' do
+      let(:pipeline) do
+        create(:ci_pipeline, config: { rspec: { script: 'rake' } })
+      end
+
+      it 'returns preseeded stage seeds object' do
+        expect(pipeline.stage_seeds).to be_a Gitlab::Ci::Stage::Seeds
+        expect(pipeline.stage_seeds.stages).to all(include(pipeline: pipeline))
+      end
+    end
+
     describe '#stages' do
       subject { pipeline.stages }