1 files changed, 17 insertions, 1 deletions
diff --git a/spec/models/ci/sources/pipeline_spec.rb b/spec/models/ci/sources/pipeline_spec.rb
index fdc1c111c40..707872d0a15 100644
--- a/spec/models/ci/sources/pipeline_spec.rb
+++ b/spec/models/ci/sources/pipeline_spec.rb
@@ -2,7 +2,7 @@
 
 require 'spec_helper'
 
-RSpec.describe Ci::Sources::Pipeline do
+RSpec.describe Ci::Sources::Pipeline, feature_category: :continuous_integration do
   it { is_expected.to belong_to(:project) }
   it { is_expected.to belong_to(:pipeline) }
 
@@ -31,4 +31,20 @@ RSpec.describe Ci::Sources::Pipeline do
       let!(:model) { create(:ci_sources_pipeline, project: parent) }
     end
   end
+
+  describe 'partitioning', :ci_partitioning do
+    include Ci::PartitioningHelpers
+
+    let(:new_pipeline) { create(:ci_pipeline) }
+    let(:source_pipeline) { create(:ci_sources_pipeline, pipeline: new_pipeline) }
+
+    before do
+      stub_current_partition_id
+    end
+
+    it 'assigns partition_id and source_partition_id from pipeline and source_job', :aggregate_failures do
+      expect(source_pipeline.partition_id).to eq(ci_testing_partition_id)
+      expect(source_pipeline.source_partition_id).to eq(ci_testing_partition_id)
+    end
+  end
 end