1 files changed, 47 insertions, 0 deletions
diff --git a/spec/migrations/schedule_disable_expiration_policies_linked_to_no_container_images_spec.rb b/spec/migrations/schedule_disable_expiration_policies_linked_to_no_container_images_spec.rb
new file mode 100644
index 00000000000..888d306f893
--- /dev/null
+++ b/spec/migrations/schedule_disable_expiration_policies_linked_to_no_container_images_spec.rb
@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+require_migration!
+
+RSpec.describe ScheduleDisableExpirationPoliciesLinkedToNoContainerImages do
+  let_it_be(:projects) { table(:projects) }
+  let_it_be(:container_expiration_policies) { table(:container_expiration_policies) }
+  let_it_be(:container_repositories) { table(:container_repositories) }
+  let_it_be(:namespaces) { table(:namespaces) }
+  let_it_be(:namespace) { namespaces.create!(name: 'test', path: 'test') }
+
+  let_it_be(:policy1) { create_expiration_policy(id: 1, enabled: true) }
+  let_it_be(:policy2) { create_expiration_policy(id: 2, enabled: false) }
+  let_it_be(:policy3) { create_expiration_policy(id: 3, enabled: false) }
+  let_it_be(:policy4) { create_expiration_policy(id: 4, enabled: true) }
+  let_it_be(:policy5) { create_expiration_policy(id: 5, enabled: false) }
+  let_it_be(:policy6) { create_expiration_policy(id: 6, enabled: false) }
+  let_it_be(:policy7) { create_expiration_policy(id: 7, enabled: true) }
+  let_it_be(:policy8) { create_expiration_policy(id: 8, enabled: true) }
+  let_it_be(:policy9) { create_expiration_policy(id: 9, enabled: true) }
+
+  it 'schedules background migrations', :aggregate_failures do
+    stub_const("#{described_class}::BATCH_SIZE", 2)
+
+    Sidekiq::Testing.fake! do
+      freeze_time do
+        migrate!
+
+        expect(described_class::MIGRATION).to be_scheduled_delayed_migration(2.minutes, 1, 4)
+        expect(described_class::MIGRATION).to be_scheduled_delayed_migration(4.minutes, 7, 8)
+        expect(described_class::MIGRATION).to be_scheduled_delayed_migration(6.minutes, 9, 9)
+
+        expect(BackgroundMigrationWorker.jobs.size).to eq(3)
+      end
+    end
+  end
+
+  def create_expiration_policy(id:, enabled:)
+    project = projects.create!(id: id, namespace_id: namespace.id, name: "gitlab-#{id}")
+    container_expiration_policies.create!(
+      enabled: enabled,
+      project_id: project.id
+    )
+  end
+end