Sith/sith/pytest.py

import pytest

from .composer import start_composer, stop_composer
from .settings import PROCFILE_PYTEST

# it's the first hook loaded by pytest and can only
# be defined in a proper pytest plugin
# To use the composer before pytest-django loads
# we need to define this hook and thus create a proper
# pytest plugin. We can't just use conftest.py


@pytest.hookimpl(tryfirst=True)
def pytest_load_initial_conftests(early_config, parser, args):
    """Hook that loads the composer before the pytest-django plugin"""
    if PROCFILE_PYTEST is not None:
        start_composer(PROCFILE_PYTEST)


def pytest_unconfigure(config):
    stop_composer()