diff options
Diffstat (limited to 'spec/migrations/schedule_disable_expiration_policies_linked_to_no_container_images_spec.rb')
-rw-r--r-- | spec/migrations/schedule_disable_expiration_policies_linked_to_no_container_images_spec.rb | 47 |
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 |