spec_helper.rb

# frozen_string_literal: true

$LOAD_PATH << File.expand_path(__dir__)

require 'spec_const'
require 'spec_condition'
require 'cli_helper'
require 'hide_pending_formatter'

require 'processing_for_ci'

require 'ridgepole'
require 'ridgepole/cli/config'
require 'active_support/core_ext'
require 'open3'
require 'tempfile'
require 'json'
require 'rspec/match_fuzzy'
require 'rspec/match_ruby'
require 'erbh'
require 'hash_modern_inspect'
require 'hash_order_helper'

require 'erb_helper'

RSpec.configure do |config|
  config.before(:all) do
    if condition(:debug) || condition(:verbose)
      ActiveRecord::Migration.verbose = true
      logger = Ridgepole::Logger.instance
      logger.level = ::Logger::DEBUG
      logger.verbose = condition(:verbose)
      ActiveRecord::Base.logger = logger
    else
      ActiveRecord::Migration.verbose = false
      Ridgepole::Logger.instance.level = ::Logger::ERROR
    end
  end

  config.before(:each) do |example|
    conds = example.metadata[:condition]

    if conds
      skip unless Array(conds).any? { |c| condition(c) }
    end

    case example.metadata[:file_path]
    when /mysql/
      skip if condition(:postgresql)
    when /postgresql/
      skip unless condition(:postgresql)
    end

    restore_database
  end
end

module SpecHelper
  def restore_database
    if condition(:postgresql)
      restore_database_postgresql
    else
      restore_database_mysql
    end
  end

  def system_raise_on_fail(*args)
    raise "Failed to run: #{args}" unless system(*args)
  end

  def restore_database_mysql
    sql_file = File.expand_path('mysql/ridgepole_test_database.sql', __dir__)
    system_raise_on_fail("#{MYSQL_CLI} < #{sql_file}")
  end

  def restore_database_postgresql
    sql_file = File.expand_path('postgresql/ridgepole_test_database.sql', __dir__)
    system("#{PG_CREATEDB} ridgepole_test 2>/dev/null")
    system_raise_on_fail("#{PG_PSQL} ridgepole_test --set ON_ERROR_STOP=off -q -f #{sql_file} 2>/dev/null")
  end

  def restore_tables
    if condition(:postgresql)
      restore_tables_postgresql
    else
      restore_tables_mysql
    end
  end

  def restore_tables_mysql
    sql_file = File.expand_path('mysql/ridgepole_test_tables.sql', __dir__)
    system_raise_on_fail("#{MYSQL_CLI} < #{sql_file}")
  end

  def restore_tables_postgresql
    sql_file = File.expand_path('postgresql/ridgepole_test_tables.sql', __dir__)
    system_raise_on_fail("#{PG_PSQL} ridgepole_test -q -f #{sql_file} 2>/dev/null")
  end

  def restore_tables_mysql_unknown_column_type
    sql_file = File.expand_path('mysql/ridgepole_test_tables_unknown_column_type.sql', __dir__)
    system_raise_on_fail("#{MYSQL_CLI} < #{sql_file}")
  end

  def client(options = {}, config = {})
    config = conn_spec(config)
    default_options = { debug: condition(:debug) }
    default_options[:dump_without_table_options] = true

    options = default_options.merge(options)

    Ridgepole::Client.new(config, options)
  end

  def conn_spec(config = {})
    if condition(:postgresql)
      {
        adapter: 'postgresql',
        database: TEST_SCHEMA,
        host: TEST_PG_HOST,
        port: TEST_PG_PORT,
        username: TEST_PG_USER,
        password: TEST_PG_PASS,
      }.merge(config)
    else
      {
        adapter: 'mysql2',
        database: TEST_SCHEMA,
        host: TEST_MYSQL_HOST,
        port: TEST_MYSQL_PORT,
        username: TEST_MYSQL_USER,
        password: TEST_MYSQL_PASS,
        ssl_mode: 'DISABLED',
      }.merge(config)
    end
  end

  def show_create_table(table_name)
    if condition(:postgresql)
      show_create_table_postgresql(table_name)
    else
      normalize_mysql_ddl(show_create_table_mysql(table_name))
    end
  end

  def show_create_table_mysql(table_name)
    raw_conn = ActiveRecord::Base.connection.raw_connection
    raw_conn.query("SHOW CREATE TABLE `#{table_name}`").first[1]
  end

  def show_create_table_postgresql(table_name)
    `#{PG_DUMP} --schema-only #{TEST_SCHEMA} --table=#{table_name} | awk '/^CREATE TABLE/,/);/{print} /^CREATE INDEX/{print}'`.strip
  end

  def normalize_mysql_ddl(ddl)
    ddl.gsub(/\(\d+\)/, '').gsub(/CHARSET=utf8$/, 'CHARSET=utf8mb3')
  end

  def tempfile(basename, content = '')
    path = `mktemp /tmp/#{basename}.XXXXXX`
    File.open(path, 'wb') { |f| f << content }
    FileUtils.chmod(0o777, path)
    yield(path)
  ensure
    FileUtils.rm_f(path) if path
  end

  def run_ridgepole(*args)
    Dir.chdir(File.expand_path('..', __dir__)) do
      cmd = [:bundle, :exec, './bin/ridgepole'] + args
      Open3.capture2e(cmd.join(' '))
    end
  end
end
include SpecHelper