1 files changed, 95 insertions, 0 deletions
diff --git a/test/csit/tests/dcaegen2/prh_testcases/resources/PrhLibrary.py b/test/csit/tests/dcaegen2/prh_testcases/resources/PrhLibrary.py
new file mode 100644
index 000000000..893ed9460
--- /dev/null
+++ b/test/csit/tests/dcaegen2/prh_testcases/resources/PrhLibrary.py
@@ -0,0 +1,95 @@
+'''
+Created on Apr 27, 2018
+
+@author: mwagner9
+'''
+
+import threading
+import time
+
+from robot.api import logger
+
+import PrhVariables
+import Simulator
+
+st = '{"event":{"correlationID":"NOKQTFCOC540002E", "pnfOamIpv4Address":"10.16.123.234", "pnfOamIpv6Address":"2001:0db8:85a3:0000:0000:8a2e:0370:7334"}}'
+
+class PrhLibrary(object):
+    def __init__(self):
+        pass
+
+    def setup_dmaap_server(self):
+        return _setup(PrhVariables.DMaaPHttpServerThread, 'DMaaP', PrhVariables.DMaaPIsRobotRun, Simulator, 3904)
+
+    def setup_aai_server(self):
+        return _setup(PrhVariables.AAIHttpServerThread, 'AAI', PrhVariables.AAIIsRobotRun, Simulator, 3905)
+
+    def shutdown_dmaap_server(self):
+        return _shutdown(PrhVariables.DMaaPHTTPD, 'DMaaP')
+
+    def shutdown_aai_server(self):
+        return _shutdown(PrhVariables.AAIHTTPD, 'AAI')
+
+    def is_json_empty(self, response):
+        logger.info("Enter is_json_empty: response.text: " + response.text)
+        if response.text is None or len(response.text) < 2:
+            return 'True'
+        return 'False'
+
+    def dmaap_collectorTimeStamp_receive(self, search, response):
+        return _find_element(search, response)
+
+    def AAI_Ipv4_receive(self, search, response):
+        return _find_element(search, response)
+
+    def AAI_Ipv6_receive(self, search, response):
+        return _find_element(search, response)
+
+    def check_pnf_ready(self):
+        if st in Simulator.prh_ready:
+            return 'true'
+        return 'false'
+
+def _setup(serverthread, servername, isrobotrun, module, portNum):
+    if serverthread is not None:
+        logger.console('{} Mockup Sever started'.format(servername))
+        return "true"
+
+    isrobotrun = True
+
+    module_handler = module.AAIHandler if servername is 'AAI' else module.DMaaPHandler 
+
+    module.run_server(module_handler, portNum)
+    try:
+        serverthread = threading.Thread(name='{}_HTTPServer'.format(servername), target=module.Httpd.serve_forever)
+        serverthread.start()
+        logger.console('{}  Mockup Sever started'.format(servername))
+        time.sleep(2)
+        return "true"
+    except Exception as e:
+        print (str(e))
+        return "false"
+
+def _shutdown(server, name):
+    if server is not None:
+        server.shutdown()
+        logger.console("{} Server shut down".format(name))
+        time.sleep(3)
+        return "true"
+    else:
+        return "false"
+
+def _find_element(search, response):
+    while response is not None:
+        json_data = str(response)
+        if search in json_data:
+            return 'true'
+    return 'false'
+
+if __name__ == '__main__':
+    lib = PrhLibrary()
+    ret = lib.setup_dmaap_server()
+    print ret
+    ret = lib.setup_aai_server()
+    print ret
+    time.sleep(10)