30 lines
790 B
Python
30 lines
790 B
Python
|
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('_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
|
||
|
|
)
|