debian-mirror-gitlab/spec/migrations/finalize_invalid_member_cleanup_spec.rb

# 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