mycli/test/conftest.py

import pytest
from .utils import HOST, USER, PASSWORD, PORT, CHARSET, create_db, db_connection, SSH_USER, SSH_HOST, SSH_PORT
import mycli.sqlexecute


@pytest.fixture(scope="function")
def connection():
    create_db("mycli_test_db")
    connection = db_connection("mycli_test_db")
    yield connection

    connection.close()


@pytest.fixture
def cursor(connection):
    with connection.cursor() as cur:
        return cur


@pytest.fixture
def executor(connection):
    return mycli.sqlexecute.SQLExecute(
        database="mycli_test_db",
        user=USER,
        host=HOST,
        password=PASSWORD,
        port=PORT,
        socket=None,
        charset=CHARSET,
        local_infile=False,
        ssl=None,
        ssh_user=SSH_USER,
        ssh_host=SSH_HOST,
        ssh_port=SSH_PORT,
        ssh_password=None,
        ssh_key_filename=None,
    )