gitlabhq/spec/migrations/move_ci_pipelines_to_dynamic_schema_spec.rb

# frozen_string_literal: true

require 'spec_helper'
require_migration!

RSpec.describe MoveCiPipelinesToDynamicSchema, :migration, feature_category: :continuous_integration do
  let(:migration) { described_class.new }
  let(:table_name) { described_class::TABLE_NAME }
  let(:identifier) { "#{described_class::DYNAMIC_SCHEMA}.#{table_name}" }

  describe '#up' do
    it 'moves the table into the dynamic schema' do
      expect(table_exists?(table_name)).to be_truthy

      migration.up

      expect(table_exists?(table_name)).to be_falsey
      expect(table_exists?(identifier)).to be_truthy
    end
  end

  describe '#down' do
    context 'when the partition exists in the dynamic schema' do
      before do
        migration.up

        drop_partition("#{identifier}_100")
        drop_partition("#{identifier}_101")
        drop_partition("#{identifier}_102")

        ApplicationRecord.connection.execute(<<~SQL)
          CREATE TABLE IF NOT EXISTS #{identifier} PARTITION OF p_#{table_name} FOR VALUES IN (100, 101, 102);
        SQL
      end

      it 'moves the table into the current schema' do
        expect(table_exists?(identifier)).to be_truthy

        migration.down

        expect(table_exists?(table_name)).to be_truthy
      end
    end

    context 'when the partition does not exist in the dynamic schema' do
      before do
        migration.up

        drop_partition(identifier)

        ApplicationRecord.connection.execute(<<~SQL)
          CREATE TABLE IF NOT EXISTS #{identifier}_100 PARTITION OF p_#{table_name} FOR VALUES IN (100);
          CREATE TABLE IF NOT EXISTS #{identifier}_101 PARTITION OF p_#{table_name} FOR VALUES IN (101);
          CREATE TABLE IF NOT EXISTS #{identifier}_102 PARTITION OF p_#{table_name} FOR VALUES IN (102);
        SQL
      end

      it 'creates the table into the current schema' do
        expect(table_exists?(identifier)).to be_falsey
        expect(table_exists?("#{identifier}_100")).to be_truthy
        expect(table_exists?("#{identifier}_101")).to be_truthy
        expect(table_exists?("#{identifier}_102")).to be_truthy

        migration.down

        expect(table_exists?(table_name)).to be_truthy
      end
    end
  end

  def table_exists?(name)
    ApplicationRecord.connection.table_exists?(name)
  end

  def drop_partition(name)
    return unless table_exists?(name)

    ApplicationRecord.connection.execute(<<~SQL)
      ALTER TABLE p_#{table_name} DETACH PARTITION #{name};
      DROP TABLE IF EXISTS #{name};
    SQL
  end
end