Function create_test_db

tests/mssqltestutils.py:93–128

Creates database for test, using various status checks and retry logic for reliability. - Calls helper method to check status of create db task, if possible - Exits on successful response or retry period exceeds time limit

()

    return  os.path.abspath(tempPath)

def create_test_db():
    """
    Creates database for test, using various status checks and retry logic for reliability.
    - Calls helper method to check status of create db task, if possible
    - Exits on successful response or retry period exceeds time limit
    """
    options = create_mssql_cli_options(database='master')
    client = create_mssql_cli_client(options)

    local_machine_name = socket.gethostname().replace(
        '-', '_').replace('.', '_')

    test_db_name = u'mssqlcli_testdb_{0}_{1}'.format(
        local_machine_name, random_str())
    query_db_create = u"CREATE DATABASE {0};".format(test_db_name)

    try:
        for _, _, status, _, is_create_error in client.execute_query(query_db_create):
            if _is_client_db_on_cloud(client):
                # retry logic is only supported for sql azure
                create_db_status, create_db_error = _check_create_db_status(test_db_name, client)

                if create_db_status == 'FAILED':
                    # break loop to assert db creation failure
                    raise AssertionError("Database creation failed. Retry logic for SQL " \
                                        "Azure DB was unsuccessful with the following error: " \
                                        "\n{}".format(create_db_error))

            if is_create_error:
                # break loop to assert db creation failure
                raise AssertionError("Database creation failed: {}".format(status))

        return test_db_name

    finally:
        shutdown(client)

def _is_client_db_on_cloud(client):
    """