Function cleanup_database

tests/conftest.py:373–396

(client, db_name, must_exist)

                       'characters.'.format(db_name))

  def cleanup_database(client, db_name, must_exist):
    # By default, dropping a database will move the files to the trash. Since the dev
    # environment uses a long fs.trash.interval, these files will accumulate and
    # waste disk space. To avoid that, we blast away the database directory (skipping
    # trash) before dropping the database.
    #
    # When there are external tables, the external locations are not removed by cascade.
    # These preexisting files/directories can cause errors when tests run repeatedly or
    # use a data snapshot (see IMPALA-9702). Deleting the database directory prevents
    # this as well.
    db_location = "{0}/{1}.db".format(WAREHOUSE, db_name)
    request.instance.filesystem_client.delete_file_dir(db_location, recursive=True)
    for i in range(2):
      try:
        result = client.execute('DROP DATABASE {0} `{1}` CASCADE'.format(
          "" if must_exist else "IF EXISTS", db_name))
        break
      except Exception as e:
        if i == 0:
          # Retry in case we hit IMPALA-14228.
          LOG.warn("Ignored cleanup failure once: " + str(e))
        else:
          raise e
    assert result.success

  def cleanup():
    with request.instance.create_impala_client(protocol=HS2) as client: