Add exclusive lease for stuck_ci_builds_worker

1 files changed, 16 insertions, 1 deletions

diff --git a/spec/workers/stuck_ci_builds_worker_spec.rb b/spec/workers/stuck_ci_builds_worker_spec.rb
index e2386336447..c88a0dfa4a6 100644
--- a/spec/workers/stuck_ci_builds_worker_spec.rb
+++ b/spec/workers/stuck_ci_builds_worker_spec.rb
@@ -10,7 +10,10 @@ describe StuckCiBuildsWorker do
     build.status
   end
 
-  before { build.update!(status: status, updated_at: updated_at) }
+  before do
+    build.update!(status: status, updated_at: updated_at)
+    allow_any_instance_of(Gitlab::ExclusiveLease).to receive(:try_obtain).and_return(true)
+  end
 
   shared_examples 'build is dropped' do
     it 'changes status' do
@@ -96,4 +99,16 @@ describe StuckCiBuildsWorker do
       worker.perform
     end
   end
+
+  describe 'exclusive lease' do
+    let(:status) { 'running' }
+    let(:updated_at) { 2.days.ago }
+
+    it 'is guard by exclusive lease' do
+      worker.perform
+      allow_any_instance_of(Gitlab::ExclusiveLease).to receive(:try_obtain).and_return(false)
+      expect(worker).not_to receive(:drop)
+      worker.perform
+    end
+  end
 end