1 files changed, 45 insertions, 0 deletions
diff --git a/spec/services/ci/runners/stale_machines_cleanup_service_spec.rb b/spec/services/ci/runners/stale_machines_cleanup_service_spec.rb
new file mode 100644
index 00000000000..456dbcebb84
--- /dev/null
+++ b/spec/services/ci/runners/stale_machines_cleanup_service_spec.rb
@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Ci::Runners::StaleMachinesCleanupService, feature_category: :runner_fleet do
+  let(:service) { described_class.new }
+  let!(:runner_machine3) { create(:ci_runner_machine, created_at: 6.months.ago, contacted_at: Time.current) }
+
+  subject(:response) { service.execute }
+
+  context 'with no stale runner machines' do
+    it 'does not clean any runner machines and returns :success status' do
+      expect do
+        expect(response).to be_success
+        expect(response.payload).to match({ deleted_machines: false })
+      end.not_to change { Ci::RunnerMachine.count }.from(1)
+    end
+  end
+
+  context 'with some stale runner machines' do
+    before do
+      create(:ci_runner_machine, :stale)
+      create(:ci_runner_machine, :stale, contacted_at: nil)
+    end
+
+    it 'only leaves non-stale runners' do
+      expect(response).to be_success
+      expect(response.payload).to match({ deleted_machines: true })
+      expect(Ci::RunnerMachine.all).to contain_exactly(runner_machine3)
+    end
+
+    context 'with more stale runners than MAX_DELETIONS' do
+      before do
+        stub_const("#{described_class}::MAX_DELETIONS", 1)
+      end
+
+      it 'only leaves non-stale runners' do
+        expect do
+          expect(response).to be_success
+          expect(response.payload).to match({ deleted_machines: true })
+        end.to change { Ci::RunnerMachine.count }.by(-Ci::Runners::StaleMachinesCleanupService::MAX_DELETIONS)
+      end
+    end
+  end
+end