module ActiveRecord::MigrationChecker

def prepare_test_db
  current_config = ::ActiveRecord::Base.connection_db_config
  all_configs = ::ActiveRecord::Base.configurations.configs_for(env_name: Jets.env)
  needs_update = !all_configs.all? do |db_config|
    ::ActiveRecord::Tasks::DatabaseTasks.schema_up_to_date?(
      db_config.configuration_hash, 
      ::ActiveRecord::Base.schema_format, 
      nil
    )
  end
  if needs_update
    # Roundtrip to Rake to allow plugins to hook into database initialization.
    FileUtils.cd(Jets.root) do
      ::ActiveRecord::Base.clear_all_connections!
      system("jets db:test:prepare")
    end
  end
  # Establish a new connection, the old database may be gone (db:test:prepare uses purge)
  ::ActiveRecord::Base.establish_connection(current_config)
  begin
    check_pending!
  rescue ::ActiveRecord::PendingMigrationError
    puts "Migrations are pending. To resolve this issue, run:\n\n        jets db:migrate"
    exit 1
  end
end