class Gitlab::QA::Scenario::Test::Integration::Praefect

def omnibus_config_with_praefect
  <<~OMNIBUS
    gitaly['enable'] = true;
    gitaly['auth_token'] = 'praefect-gitaly-token';
    gitaly['storage'] = [
      {
        'name' => 'praefect-gitaly-0',
        'path' => '/var/opt/gitlab/git-data/repositories'
      }
    ];
    praefect['enable'] = true;
    praefect['listen_addr'] = '0.0.0.0:2305';
    praefect['auth_token'] = 'praefect-token';
    praefect['virtual_storages'] = {
      'default' => {
        'praefect-gitaly-0' => {
          'address' => 'unix:/var/opt/gitlab/gitaly/gitaly.socket',
          'token'   => 'praefect-gitaly-token',
          'primary' => true
        }
      }
    };
    praefect['database_host'] = '/var/opt/gitlab/postgresql';
    praefect['database_user'] = 'gitlab';
    praefect['database_dbname'] = 'praefect_production';
    praefect['postgres_queue_enabled'] = true;
    gitlab_rails['gitaly_token'] = 'praefect-token';
    git_data_dirs({
      'default' => {
        'gitaly_address' => 'tcp://localhost:2305'
      }
    });
  OMNIBUS
end

def perform(release, *rspec_args)
  Docker::Volumes.new.with_temporary_volumes do |volumes|
    # Create the Praefect database before enabling Praefect
    Component::Gitlab.perform do |gitlab|
      gitlab.release = Release.new(release)
      gitlab.name = 'gitlab'
      gitlab.network = 'test'
      gitlab.volumes = volumes
      gitlab.exec_commands = ['gitlab-psql -d template1 -c "CREATE DATABASE praefect_production OWNER gitlab"']
      gitlab.act(omnibus_config_with_praefect) do |new_config|
        prepare
        start
        reconfigure
        process_exec_commands
        wait
        teardown
      end
    end
    # Restart GitLab with Praefect enabled and then run the tests
    Component::Gitlab.perform do |gitlab|
      gitlab.release = Release.new(release)
      gitlab.name = 'gitlab'
      gitlab.network = 'test'
      gitlab.volumes = volumes
      gitlab.omnibus_config = omnibus_config_with_praefect
      gitlab.act do
        start
        reconfigure
        wait
        puts "Running Praefect specs!"
        Component::Specs.perform do |specs|
          specs.suite = 'Test::Instance::All'
          specs.release = gitlab.release
          specs.network = gitlab.network
          specs.args = [gitlab.address, *rspec_args]
          specs.env = { QA_PRAEFECT_REPOSITORY_STORAGE: 'default' }
        end
        teardown
      end
    end
  end
end