mirror of https://github.com/dbcli/pgcli
55 lines
1.1 KiB
Python
55 lines
1.1 KiB
Python
import os
|
|
import pytest
|
|
from utils import (
|
|
POSTGRES_HOST,
|
|
POSTGRES_PORT,
|
|
POSTGRES_USER,
|
|
POSTGRES_PASSWORD,
|
|
create_db,
|
|
db_connection,
|
|
drop_tables,
|
|
)
|
|
import pgcli.main
|
|
import pgcli.pgexecute
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def connection():
|
|
create_db("_test_db")
|
|
connection = db_connection("_test_db")
|
|
yield connection
|
|
|
|
drop_tables(connection)
|
|
connection.close()
|
|
|
|
|
|
@pytest.fixture
|
|
def cursor(connection):
|
|
with connection.cursor() as cur:
|
|
return cur
|
|
|
|
|
|
@pytest.fixture
|
|
def executor(connection):
|
|
return pgcli.pgexecute.PGExecute(
|
|
database="_test_db",
|
|
user=POSTGRES_USER,
|
|
host=POSTGRES_HOST,
|
|
password=POSTGRES_PASSWORD,
|
|
port=POSTGRES_PORT,
|
|
dsn=None,
|
|
notify_callback=pgcli.main.notify_callback,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def exception_formatter():
|
|
return lambda e: str(e)
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def temp_config(tmpdir_factory):
|
|
# this function runs on start of test session.
|
|
# use temporary directory for config home so user config will not be used
|
|
os.environ["XDG_CONFIG_HOME"] = str(tmpdir_factory.mktemp("data"))
|