Change pytest plugin to use the same importer as other components

Fixes #166

Author: ogenstad

changelog/166.fixed.md

@@ -0,0 +1 @@
+Fix relative imports for the pytest plugin, note that the relative imports can't be at the top level of the repository alongside .infrahub.yml. They have to be located within a subfolder.

infrahub_sdk/pytest_plugin/items/check.py

@@ -1,12 +1,12 @@
 from __future__ import annotations
 
 import asyncio
+from pathlib import Path
 from typing import TYPE_CHECKING, Any, Optional
 
 import ujson
 from httpx import HTTPStatusError
 
-from ...checks import get_check_class_instance
 from ..exceptions import CheckDefinitionError, CheckResultError
 from ..models import InfrahubTestExpectedResult
 from .base import InfrahubItem
@@ -33,10 +33,10 @@ def __init__(
         self.check_instance: InfrahubCheck
 
     def instantiate_check(self) -> None:
-        self.check_instance = get_check_class_instance(
-            check_config=self.resource_config,  # type: ignore[arg-type]
-            search_path=self.session.infrahub_config_path.parent,  # type: ignore[attr-defined]
+        relative_path = (
+            str(self.resource_config.file_path.parent) if self.resource_config.file_path.parent != Path() else None  # type: ignore[attr-defined]
         )
+        self.check_instance = self.resource_config.load_class(import_root=str(Path.cwd()), relative_path=relative_path)  # type: ignore[attr-defined]
 
     def run_check(self, variables: dict[str, Any]) -> Any:
         self.instantiate_check()

infrahub_sdk/pytest_plugin/items/python_transform.py

@@ -1,12 +1,12 @@
 from __future__ import annotations
 
 import asyncio
+from pathlib import Path
 from typing import TYPE_CHECKING, Any, Optional
 
 import ujson
 from httpx import HTTPStatusError
 
-from ...transforms import get_transform_class_instance
 from ..exceptions import OutputMatchError, PythonTransformDefinitionError
 from ..models import InfrahubTestExpectedResult
 from .base import InfrahubItem
@@ -33,9 +33,11 @@ def __init__(
         self.transform_instance: InfrahubTransform
 
     def instantiate_transform(self) -> None:
-        self.transform_instance = get_transform_class_instance(
-            transform_config=self.resource_config,  # type: ignore[arg-type]
-            search_path=self.session.infrahub_config_path.parent,  # type: ignore[attr-defined]
+        relative_path = (
+            str(self.resource_config.file_path.parent) if self.resource_config.file_path.parent != Path() else None  # type: ignore[attr-defined]
+        )
+        self.transform_instance = self.resource_config.load_class(  # type: ignore[attr-defined]
+            import_root=str(Path.cwd()), relative_path=relative_path
         )
 
     def run_transform(self, variables: dict[str, Any]) -> Any: