summaryrefslogtreecommitdiffstats
path: root/test/conftest.py
blob: cf6d721be510e5f68cefad735591d8e5e89ca63d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import pytest
from .utils import (HOST, USER, PASSWORD, PORT, CHARSET, create_db,
                    db_connection, SSH_USER, SSH_HOST, SSH_PORT)
import mycli.sqlexecute


@pytest.yield_fixture(scope="function")
def connection():
    create_db('_test_db')
    connection = db_connection('_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='_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
    )