# frozen_string_literal: true require 'spec_helper' require_migration! RSpec.describe FinalizeInvalidMemberCleanup, :migration, feature_category: :subgroups do let(:batched_migrations) { table(:batched_background_migrations) } let!(:migration) { described_class::MIGRATION } describe '#up' do shared_examples 'finalizes the migration' do it 'finalizes the migration' do allow_next_instance_of(Gitlab::Database::BackgroundMigration::BatchedMigrationRunner) do |runner| expect(runner).to receive(:finalize).with('DestroyInvalidMembers', :members, :id, []) end end end context 'when migration is missing' do before do batched_migrations.where(job_class_name: migration).delete_all end it 'warns migration not found' do expect(Gitlab::AppLogger) .to receive(:warn).with(/Could not find batched background migration for the given configuration:/) migrate! end end context 'with migration present' do let!(:destroy_invalid_member_migration) do batched_migrations.create!( job_class_name: 'DestroyInvalidMembers', table_name: :members, column_name: :id, job_arguments: [], interval: 2.minutes, min_value: 1, max_value: 2, batch_size: 1000, sub_batch_size: 200, gitlab_schema: :gitlab_main, status: 3 # finished ) end context 'when migration finished successfully' do it 'does not raise exception' do expect { migrate! }.not_to raise_error end end context 'with different migration statuses' do using RSpec::Parameterized::TableSyntax where(:status, :description) do 0 | 'paused' 1 | 'active' 4 | 'failed' 5 | 'finalizing' end with_them do before do destroy_invalid_member_migration.update!(status: status) end it_behaves_like 'finalizes the migration' end end end end end