1 files changed, 52 insertions, 0 deletions
diff --git a/spec/workers/gitlab/export/prune_project_export_jobs_worker_spec.rb b/spec/workers/gitlab/export/prune_project_export_jobs_worker_spec.rb
new file mode 100644
index 00000000000..eded07c7a2f
--- /dev/null
+++ b/spec/workers/gitlab/export/prune_project_export_jobs_worker_spec.rb
@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::Export::PruneProjectExportJobsWorker, feature_category: :importers do
+  let_it_be(:old_job_1) { create(:project_export_job, updated_at: 37.months.ago) }
+  let_it_be(:old_job_2) { create(:project_export_job, updated_at: 12.months.ago) }
+  let_it_be(:old_job_3) { create(:project_export_job, updated_at: 8.days.ago) }
+  let_it_be(:fresh_job_1) { create(:project_export_job, updated_at: 1.day.ago) }
+  let_it_be(:fresh_job_2) { create(:project_export_job, updated_at: 2.days.ago) }
+  let_it_be(:fresh_job_3) { create(:project_export_job, updated_at: 6.days.ago) }
+
+  let_it_be(:old_relation_export_1) { create(:project_relation_export, project_export_job_id: old_job_1.id) }
+  let_it_be(:old_relation_export_2) { create(:project_relation_export, project_export_job_id: old_job_2.id) }
+  let_it_be(:old_relation_export_3) { create(:project_relation_export, project_export_job_id: old_job_3.id) }
+  let_it_be(:fresh_relation_export_1) { create(:project_relation_export, project_export_job_id: fresh_job_1.id) }
+
+  let_it_be(:old_upload_1) { create(:relation_export_upload, project_relation_export_id: old_relation_export_1.id) }
+  let_it_be(:old_upload_2) { create(:relation_export_upload, project_relation_export_id: old_relation_export_2.id) }
+  let_it_be(:old_upload_3) { create(:relation_export_upload, project_relation_export_id: old_relation_export_3.id) }
+  let_it_be(:fresh_upload_1) { create(:relation_export_upload, project_relation_export_id: fresh_relation_export_1.id) }
+
+  subject(:worker) { described_class.new }
+
+  describe '#perform' do
+    include_examples 'an idempotent worker' do
+      it 'prunes jobs and associations older than 7 days' do
+        expect { perform_multiple }.to change { ProjectExportJob.count }.by(-3)
+        expect(ProjectExportJob.find_by(id: old_job_1.id)).to be_nil
+        expect(ProjectExportJob.find_by(id: old_job_2.id)).to be_nil
+        expect(ProjectExportJob.find_by(id: old_job_3.id)).to be_nil
+
+        expect(Projects::ImportExport::RelationExport.find_by(id: old_relation_export_1.id)).to be_nil
+        expect(Projects::ImportExport::RelationExport.find_by(id: old_relation_export_2.id)).to be_nil
+        expect(Projects::ImportExport::RelationExport.find_by(id: old_relation_export_3.id)).to be_nil
+
+        expect(Projects::ImportExport::RelationExportUpload.find_by(id: old_upload_1.id)).to be_nil
+        expect(Projects::ImportExport::RelationExportUpload.find_by(id: old_upload_2.id)).to be_nil
+        expect(Projects::ImportExport::RelationExportUpload.find_by(id: old_upload_3.id)).to be_nil
+      end
+
+      it 'leaves fresh jobs and associations' do
+        perform_multiple
+        expect(fresh_job_1.reload).to be_present
+        expect(fresh_job_2.reload).to be_present
+        expect(fresh_job_3.reload).to be_present
+        expect(fresh_relation_export_1.reload).to be_present
+        expect(fresh_upload_1.reload).to be_present
+      end
+    end
+  end
+end