Improve trigger_schedule.rb

1 files changed, 21 insertions, 2 deletions

diff --git a/spec/workers/trigger_schedule_worker_spec.rb b/spec/workers/trigger_schedule_worker_spec.rb
index 3997369489f..c48f1406c34 100644
--- a/spec/workers/trigger_schedule_worker_spec.rb
+++ b/spec/workers/trigger_schedule_worker_spec.rb
@@ -8,10 +8,12 @@ describe TriggerScheduleWorker do
   end
 
   context 'when there is a scheduled trigger within next_run_at' do
-    let!(:trigger_schedule) { create(:ci_trigger_schedule, :nightly, :force_triggable) }
-    let(:next_time) { Gitlab::Ci::CronParser.new(trigger_schedule.cron, trigger_schedule.cron_timezone).next_time_from(Time.now) }
+    let!(:trigger_schedule) { create(:ci_trigger_schedule, :nightly) }
+    let(:next_time) { Gitlab::Ci::CronParser.new(trigger_schedule.cron, trigger_schedule.cron_timezone).next_time_from(@time_future) }
 
     before do
+      @time_future = Time.now + 10.days
+      allow(Time).to receive(:now).and_return(@time_future)
       worker.perform
     end
 
@@ -43,4 +45,21 @@ describe TriggerScheduleWorker do
       expect(trigger_schedule.next_run_at).to eq(Ci::TriggerSchedule.last.next_run_at)
     end
   end
+
+  context 'when next_run_at is nil' do
+    let!(:trigger_schedule) { create(:ci_trigger_schedule, :nightly) }
+
+    before do
+      trigger_schedule.update_attribute(:next_run_at, nil)
+      worker.perform
+    end
+
+    it 'does not create a new pipeline' do
+      expect(Ci::Pipeline.count).to eq(0)
+    end
+
+    it 'does not update next_run_at' do
+      expect(trigger_schedule.next_run_at).to eq(Ci::TriggerSchedule.last.next_run_at)
+    end
+  end
 end