# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Gitlab::BackgroundMigration::DestroyInvalidGroupMembers, :migration, schema: 20220809002011 do
  # rubocop: disable Layout/LineLength
  # rubocop: disable RSpec/ScatteredLet
  let!(:migration_attrs) do
    {
      start_id: 1,
      end_id: 1000,
      batch_table: :members,
      batch_column: :id,
      sub_batch_size: 100,
      pause_ms: 0,
      connection: ApplicationRecord.connection
    }
  end

  let!(:migration) { described_class.new(**migration_attrs) }

  subject(:perform_migration) { migration.perform }

  let(:users_table) { table(:users) }
  let(:namespaces_table) { table(:namespaces) }
  let(:members_table) { table(:members) }
  let(:projects_table) { table(:projects) }

  let(:user1) { users_table.create!(name: 'user1', email: 'user1@example.com', projects_limit: 5) }
  let(:user2) { users_table.create!(name: 'user2', email: 'user2@example.com', projects_limit: 5) }
  let(:user3) { users_table.create!(name: 'user3', email: 'user3@example.com', projects_limit: 5) }
  let(:user4) { users_table.create!(name: 'user4', email: 'user4@example.com', projects_limit: 5) }
  let(:user5) { users_table.create!(name: 'user5', email: 'user5@example.com', projects_limit: 5) }
  let(:user6) { users_table.create!(name: 'user6', email: 'user6@example.com', projects_limit: 5) }

  let!(:group1) { namespaces_table.create!(name: 'marvellous group 1', path: 'group-path-1', type: 'Group') }

  let!(:group2) { namespaces_table.create!(name: 'outstanding group 2', path: 'group-path-2', type: 'Group') }

  # create group member records, a mix of both valid and invalid
  # project members will have already been filtered out.
  let!(:group_member1) { create_invalid_group_member(id: 1, user_id: user1.id) }

  let!(:group_member4) { create_valid_group_member(id: 4, user_id: user2.id, group_id: group1.id) }

  let!(:group_member5) { create_valid_group_member(id: 5, user_id: user3.id, group_id: group2.id) }

  let!(:group_member6) { create_invalid_group_member(id: 6, user_id: user4.id) }

  let!(:group_member7) { create_valid_group_member(id: 7, user_id: user5.id, group_id: group1.id) }

  let!(:group_member8) { create_invalid_group_member(id: 8, user_id: user6.id) }

  it 'removes invalid memberships but keeps valid ones', :aggregate_failures do
    expect(members_table.where(type: 'GroupMember').count).to eq 6

    queries = ActiveRecord::QueryRecorder.new do
      perform_migration
    end

    expect(queries.count).to eq(4)
    expect(members_table.where(type: 'GroupMember').pluck(:id)).to match_array([group_member4, group_member5, group_member7].map(&:id))
  end

  it 'tracks timings of queries' do
    expect(migration.batch_metrics.timings).to be_empty

    expect { perform_migration }.to change { migration.batch_metrics.timings }
  end

  it 'logs IDs of deleted records' do
    expect(Gitlab::AppLogger).to receive(:info).with({ message: 'Removing invalid group member records',
                                                       deleted_count: 3, ids: [group_member1, group_member6, group_member8].map(&:id) })

    perform_migration
  end

  def create_invalid_group_member(id:, user_id:)
    members_table.create!(id: id, user_id: user_id, source_id: non_existing_record_id, access_level: Gitlab::Access::MAINTAINER,
                          type: "GroupMember", source_type: "Namespace", notification_level: 3, member_namespace_id: nil)
  end

  def create_valid_group_member(id:, user_id:, group_id:)
    members_table.create!(id: id, user_id: user_id, source_id: group_id, access_level: Gitlab::Access::MAINTAINER,
                          type: "GroupMember", source_type: "Namespace", member_namespace_id: group_id, notification_level: 3)
  end
  # rubocop: enable Layout/LineLength
  # rubocop: enable RSpec/ScatteredLet
end