1 files changed, 80 insertions, 0 deletions
diff --git a/spec/services/deployments/archive_in_project_service_spec.rb b/spec/services/deployments/archive_in_project_service_spec.rb
new file mode 100644
index 00000000000..d4039ee7b4a
--- /dev/null
+++ b/spec/services/deployments/archive_in_project_service_spec.rb
@@ -0,0 +1,80 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Deployments::ArchiveInProjectService do
+  let_it_be(:project) { create(:project, :repository) }
+
+  let(:service) { described_class.new(project, nil) }
+
+  describe '#execute' do
+    subject { service.execute }
+
+    context 'when there are archivable deployments' do
+      let!(:deployments) { create_list(:deployment, 3, project: project) }
+      let!(:deployment_refs) { deployments.map(&:ref_path) }
+
+      before do
+        deployments.each(&:create_ref)
+        allow(Deployment).to receive(:archivables_in) { deployments }
+      end
+
+      it 'returns result code' do
+        expect(subject[:result]).to eq(:archived)
+        expect(subject[:status]).to eq(:success)
+        expect(subject[:count]).to eq(3)
+      end
+
+      it 'archives the deployment' do
+        expect(deployments.map(&:archived?)).to be_all(false)
+        expect(deployment_refs_exist?).to be_all(true)
+
+        subject
+
+        deployments.each(&:reload)
+        expect(deployments.map(&:archived?)).to be_all(true)
+        expect(deployment_refs_exist?).to be_all(false)
+      end
+
+      context 'when ref does not exist by some reason' do
+        before do
+          project.repository.delete_refs(*deployment_refs)
+        end
+
+        it 'does not raise an error' do
+          expect(deployment_refs_exist?).to be_all(false)
+
+          expect { subject }.not_to raise_error
+
+          expect(deployment_refs_exist?).to be_all(false)
+        end
+      end
+
+      context 'when deployments_archive feature flag is disabled' do
+        before do
+          stub_feature_flags(deployments_archive: false)
+        end
+
+        it 'does not do anything' do
+          expect(subject[:status]).to eq(:error)
+          expect(subject[:message]).to eq('Feature flag is not enabled')
+        end
+      end
+
+      def deployment_refs_exist?
+        deployment_refs.map { |path| project.repository.ref_exists?(path) }
+      end
+    end
+
+    context 'when there are no archivable deployments' do
+      before do
+        allow(Deployment).to receive(:archivables_in) { Deployment.none }
+      end
+
+      it 'returns result code' do
+        expect(subject[:result]).to eq(:empty)
+        expect(subject[:status]).to eq(:success)
+      end
+    end
+  end
+end