1 files changed, 6 insertions, 1 deletions
diff --git a/spec/models/ci/sources/pipeline_spec.rb b/spec/models/ci/sources/pipeline_spec.rb
index 707872d0a15..036708ed61e 100644
--- a/spec/models/ci/sources/pipeline_spec.rb
+++ b/spec/models/ci/sources/pipeline_spec.rb
@@ -6,6 +6,11 @@ RSpec.describe Ci::Sources::Pipeline, feature_category: :continuous_integration
   it { is_expected.to belong_to(:project) }
   it { is_expected.to belong_to(:pipeline) }
 
+  it do
+    is_expected.to belong_to(:build).class_name('Ci::Build')
+      .with_foreign_key(:source_job_id).inverse_of(:sourced_pipelines)
+  end
+
   it { is_expected.to belong_to(:source_project).class_name('::Project') }
   it { is_expected.to belong_to(:source_job) }
   it { is_expected.to belong_to(:source_bridge) }
@@ -32,7 +37,7 @@ RSpec.describe Ci::Sources::Pipeline, feature_category: :continuous_integration
     end
   end
 
-  describe 'partitioning', :ci_partitioning do
+  describe 'partitioning', :ci_partitionable do
     include Ci::PartitioningHelpers
 
     let(:new_pipeline) { create(:ci_pipeline) }