1 files changed, 22 insertions, 0 deletions
diff --git a/spec/models/ci/pending_build_spec.rb b/spec/models/ci/pending_build_spec.rb
index 4bb43233dbd..331522070df 100644
--- a/spec/models/ci/pending_build_spec.rb
+++ b/spec/models/ci/pending_build_spec.rb
@@ -196,6 +196,28 @@ RSpec.describe Ci::PendingBuild do
     end
   end
 
+  describe 'partitioning', :ci_partitionable do
+    include Ci::PartitioningHelpers
+
+    before do
+      stub_current_partition_id
+    end
+
+    let(:new_pipeline ) { create(:ci_pipeline, project: pipeline.project) }
+    let(:new_build) { create(:ci_build, pipeline: new_pipeline) }
+
+    it 'assigns the same partition id as the one that build has', :aggregate_failures do
+      expect(new_build.partition_id).to eq ci_testing_partition_id
+      expect(new_build.partition_id).not_to eq pipeline.partition_id
+
+      described_class.upsert_from_build!(build)
+      described_class.upsert_from_build!(new_build)
+
+      expect(build.reload.queuing_entry.partition_id).to eq pipeline.partition_id
+      expect(new_build.reload.queuing_entry.partition_id).to eq ci_testing_partition_id
+    end
+  end
+
   it_behaves_like 'cleanup by a loose foreign key' do
     let!(:parent) { create(:namespace) }
     let!(:model) { create(:ci_pending_build, namespace: parent) }