From 1018811d152f8861db476a61aa5ca3d3b75365d9 Mon Sep 17 00:00:00 2001 From: Patrick Ogenstad Date: Thu, 5 Jun 2025 16:46:54 +0200 Subject: [PATCH] Add transform infrahub_node argument --- infrahub_sdk/pytest_plugin/items/python_transform.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/infrahub_sdk/pytest_plugin/items/python_transform.py b/infrahub_sdk/pytest_plugin/items/python_transform.py index a030f8a1..02977f48 100644 --- a/infrahub_sdk/pytest_plugin/items/python_transform.py +++ b/infrahub_sdk/pytest_plugin/items/python_transform.py @@ -7,6 +7,7 @@ import ujson from httpx import HTTPStatusError +from ...node import InfrahubNode from ..exceptions import OutputMatchError, PythonTransformDefinitionError from ..models import InfrahubTestExpectedResult from .base import InfrahubItem @@ -41,7 +42,7 @@ def instantiate_transform(self) -> None: ) client = self.session.infrahub_client # type: ignore[attr-defined] # TODO: Look into seeing how a transform class may use the branch, but set as a empty string for the time being to keep current behaviour - self.transform_instance = transform_class(branch="", client=client) + self.transform_instance = transform_class(branch="", client=client, infrahub_node=InfrahubNode) def run_transform(self, variables: dict[str, Any]) -> Any: self.instantiate_transform()