import inspect import os import pathlib from types import ModuleType import pytest from osbuild import sources, testutil from osbuild.testutil.imports import import_module_from_path @pytest.fixture(name="sources_module") def sources_module_fixture(request: pytest.FixtureRequest) -> ModuleType: """sources_module is a fixture that imports a stage module by its name defined in SOURCES_NAME in the test module. """ if not hasattr(request.module, "SOURCES_NAME"): raise ValueError("sources_module fixture must be used in a test module that defines SOURCES_NAME") sources_name = request.module.SOURCES_NAME caller_dir = pathlib.Path(request.node.fspath).parent module_path = caller_dir.parent / sources_name return import_module_from_path("sources", os.fspath(module_path)) @pytest.fixture def sources_service(sources_module) -> ModuleType: """sources_service is a fixture that imports a sources module by its name defined in SOURCES_NAME in the test module and returns a SourcesService """ service_cls = None for memb in inspect.getmembers( sources_module, predicate=lambda obj: inspect.isclass(obj) and issubclass( obj, sources.SourceService)): if service_cls: raise ValueError(f"already have {service_cls}, also found {memb}") service_cls = memb[1] fd = testutil.make_fake_service_fd() services_obj = service_cls.from_args(["--service-fd", str(fd)]) return services_obj