d504165c80
Similar to `stages` and `sources` we need some basic infrastructure so that we can use a `mounts_module` fixture for the coming tests to the mount modules.
33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
import os
|
|
import pathlib
|
|
from types import ModuleType
|
|
|
|
import pytest
|
|
|
|
from osbuild import mounts, testutil
|
|
from osbuild.testutil.imports import import_module_from_path
|
|
|
|
|
|
@pytest.fixture(name="mounts_module")
|
|
def mounts_module_fixture(request: pytest.FixtureRequest) -> ModuleType:
|
|
"""mounts_module is a fixture that imports a stage module by its name
|
|
defined in MOUNTS_NAME in the test module.
|
|
"""
|
|
if not hasattr(request.module, "MOUNTS_NAME"):
|
|
raise ValueError("mounts_module fixture must be used in a test module that defines MOUNTS_NAME")
|
|
|
|
mounts_name = request.module.MOUNTS_NAME
|
|
caller_dir = pathlib.Path(request.node.fspath).parent
|
|
module_path = caller_dir.parent / mounts_name
|
|
return import_module_from_path("mounts", os.fspath(module_path))
|
|
|
|
|
|
@pytest.fixture
|
|
def mounts_service(mounts_module) -> ModuleType:
|
|
"""mounts_service is a fixture that imports a mounts module by its name
|
|
defined in MOUNTS_NAME in the test module and returns a MountsService
|
|
"""
|
|
service_cls = testutil.find_one_subclass_in_module(mounts_module, mounts.MountService)
|
|
fd = testutil.make_fake_service_fd()
|
|
srv_obj = service_cls.from_args(["--service-fd", str(fd)])
|
|
return srv_obj
|