callable_marker: add type hints

Signed-off-by: Gaëtan Lehmann <[email protected]>
Co-Authored-By: Yann Dirson <[email protected]>

Author: glehmann

lib/common.py

@@ -9,7 +9,7 @@
 import traceback
 from enum import Enum
 from pydantic import TypeAdapter, ValidationError
-from typing import Any, Dict, Literal, Optional, Type, TypeVar, cast, overload, TYPE_CHECKING, Union
+from typing import Any, Callable, Dict, Literal, Optional, Type, TypeVar, cast, overload, TYPE_CHECKING, Union
 from uuid import UUID
 
 import pytest
@@ -90,7 +90,7 @@ def ensure_type(typ: Type[T], value: Any) -> T:
             return ta.validate_python(value)
     raise TypeError(f"'{type(value).__name__}' object is not of the expected type '{typ.__name__}'")
 
-def callable_marker(value, request):
+def callable_marker(value: Union[T, Callable[..., T]], request: pytest.FixtureRequest) -> T:
     """
     Process value optionally generated by fixture-dependent callable.
 
@@ -106,8 +106,12 @@ def callable_marker(value, request):
                       for arg_name in inspect.getfullargspec(value).args}
         except pytest.FixtureLookupError as e:
             raise RuntimeError("fixture in mapping not found on test") from e
-        value = value(**params)
-    return value
+        # callable ensures the value is of type Callable[..., object], which is not enough in that case
+        # we can trust the static checker though, and thus use cast
+        fn = cast(Callable[..., T], value)
+        return fn(**params)
+    else:
+        return value
 
 def wait_for(fn, msg=None, timeout_secs=2 * 60, retry_delay_secs=2, invert=False):
     if msg is not None:

tests/install/conftest.py

@@ -1,13 +1,15 @@
+from __future__ import annotations
+
 import logging
 import os
-from typing import Sequence
+from typing import Callable, Generator, Sequence, Union
 import pytest
 import pytest_dependency        # type: ignore
 import tempfile
 import xml.etree.ElementTree as ET
 
 from lib import installer, pxe
-from lib.common import callable_marker, url_download, wait_for
+from lib.common import callable_marker, ensure_type, url_download, wait_for
 from lib.installer import AnswerFile
 from lib.commands import local_cmd
 
@@ -39,7 +41,7 @@ def skip_package_source(version, package_source):
     return True, "unknown source type {}".format(package_source)
 
 @pytest.fixture(scope='function')
-def answerfile(request):
+def answerfile(request: pytest.FixtureRequest) -> Generator[Union[AnswerFile, None], None, None]:
     """
     Makes an AnswerFile object available to test and other fixtures.
 
@@ -66,8 +68,8 @@ def answerfile(request):
         return
 
     # construct answerfile definition from option "base", and explicit bits
-    answerfile_def = callable_marker(marker.args[0], request)
-    assert isinstance(answerfile_def, AnswerFile)
+    marker_args = ensure_type(tuple[Union[AnswerFile, Callable[..., AnswerFile]]], marker.args)
+    answerfile_def = callable_marker(marker_args[0], request)
 
     yield answerfile_def
 
@@ -322,8 +324,8 @@ def xcpng_chained(request):
     # take test name from mark
     marker = request.node.get_closest_marker("continuation_of")
     assert marker is not None, "xcpng_chained fixture requires 'continuation_of' marker"
-    continuation_of = callable_marker(marker.args[0], request)
-    assert isinstance(continuation_of, Sequence)
+    marker_args = ensure_type(tuple[Union[Sequence[dict], Callable[..., Sequence[dict]]]], marker.args)
+    continuation_of = callable_marker(marker_args[0], request)
 
     vm_defs = [dict(name=vm_spec['vm'],
                     image_test=vm_spec['image_test'],