class Baza::InfoActiveRecord
  attr_reader :db

  def self.connection
    require "active_record"

    if RUBY_ENGINE == "jruby"
      require "/usr/share/java/mysql-connector-java.jar" if File.exists?("/usr/share/java/mysql-connector-java.jar")
      @conn_pool ||= ::ActiveRecord::Base.establish_connection(
        adapter: "mysql",
        host: "localhost",
        database: "baza",
        username: "shippa"
      )
      @conn ||= @conn_pool.connection
    else
      @conn_pool ||= ::ActiveRecord::Base.establish_connection(
        adapter: "mysql2",
        host: "localhost",
        database: "baza",
        username: "shippa"
      )
      @conn ||= @conn_pool.connection
    end

    return {pool: @conn_pool, conn: @conn}
  end

  def initialize
    data = Baza::InfoActiveRecord.connection

    @db = Baza::Db.new(
      type: :active_record,
      conn: data[:conn]
    )
  end

  def before
    @db.tables.list.each do |name, table|
      table.drop
    end
  end

  def after
    @db.close
  end
end