tests/dcaegen2/prh-testcases/resources/PrhLibrary.py

   1 import json
   2
   3 import docker
   4 import time
   5
   6
   7 class PrhLibrary(object):
   8
   9     def __init__(self):
  10         pass
  11
  12     @staticmethod
  13     def check_for_log(search_for):
  14         client = docker.from_env()
  15         container = client.containers.get('prh')
  16         print ("Check for log searches for pattern: ", search_for )
  17         for line in container.logs(stream=True):
  18             print ("Check for log analysis line: ", line )
  19             if search_for in line.strip():
  20                 return True
  21         else:
  22             return False
  23
  24     @staticmethod
  25     def create_invalid_notification(json_file):
  26         json_to_python = json.loads(json_file)
  27         correlation_id = PrhLibrary.extract_correlation_id_value(json_to_python, "correlationId")
  28         ipv4 = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "oamV4IpAddress", "oamV4IpAddress")
  29         ipv6 = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "oamV6IpAddress", "oamV6IpAddress")
  30         serial_number = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "serialNumber", "serialNumber")
  31         vendor_name = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "vendorName", "vendorName")
  32         model_number = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "modelNumber", "modelNumber")
  33         unit_type = PrhLibrary.extract_value_from_pnfRegistrationFields(json_to_python, "unitType", "unitType")
  34
  35         additional_fields = PrhLibrary.extract_additional_fields(json_to_python)
  36
  37         str_json = '{' + correlation_id + ipv4 + ipv6 + serial_number + vendor_name + model_number + unit_type + '"nfNamingCode":""' + "," + '"softwareVersion":"",' + additional_fields
  38         return json.dumps(str_json).replace("\\", "")[1:-1].replace("\":", "\": ").rstrip(',') + '\\n}'
  39
  40     @staticmethod
  41     def create_pnf_ready_notification_as_pnf_ready(json_file):
  42         json_to_python = json.loads(json_file)
  43         correlation_id = PrhLibrary.extract_correlation_id_value(json_to_python, "correlationId")
  44
  45         additional_fields = PrhLibrary.extract_additional_fields_value(json_to_python)
  46
  47         str_json = '{' + correlation_id + additional_fields
  48
  49         return json.dumps(str_json.rstrip(',') + '}').replace("\\", "")[1:-1]
  50
  51     @staticmethod
  52     def extract_additional_fields_value(content):
  53         fields = PrhLibrary.get_additional_fields_as_key_value_pairs(content)
  54         if len(fields) == 0:
  55             return ""
  56         return PrhLibrary.build_additional_fields_json(fields)
  57
  58     @staticmethod
  59     def extract_additional_fields(content):
  60         fields = PrhLibrary.get_additional_fields_as_key_value_pairs(content)
  61         if fields == []:
  62             return '"additionalFields":null'
  63         return PrhLibrary.build_additional_fields_json(fields)
  64
  65     @staticmethod
  66     def get_additional_fields_as_key_value_pairs(content):
  67         return content.get("event").get("pnfRegistrationFields").get(
  68             "additionalFields") if "additionalFields" in content["event"]["pnfRegistrationFields"] else []
  69
  70     @staticmethod
  71     def build_additional_fields_json(fields):
  72         res = '"additionalFields":{'
  73         for f in fields:
  74             res += '"' + f + '":"' + fields.get(f) + '",'
  75         return res.rstrip(',') + '},'
  76
  77     @staticmethod
  78     def extract_value_from_pnfRegistrationFields(content, name, key):
  79         return '"' + name + '":"' + (content.get("event").get("pnfRegistrationFields").get(key) + '",' if key in content["event"]["pnfRegistrationFields"] else '",')
  80
  81     @staticmethod
  82     def extract_correlation_id_value(content, name):
  83         return '"' + name + '":"' + (content.get("event").get("commonEventHeader").get("sourceName") + '",' if "sourceName" in content["event"]["commonEventHeader"] else '",')
  84
  85     @staticmethod
  86     def create_pnf_name(json_file):
  87         json_to_python = json.loads(json_file)
  88         correlation_id = json_to_python.get("event").get("commonEventHeader").get("sourceName") + '",' if "sourceName" in json_to_python["event"]["commonEventHeader"] else '",'
  89         return correlation_id
  90
  91     @staticmethod
  92     def ensure_container_is_running(name):
  93         client = docker.from_env()
  94
  95         if not PrhLibrary.is_in_status(client, name, "running"):
  96             print ("starting container", name)
  97             container = client.containers.get(name)
  98             container.start()
  99             PrhLibrary.wait_for_status(client, name, "running")
 100
 101         PrhLibrary.print_status(client)
 102
 103     @staticmethod
 104     def ensure_container_is_exited(name):
 105         client = docker.from_env()
 106
 107         if not PrhLibrary.is_in_status(client, name, "exited"):
 108             print ("stopping container", name)
 109             container = client.containers.get(name)
 110             container.stop()
 111             PrhLibrary.wait_for_status(client, name, "exited")
 112
 113         PrhLibrary.print_status(client)
 114
 115     @staticmethod
 116     def print_status(client):
 117         print("containers status")
 118         for c in client.containers.list(all=True):
 119             print(c.name, "   ", c.status)
 120
 121     @staticmethod
 122     def wait_for_status(client, name, status):
 123         while not PrhLibrary.is_in_status(client, name, status):
 124             print ("waiting for container: ", name, "to be in status: ", status)
 125             time.sleep(3)
 126
 127     @staticmethod
 128     def is_in_status(client, name, status):
 129         return len(client.containers.list(all=True, filters={"name": "^/"+name+"$", "status": status})) == 1