devices: add pytest conftest infra for devices

This is similar to the other `inputs`, `sources` etc conftest.py
modules and allows us to write unit test for the devices code.
This commit is contained in:
Michael Vogt 2024-08-21 16:55:04 +02:00 committed by Achilleas Koutsou
parent 6ba0561187
commit 6788a2c840
2 changed files with 34 additions and 1 deletions

33
devices/test/conftest.py Normal file
View file

@ -0,0 +1,33 @@
import os
import pathlib
from types import ModuleType
import pytest
from osbuild import devices, testutil
from osbuild.testutil.imports import import_module_from_path
@pytest.fixture(name="devices_module")
def devices_module_fixture(request: pytest.FixtureRequest) -> ModuleType:
"""devices_module is a fixture that imports a stage module by its name
defined in DEVICES_NAME in the test module.
"""
if not hasattr(request.module, "DEVICES_NAME"):
raise ValueError("devices_module fixture must be used in a test module that defines DEVICES_NAME")
devices_name = request.module.DEVICES_NAME
caller_dir = pathlib.Path(request.node.fspath).parent
module_path = caller_dir.parent / devices_name
return import_module_from_path("devices", os.fspath(module_path))
@pytest.fixture
def devices_service(devices_module) -> ModuleType:
"""devices_service is a fixture that imports a devices module by its name
defined in DEVICES_NAME in the test module and returns a Deviceservice
"""
service_cls = testutil.find_one_subclass_in_module(devices_module, devices.DeviceService)
fd = testutil.make_fake_service_fd()
srv_obj = service_cls.from_args(["--service-fd", str(fd)])
return srv_obj

View file

@ -24,7 +24,7 @@ deps =
requests requests
setenv = setenv =
LINTABLES = osbuild/ assemblers/* devices/* inputs/*.* mounts/*.* mounts/test/*.py runners/* sources/*.* stages/*.* inputs/test/*.py stages/test/*.py sources/test/*.py test/ tools/ LINTABLES = osbuild/ assemblers/* devices/*.* devices/test/*.py inputs/*.* mounts/*.* mounts/test/*.py runners/* sources/*.* stages/*.* inputs/test/*.py stages/test/*.py sources/test/*.py test/ tools/
LINTABLES_EXCLUDES = "*.json,*.sh" LINTABLES_EXCLUDES = "*.json,*.sh"
LINTABLES_EXCLUDES_RE = ".*\.json$,.*\.sh" LINTABLES_EXCLUDES_RE = ".*\.json$,.*\.sh"
TYPEABLES = osbuild TYPEABLES = osbuild