test: ✅ Update tests to add missing pieces like id key

nautobot · Jan 7, 2025 · cd4981e · cd4981e
1 parent 78d5d98
commit cd4981e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/nautobot_ssot/tests/dna_center/test_adapters_dna_center.py b/nautobot_ssot/tests/dna_center/test_adapters_dna_center.py
@@ -216,6 +216,8 @@ def test_load_device_location_tree_w_floor(self):
         self.dna_center.building_map = {
             "5": {
                 "name": "HQ",
+                "id": "5",
+                "parentId": "4",
                 "additionalInfo": [
                     {
                         "nameSpace": "Location",
@@ -229,6 +231,7 @@ def test_load_device_location_tree_w_floor(self):
                         },
                     }
                 ],
+                "siteHierarchy": "/1/2/3/4/5/",
             },
         }
         mock_loc_data = {"areas": ["Global", "USA", "New York", "NYC"], "building": "HQ", "floor": "1st Floor"}
@@ -277,6 +280,8 @@ def test_load_device_location_tree_wo_floor(self):
         self.dna_center.building_map = {
             "5": {
                 "name": "HQ",
+                "id": "5",
+                "parentId": "4",
                 "additionalInfo": [
                     {
                         "nameSpace": "Location",
@@ -290,6 +295,7 @@ def test_load_device_location_tree_wo_floor(self):
                         },
                     }
                 ],
+                "siteHierarchy": "/1/2/3/4/5/",
             },
         }
         mock_loc_data = {"areas": ["Global", "USA", "New York", "NYC"], "building": "HQ"}