build relationships only if include contains rel name

Author: LucasG0

infrahub_sdk/client.py

@@ -568,7 +568,7 @@ async def _process_nodes_and_relationships(
             node = await InfrahubNode.from_graphql(client=self, branch=branch, data=item, timeout=timeout)
             nodes.append(node)
 
-            if prefetch_relationships or include is not None:
+            if prefetch_relationships or (include and any(rel in include for rel in node._relationships)):
                 await node._process_relationships(
                     node_data=item,
                     branch=branch,
@@ -1864,7 +1864,7 @@ def _process_nodes_and_relationships(
             node = InfrahubNodeSync.from_graphql(client=self, branch=branch, data=item, timeout=timeout)
             nodes.append(node)
 
-            if prefetch_relationships or include is not None:
+            if prefetch_relationships or (include and any(rel in include for rel in node._relationships)):
                 node._process_relationships(node_data=item, branch=branch, related_nodes=related_nodes, timeout=timeout)
 
         return ProcessRelationsNodeSync(nodes=nodes, related_nodes=related_nodes)

tests/unit/sdk/conftest.py

@@ -2646,20 +2646,3 @@ async def mock_query_tasks_05(httpx_mock: HTTPXMock) -> HTTPXMock:
         is_reusable=True,
     )
     return httpx_mock
-
-
-async def set_builtin_tag_schema_cache(client) -> None:
-    # Set tag schema in cache to avoid needed to request the server.
-    builtin_tag_schema = {
-        "version": "1.0",
-        "nodes": [
-            {
-                "name": "Tag",
-                "namespace": "Builtin",
-                "default_filter": "name__value",
-                "display_label": "name__value",
-                "branch": "aware",
-            }
-        ],
-    }
-    client.schema.set_cache(builtin_tag_schema)

tests/unit/sdk/test_node.py

@@ -17,7 +17,6 @@
 from infrahub_sdk.node.constants import SAFE_VALUE
 from infrahub_sdk.node.related_node import RelatedNode, RelatedNodeSync
 from infrahub_sdk.schema import GenericSchema, NodeSchemaAPI
-from tests.unit.sdk.conftest import set_builtin_tag_schema_cache
 
 if TYPE_CHECKING:
     from infrahub_sdk.client import InfrahubClient, InfrahubClientSync
@@ -54,6 +53,23 @@
 ]
 
 
+async def set_builtin_tag_schema_cache(client) -> None:
+    # Set tag schema in cache to avoid needed to request the server.
+    builtin_tag_schema = {
+        "version": "1.0",
+        "nodes": [
+            {
+                "name": "Tag",
+                "namespace": "Builtin",
+                "default_filter": "name__value",
+                "display_label": "name__value",
+                "branch": "aware",
+            }
+        ],
+    }
+    client.schema.set_cache(builtin_tag_schema)
+
+
 async def test_method_sanity() -> None:
     """Validate that there is at least one public method and that both clients look the same."""
     assert async_node_methods