update InfrahubNodeSync too, add unit test

Author: ajtmccarty

infrahub_sdk/node/node.py

@@ -1090,11 +1090,18 @@ def _init_relationships(self, data: dict | None = None) -> None:
 
             if rel_schema.cardinality == "one":
                 if isinstance(rel_data, RelatedNodeSync):
-                    self._relationship_cardinality_one_data[rel_schema.name] = rel_data
-                else:
-                    self._relationship_cardinality_one_data[rel_schema.name] = RelatedNodeSync(
-                        name=rel_schema.name, branch=self._branch, client=self._client, schema=rel_schema, data=rel_data
-                    )
+                    peer_id_data: dict[str, Any] = {}
+                    if rel_data.id:
+                        peer_id_data["id"] = rel_data.id
+                    if rel_data.hfid:
+                        peer_id_data["hfid"] = rel_data.hfid
+                    if peer_id_data:
+                        rel_data = peer_id_data
+                    else:
+                        rel_data = None
+                self._relationship_cardinality_one_data[rel_schema.name] = RelatedNodeSync(
+                    name=rel_schema.name, branch=self._branch, client=self._client, schema=rel_schema, data=rel_data
+                )
             else:
                 self._relationship_cardinality_many_data[rel_schema.name] = RelationshipManagerSync(
                     name=rel_schema.name,

tests/integration/test_node.py

@@ -74,9 +74,9 @@ async def test_node_create_with_relationships_using_related_node(
     ):
         related_node = car_golf.owner
         node = await client.create(
-            kind=TESTING_CAR, name="CoolerTiguan", color="Black", manufacturer=manufacturer_mercedes, owner=related_node
+            kind=TESTING_CAR, name="Tiguan", color="Black", manufacturer=manufacturer_mercedes, owner=related_node
         )
-        await node.save()
+        await node.save(allow_upsert=True)
         assert node.id is not None
 
         node_after = await client.get(kind=TESTING_CAR, id=node.id, prefetch_relationships=True)

tests/unit/sdk/test_node.py

@@ -15,6 +15,7 @@
     parse_human_friendly_id,
 )
 from infrahub_sdk.node.constants import SAFE_VALUE
+from infrahub_sdk.node.related_node import RelatedNode, RelatedNodeSync
 from infrahub_sdk.schema import GenericSchema, NodeSchemaAPI
 
 if TYPE_CHECKING:
@@ -194,6 +195,50 @@ async def test_init_node_data_user_with_relationships(client, location_schema: N
     assert "get_kind" in keys
 
 
+@pytest.mark.parametrize("client_type", client_types)
+async def test_init_node_data_user_with_relationships_using_related_node(
+    client, location_schema: NodeSchemaAPI, client_type
+):
+    rel_schema = location_schema.get_relationship(name="primary_tag")
+    if client_type == "standard":
+        primary_tag = RelatedNode(
+            name="primary_tag", branch="main", client=client, schema=rel_schema, data={"id": "pppppppp"}
+        )
+    else:
+        primary_tag = RelatedNodeSync(
+            name="primary_tag", branch="main", client=client, schema=rel_schema, data={"id": "pppppppp"}
+        )
+
+    data = {
+        "name": {"value": "JFK1"},
+        "description": {"value": "JFK Airport"},
+        "type": {"value": "SITE"},
+        "primary_tag": primary_tag,
+        "tags": [{"id": "aaaaaa"}, {"id": "bbbb"}],
+    }
+    if client_type == "standard":
+        node = InfrahubNode(client=client, schema=location_schema, data=data)
+    else:
+        node = InfrahubNodeSync(client=client, schema=location_schema, data=data)
+
+    assert node.name.value == "JFK1"
+    assert node.name.is_protected is None
+    assert node.description.value == "JFK Airport"
+    assert node.type.value == "SITE"
+
+    assert isinstance(node.tags, RelationshipManagerBase)
+    assert len(node.tags.peers) == 2
+    assert isinstance(node.tags.peers[0], RelatedNodeBase)
+    assert isinstance(node.primary_tag, RelatedNodeBase)
+    assert node.primary_tag.id == "pppppppp"
+
+    keys = dir(node)
+    assert "name" in keys
+    assert "type" in keys
+    assert "tags" in keys
+    assert "get_kind" in keys
+
+
 @pytest.mark.parametrize("property_test", property_tests)
 @pytest.mark.parametrize("client_type", client_types)
 async def test_init_node_data_graphql(