1 files changed, 21 insertions, 0 deletions
diff --git a/spec/models/ci/pipeline_variable_spec.rb b/spec/models/ci/pipeline_variable_spec.rb
index 4e8d49585d0..fdcec0e96af 100644
--- a/spec/models/ci/pipeline_variable_spec.rb
+++ b/spec/models/ci/pipeline_variable_spec.rb
@@ -17,4 +17,25 @@ RSpec.describe Ci::PipelineVariable do
     it { is_expected.to be_a(Hash) }
     it { is_expected.to eq({ key: 'foo', value: 'bar' }) }
   end
+
+  describe 'partitioning' do
+    context 'with pipeline' do
+      let(:pipeline) { build(:ci_pipeline, partition_id: 123) }
+      let(:variable) { build(:ci_pipeline_variable, pipeline: pipeline, partition_id: nil) }
+
+      it 'copies the partition_id from pipeline' do
+        expect { variable.valid? }.to change(variable, :partition_id).from(nil).to(123)
+      end
+    end
+
+    context 'without pipeline' do
+      subject(:variable) { build(:ci_pipeline_variable, pipeline: nil, partition_id: nil) }
+
+      it { is_expected.to validate_presence_of(:partition_id) }
+
+      it 'does not change the partition_id value' do
+        expect { variable.valid? }.not_to change(variable, :partition_id)
+      end
+    end
+  end
 end