Hands-On: Python Test Framework

"""Complete conftest.py for AEB ECU test project."""
import pytest, time

def pytest_addoption(parser):
    parser.addoption("--can-channel",  default="mock")
    parser.addoption("--dbc",          default="vehicle.dbc")
    parser.addoption("--uds-txid",     default="0x7E0")
    parser.addoption("--uds-rxid",     default="0x7E8")

@pytest.fixture(scope="session")
def ecu(request):
    channel = request.config.getoption("--can-channel")
    if channel == "mock":
        from lib.mock_ecu import MockAebEcu
        yield MockAebEcu()
    else:
        from lib.can_ecu import CanAebEcu
        dbc = request.config.getoption("--dbc")
        iface = CanAebEcu(channel, dbc)
        yield iface
        iface.shutdown()

@pytest.fixture(scope="function")
def clean_ecu(ecu):
    """Reset ECU to clean state; clear DTCs."""
    ecu.reset()
    time.sleep(0.1)
    yield ecu
    ecu.clear_dtcs()

@pytest.fixture(scope="session")
def uds(request, ecu):
    """UDS diagnostic client bound to same session as ecu."""
    tx = int(request.config.getoption("--uds-txid"), 16)
    rx = int(request.config.getoption("--uds-rxid"), 16)
    return ecu.get_uds_client(tx, rx)