diff options
Diffstat (limited to 'spec/migrations/schedule_set_correct_vulnerability_state_spec.rb')
-rw-r--r-- | spec/migrations/schedule_set_correct_vulnerability_state_spec.rb | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/spec/migrations/schedule_set_correct_vulnerability_state_spec.rb b/spec/migrations/schedule_set_correct_vulnerability_state_spec.rb new file mode 100644 index 00000000000..08dccf1f37a --- /dev/null +++ b/spec/migrations/schedule_set_correct_vulnerability_state_spec.rb @@ -0,0 +1,33 @@ +# frozen_string_literal: true + +require 'spec_helper' + +require_migration! + +RSpec.describe ScheduleSetCorrectVulnerabilityState do + let_it_be(:migration) { described_class::MIGRATION_NAME } + + describe '#up' do + it 'schedules background jobs for each batch of vulnerabilities' do + migrate! + + expect(migration).to have_scheduled_batched_migration( + table_name: :vulnerabilities, + column_name: :id, + interval: described_class::JOB_INTERVAL, + batch_size: described_class::MAX_BATCH_SIZE, + batch_class_name: described_class::BATCH_CLASS_NAME, + sub_batch_size: described_class::SUB_BATCH_SIZE + ) + end + end + + describe '#down' do + it 'deletes all batched migration records' do + migrate! + schema_migrate_down! + + expect(migration).not_to have_scheduled_batched_migration + end + end +end |