From 469a3efd4b959afc971217003676e3b9910e18a1 Mon Sep 17 00:00:00 2001 From: Joanna Jeremicz Date: Tue, 19 Mar 2019 09:37:59 +0100 Subject: Add new testcase to PRH CSITs Change-Id: I14fec39acc2794e6cabd89fa95a457da88a36667 Issue-ID: INT-972 Signed-off-by: Joanna Jeremicz --- .../dcaegen2/prh-testcases/resources/PrhLibrary.py | 38 +++++++++++++--------- 1 file changed, 23 insertions(+), 15 deletions(-) (limited to 'tests/dcaegen2/prh-testcases/resources') diff --git a/tests/dcaegen2/prh-testcases/resources/PrhLibrary.py b/tests/dcaegen2/prh-testcases/resources/PrhLibrary.py index ee89be68..6667bd78 100644 --- a/tests/dcaegen2/prh-testcases/resources/PrhLibrary.py +++ b/tests/dcaegen2/prh-testcases/resources/PrhLibrary.py @@ -31,7 +31,8 @@ class PrhLibrary(object): vendor_name = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "vendorName", "vendorName") model_number = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "modelNumber", "modelNumber") unit_type = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "unitType", "unitType") - additional_fields = PrhLibrary.extract_additional_fields(json_to_python, "additionalFields") + + additional_fields = PrhLibrary.extract_additional_fields(json_to_python) str_json = '{' + correlation_id + ipv4 + ipv6 + serial_number + vendor_name + model_number + unit_type + '"nfNamingCode":""' + "," + '"softwareVersion":"",' + additional_fields return json.dumps(str_json).replace("\\", "")[1:-1].replace("\":", "\": ").rstrip(',') + '\\n}' @@ -44,7 +45,8 @@ class PrhLibrary(object): vendor_name = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "equip-vendor", "vendorName") model_number = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "equip-model", "modelNumber") unit_type = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "equip-type", "unitType") - additional_fields = PrhLibrary.extract_additional_fields_value(json_to_python, "additionalFields") + + additional_fields = PrhLibrary.extract_additional_fields_value(json_to_python) nf_role = json_to_python.get("event").get("commonEventHeader").get("nfNamingCode") if "nfNamingCode" in json_to_python["event"]["commonEventHeader"] else "" @@ -53,23 +55,29 @@ class PrhLibrary(object): return json.dumps(str_json.rstrip(',') + '}').replace("\\", "")[1:-1] @staticmethod - def extract_additional_fields(content, name): - fields = content.get("event").get("pnfRegistrationFields").get(name) if name in content["event"]["pnfRegistrationFields"] else [] + def extract_additional_fields_value(content): + fields = PrhLibrary.get_additional_fields_as_key_value_pairs(content) + if len(fields) == 0: + return "" + return PrhLibrary.build_additional_fields_json(fields) + + @staticmethod + def extract_additional_fields(content): + fields = PrhLibrary.get_additional_fields_as_key_value_pairs(content) if fields == []: - return '"additionalFields":' + 'null' - res = '"' + name + '":{' - for f in fields: - res += '"' + f + '"' + ':' + '"' + fields.get(f) + '",' - return res.rstrip(',') + '},' + return '"additionalFields":null' + return PrhLibrary.build_additional_fields_json(fields) @staticmethod - def extract_additional_fields_value(content, name): - fields = content.get("event").get("pnfRegistrationFields").get(name) if name in content["event"]["pnfRegistrationFields"] else [] - if fields == [] or len(fields) == 0: - return "" - res = '"' + name + '":{' + def get_additional_fields_as_key_value_pairs(content): + return content.get("event").get("pnfRegistrationFields").get( + "additionalFields") if "additionalFields" in content["event"]["pnfRegistrationFields"] else [] + + @staticmethod + def build_additional_fields_json(fields): + res = '"additionalFields":{' for f in fields: - res += '"' + f + '"' + ':' + '"' + fields.get(f) + '",' + res += '"' + f + '":"' + fields.get(f) + '",' return res.rstrip(',') + '},' @staticmethod -- cgit 1.2.3-korg