summaryrefslogtreecommitdiff
path: root/spec/migrations/schedule_set_correct_vulnerability_state_spec.rb
blob: e888a1132c079f8ea2f483a946e72141747e308e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# frozen_string_literal: true

require 'spec_helper'

require_migration!

RSpec.describe ScheduleSetCorrectVulnerabilityState, feature_category: :vulnerability_management do
  let!(: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