conftest.py

"""For testing."""

import sys

import pytest
from twisted.python import log

from pywwa import CTX, CTX_DEFAULTS
from pywwa.database import get_dbconnc

# Dragons
# pytest + twisted + click == too much magic for poor me
# So, this effectively disables reactor.stop() from working
CTX_DEFAULTS["shutdown_delay"] = 1800
log.startLogging(sys.stdout)


@pytest.fixture(autouse=True)
def setup_and_teardown():
    """Something to fix the hacky global namespace mucking I do."""
    CTX.update(CTX_DEFAULTS)
    yield
    CTX.update(CTX_DEFAULTS)


@pytest.fixture(scope="function")
def cursor(database):
    """Return a disposable database cursor."""
    pgconn, cursor = get_dbconnc(database)
    yield cursor
    pgconn.close()