debian-mirror-gitlab/spec/support_specs/helpers/migrations_helpers_spec.rb

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe MigrationsHelpers, feature_category: :database do
  let(:helper_class) do
    Class.new.tap do |klass|
      klass.include described_class
      allow(klass).to receive(:metadata).and_return(metadata)
    end
  end

  let(:metadata) { {} }
  let(:helper) { helper_class.new }

  describe '#active_record_base' do
    it 'returns the main base model' do
      expect(helper.active_record_base).to eq(ActiveRecord::Base)
    end

    context 'ci database configured' do
      before do
        skip_if_multiple_databases_not_setup(:ci)
      end

      it 'returns the CI base model' do
        expect(helper.active_record_base(database: :ci)).to eq(Ci::ApplicationRecord)
      end
    end

    context 'ci database not configured' do
      before do
        skip_if_multiple_databases_are_setup(:ci)
      end

      it 'returns the main base model' do
        expect(helper.active_record_base(database: :ci)).to eq(ActiveRecord::Base)
      end
    end

    it 'raises ArgumentError for bad database argument' do
      expect { helper.active_record_base(database: :non_existent) }.to raise_error(ArgumentError)
    end
  end

  describe '#table' do
    it 'creates a class based on main base model' do
      klass = helper.table(:projects)
      expect(klass.connection_specification_name).to eq('ActiveRecord::Base')
    end

    context 'ci database configured' do
      before do
        skip_if_multiple_databases_not_setup(:ci)
      end

      it 'create a class based on the CI base model' do
        klass = helper.table(:ci_builds, database: :ci)
        expect(klass.connection_specification_name).to eq('Ci::ApplicationRecord')
      end
    end

    context 'ci database not configured' do
      before do
        skip_if_multiple_databases_are_setup(:ci)
      end

      it 'creates a class based on main base model' do
        klass = helper.table(:ci_builds, database: :ci)
        expect(klass.connection_specification_name).to eq('ActiveRecord::Base')
      end
    end
  end

  describe '#reset_column_information' do
    context 'with a regular ActiveRecord model class' do
      let(:klass) { Project }

      it 'calls reset_column_information' do
        expect(klass).to receive(:reset_column_information)

        helper.reset_column_information(klass)
      end
    end

    context 'with an anonymous class with table name defined' do
      let(:klass) do
        Class.new(ActiveRecord::Base) do
          self.table_name = :projects
        end
      end

      it 'calls reset_column_information' do
        expect(klass).to receive(:reset_column_information)

        helper.reset_column_information(klass)
      end
    end

    context 'with an anonymous class with no table name defined' do
      let(:klass) { Class.new(ActiveRecord::Base) }

      it 'does not call reset_column_information' do
        expect(klass).not_to receive(:reset_column_information)

        helper.reset_column_information(klass)
      end
    end
  end
end