fix tests

Author: wvandeun

infrahub_sdk/node/attribute.py

@@ -106,7 +106,7 @@ def _generate_input_data(self) -> dict | None:
         return {"data": data, "variables": variables}
 
     def _generate_query_data(self, property: bool = False) -> dict | None:
-        data: dict[str, Any] = {"value": None, "updated_at": None}
+        data: dict[str, Any] = {"value": None}
 
         if property:
             data.update({"is_default": None, "is_from_profile": None})

infrahub_sdk/node/constants.py

@@ -1,7 +1,7 @@
 import ipaddress
 import re
 
-PROPERTIES_FLAG = ["is_visible", "is_protected"]
+PROPERTIES_FLAG = ["is_visible", "is_protected", "updated_at"]
 PROPERTIES_OBJECT = ["source", "owner"]
 SAFE_VALUE = re.compile(r"(^[\. /:a-zA-Z0-9_-]+$)|(^$)")
 

infrahub_sdk/node/related_node.py

@@ -163,15 +163,14 @@ def _generate_query_data(cls, peer_data: dict[str, Any] | None = None, property:
                 and typename. The method also includes additional properties and any peer_data provided.
         """
         data: dict[str, Any] = {"node": {"id": None, "hfid": None, "display_label": None, "__typename": None}}
-        properties: dict[str, Any] = {"updated_at": None}
+        properties: dict[str, Any] = {}
 
         if property:
             for prop_name in PROPERTIES_FLAG:
                 properties[prop_name] = None
             for prop_name in PROPERTIES_OBJECT:
                 properties[prop_name] = {"id": None, "display_label": None, "__typename": None}
 
-        if properties:
             data["properties"] = properties
         if peer_data:
             data["node"].update(peer_data)

infrahub_sdk/node/relationship.py

@@ -87,9 +87,8 @@ def _generate_query_data(cls, peer_data: dict[str, Any] | None = None, property:
                 properties[prop_name] = None
             for prop_name in PROPERTIES_OBJECT:
                 properties[prop_name] = {"id": None, "display_label": None, "__typename": None}
-
-        if properties:
             data["edges"]["properties"] = properties
+
         if peer_data:
             data["edges"]["node"].update(peer_data)
 

tests/unit/sdk/conftest.py

@@ -468,20 +468,23 @@ async def location_data01_property() -> dict[str, Any]:
                 "is_visible": True,
                 "owner": None,
                 "source": None,
+                "updated_at": "2024-01-15T10:30:00.000000Z",
                 "value": "DFW",
             },
             "description": {
                 "is_protected": False,
                 "is_visible": True,
                 "owner": None,
                 "source": None,
+                "updated_at": "2024-01-15T10:30:00.000000Z",
                 "value": None,
             },
             "type": {
                 "is_protected": True,
                 "is_visible": True,
                 "owner": None,
                 "source": None,
+                "updated_at": "2024-01-15T10:30:00.000000Z",
                 "value": "SITE",
             },
             "primary_tag": {
@@ -490,6 +493,7 @@ async def location_data01_property() -> dict[str, Any]:
                     "is_visible": True,
                     "owner": None,
                     "source": None,
+                    "updated_at": "2024-01-15T10:30:00.000000Z",
                 },
                 "node": {
                     "id": "rrrrrrrr-rrrr-rrrr-rrrr-rrrrrrrrrrrr",
@@ -506,6 +510,7 @@ async def location_data01_property() -> dict[str, Any]:
                             "is_visible": True,
                             "owner": None,
                             "source": None,
+                            "updated_at": "2024-01-15T10:30:00.000000Z",
                         },
                         "node": {
                             "id": "bbbbbbbb-bbbb-bbbb-bbbb-bbbbbbbbbbbb",
@@ -574,13 +579,15 @@ async def location_data02_property() -> dict[str, Any]:
                     "display_label": "CRM",
                     "id": "cccccccc-cccc-cccc-cccc-cccccccccccc",
                 },
+                "updated_at": "2024-01-15T10:30:00.000000Z",
                 "value": "dfw1",
             },
             "description": {
                 "is_protected": False,
                 "is_visible": True,
                 "owner": None,
                 "source": None,
+                "updated_at": "2024-01-15T10:30:00.000000Z",
                 "value": None,
             },
             "type": {
@@ -592,6 +599,7 @@ async def location_data02_property() -> dict[str, Any]:
                     "display_label": "CRM",
                     "id": "cccccccc-cccc-cccc-cccc-cccccccccccc",
                 },
+                "updated_at": "2024-01-15T10:30:00.000000Z",
                 "value": "SITE",
             },
             "primary_tag": {
@@ -604,6 +612,7 @@ async def location_data02_property() -> dict[str, Any]:
                         "display_label": "CRM",
                         "id": "cccccccc-cccc-cccc-cccc-cccccccccccc",
                     },
+                    "updated_at": "2024-01-15T10:30:00.000000Z",
                 },
                 "node": {
                     "id": "rrrrrrrr-rrrr-rrrr-rrrr-rrrrrrrrrrrr",
@@ -624,6 +633,7 @@ async def location_data02_property() -> dict[str, Any]:
                                 "display_label": "CRM",
                                 "id": "cccccccc-cccc-cccc-cccc-cccccccccccc",
                             },
+                            "updated_at": "2024-01-15T10:30:00.000000Z",
                         },
                         "node": {
                             "id": "bbbbbbbb-bbbb-bbbb-bbbb-bbbbbbbbbbbb",