added assertion for contacts from Related Services

test/test_worker.py

@@ -211,12 +211,16 @@ def test_transform_ims_data():
 
     customer_contacts = {
         "cu_1": ["customer_1@a.org"],
-        "cu_1_1": ["customer_1_1@a.org"]
+        "cu_1_1": ["customer_1_1@a.org"],
+        "cu_sub_1": ["customer_sub_1@a.org"],
+        "cu_sub_2": ["customer_sub_2@a.org"]
     }
 
     planned_work_contacts = {
         "cu_1": ["customer_1_PW@a.org"],
-        "cu_1_1": ["customer_1_1_PW@a.org"]
+        "cu_1_1": ["customer_1_1_PW@a.org"],
+        "cu_sub_1": ["customer_sub_1_PW@a.org"],
+        "cu_sub_2": ["customer_sub_2_PW@a.org"]
     }
 
     port_id_details = {
@@ -315,7 +319,7 @@ def test_transform_ims_data():
             "project": "customer_1",
             "carrier-circuits": ["carrier_id_1"],
             "sub-circuits": ["sub_circuit_1"],
-            "customerid": "cu_1",
+            "customerid": "cu_1"
         },
         "carrier_id_1": {
             "id": "carrier_id_1",
@@ -363,7 +367,7 @@ def test_transform_ims_data():
             "project": "customer_1",
             "carrier-circuits": ["circ_id_1"],
             "sub-circuits": ["sub_circuit_2"],
-            "customerid": "cu_1",
+            "customerid": "cu_Sub_1",
         },
         "sub_circuit_2": {
             "id": "sub_circuit_2",
@@ -375,7 +379,7 @@ def test_transform_ims_data():
             "project": "customer_1",
             "carrier-circuits": ["sub_circuit_1"],
             "sub-circuits": [],
-            "customerid": "cu_1",
+            "customerid": "cu_sub_2",
         },
         702203: {
             "id": 702203,
@@ -458,12 +462,16 @@ def test_transform_ims_data():
         "CAM01-MTC6-3:1-A-1-S1-1",
         "CAM01-MTC6-3:1-A-1-L1-1"
     ]
+    assert hierarchy["circ_id_1"]["contacts"] == ["customer_1@a.org", "customer_1_1@a.org"]
+
     for v in ifs.values():
         assert len(v) == 1
         assert len(v[0]["related-services"]) == 1
         assert v[0]["related-services"][0]["id"] == "sub_circuit_2"
         assert len(v[0]["fibre-routes"]) == 1
         assert v[0]["fibre-routes"][0]["id"] == "carrier_id_3"
+        assert v[0]["contacts"] == ["customer_sub_2@a.org"]
+        assert v[0]["planned_work_contacts"] == ["customer_sub_2_PW@a.org"]
 
     nps = res["node_pair_services"]
     assert list(nps.keys()) == ["eq_a/eq_b", "eq_b/eq_a"]