From f7f60ab54ab69fb4d0c3a43406a9809edab7d762 Mon Sep 17 00:00:00 2001
From: Shinya Maeda <shinya@gitlab.com>
Date: Fri, 1 Jun 2018 14:53:00 +0900
Subject: [PATCH] Add spec for variables expressions with pipeline iid

---
 .../gitlab/ci/pipeline/chain/populate_spec.rb | 45 +++++++++++++------
 1 file changed, 32 insertions(+), 13 deletions(-)

diff --git a/spec/lib/gitlab/ci/pipeline/chain/populate_spec.rb b/spec/lib/gitlab/ci/pipeline/chain/populate_spec.rb
index feed7728f5a..6b18c615430 100644
--- a/spec/lib/gitlab/ci/pipeline/chain/populate_spec.rb
+++ b/spec/lib/gitlab/ci/pipeline/chain/populate_spec.rb
@@ -156,22 +156,41 @@ describe Gitlab::Ci::Pipeline::Chain::Populate do
     end
   end
 
-  context 'when using only/except build policies' do
-    let(:config) do
-      { rspec: { script: 'rspec', stage: 'test', only: ['master'] },
-        prod: { script: 'cap prod', stage: 'deploy', only: ['tags'] } }
-    end
+  context 'when variables policy is specified' do
+    context 'when using only/except build policies' do
+      let(:config) do
+        { rspec: { script: 'rspec', stage: 'test', only: ['master'] },
+          prod: { script: 'cap prod', stage: 'deploy', only: ['tags'] } }
+      end
 
-    let(:pipeline) do
-      build(:ci_pipeline, ref: 'master', config: config)
-    end
+      let(:pipeline) do
+        build(:ci_pipeline, ref: 'master', config: config)
+      end
 
-    it 'populates pipeline according to used policies' do
-      step.perform!
+      it 'populates pipeline according to used policies' do
+        step.perform!
 
-      expect(pipeline.stages.size).to eq 1
-      expect(pipeline.stages.first.builds.size).to eq 1
-      expect(pipeline.stages.first.builds.first.name).to eq 'rspec'
+        expect(pipeline.stages.size).to eq 1
+        expect(pipeline.stages.first.builds.size).to eq 1
+        expect(pipeline.stages.first.builds.first.name).to eq 'rspec'
+      end
+
+      context 'when variables expression is specified' do
+        let(:config) do
+          { rspec: { script: 'rspec', only: { variables: ["$CI_PIPELINE_IID == '1'"] } },
+            prod: { script: 'cap prod', only: { variables: ["$CI_PIPELINE_IID == '1000'"] } } }
+        end
+
+        context 'when pipeline iid is the subject' do
+          it 'populates pipeline according to used policies' do
+            step.perform!
+
+            expect(pipeline.stages.size).to eq 1
+            expect(pipeline.stages.first.builds.size).to eq 1
+            expect(pipeline.stages.first.builds.first.name).to eq 'rspec'
+          end 
+        end
+      end
     end
   end
 end