test/csit/tests/dcaegen2/prh_testcases/resources/Simulator.py

   1 '''
   2 Created on May 10, 2018
   3
   4 @author: mwagner9
   5 '''
   6
   7 import BaseHTTPServer
   8 import re
   9 import sys
  10
  11 from robot.api import logger
  12
  13 import PrhVariables
  14
  15 Httpd = None
  16 prh_ready = ''
  17
  18 class BaseHandler(BaseHTTPServer.BaseHTTPRequestHandler, object):
  19     def do_GET(self, param1, param2):
  20         """Serve a GET request."""
  21         #prepare GET response
  22         logger.console(self.raw_requestlinel)
  23         json_string = param1
  24         if re.search(param2, self.path) is not None:
  25             self.send_response(200)
  26             self.send_header("Content-type", "application/json")
  27             self.end_headers()
  28             self.wfile.write(json_string)
  29
  30 class AAIHandler(BaseHandler):
  31     def do_PATCH(self):
  32         #prepare PATCH response
  33         logger.console('========')
  34         logger.console(self.raw_requestline)
  35         logger.console('========')
  36         if re.search('/aai/v12/network/pnfs/pnf/NOKQTFCOC540002E', self.path)is not None:
  37             st = '{"ipaddress-v6-oam": "2001:0db8:85a3:0000:0000:8a2e:0370:7334","ipaddress-v4-oam": "11.22.33.156"}'
  38             logger.console('========')
  39             logger.console(self.raw_requestline)
  40             logger.console('========')
  41             self.send_response(200)
  42
  43     def do_GET(self):
  44         param1 = '{"pnf-name":"NOKQTFCOC540002E", "pnf-id":"NOKQTFCOC540002E","ipaddress-v4-oam":"10.16.123.234","ipaddress-v6-oam":"2001:0db8:85a3:0000:0000:8a2e:0370:7334"}'
  45         param2 = '/aai/v12/network/pnfs/pnf/NOKQTFCOC540002E'
  46         super(self.__class__, self).do_GET(param1, param2)
  47
  48 class DMaaPHandler(BaseHandler):
  49     def do_GET(self):
  50         param1 = '{"event": {"commonEventHeader": {"sourceId":"QTFCOC540002E", "startEpochMicrosec":1519837825682, "eventId":"QTFCOC540002E-reg", "nfcNamingCode":"5DU", "internalHeaderFields":{"collectorTimeStamp":"Fri, 04 27 2018 09:01:10 GMT"}, "eventType":"pnfRegistration", "priority":"Normal", "version":3, "reportingEntityName":"5GRAN_DU", "sequence":0, "domain":"other", "lastEpochMicrosec":1519837825682, "eventName":"pnfRegistration_5GDU", "sourceName":"5GRAN_DU", "nfNamingCode":"5GRAN"}, "otherFields": {"pnfLastServiceDate":1517206400, "pnfOamIpv6Address":"2001:0db8:85a3:0000:0000:8a2e:0370:7334", "pnfVendorName":"Nokia", "pnfModelNumber":"AJ02", "pnfFamily":"BBU", "pnfType":"AirScale", "otherFieldsVersion":1, "pnfOamIpv4Address":"10.16.123.234", "pnfSoftwareVersion":"v4.5.0.1", "pnfSerialNumber":"QTFCOC540002E", "pnfManufactureDate":1516406400}}}'
  51         param2 = '/events/unauthenticated.SEC_OTHER_OUTPUT/OpenDCAE-c12/c12'
  52         super(self.__class__, self).do_GET(param1, param2)
  53
  54     def do_POST(self):
  55         #Prepare POST response
  56         logger.console('========')
  57         logger.console(self.raw_requestline)
  58         logger.console('========')
  59         if re.search('/events/unauthenticated.PNF_READY', self.path) is not None:
  60             global prh_ready
  61             prh_ready = '{"event":{"correlationID":"NOKQTFCOC540002E", "pnfOamIpv4Address":"10.16.123.234", "pnfOamIpv6Address":"2001:0db8:85a3:0000:0000:8a2e:0370:7334"}}'
  62             self.send_response(200)
  63
  64 def run_server(HandlerClass, port, ServerClass = BaseHTTPServer.HTTPServer, protocol="HTTP/1.0"):
  65     server_address = ('', port)
  66
  67     HandlerClass.protocol_version = protocol
  68     httpd = ServerClass(server_address, HandlerClass)
  69
  70     global Httpd
  71     Httpd = httpd
  72     if HandlerClass is DMaaPHandler:
  73         PrhVariables.DMaaPHTTPD = httpd
  74     else:
  75         PrhVariables.AAIHTTPD = httpd
  76
  77     sa = httpd.socket.getsockname()
  78     print "Serving HTTP on", sa[0], "port", sa[1], "..."
  79
  80 def _main_ (HandlerClass, port, ServerClass = BaseHTTPServer.HTTPServer, protocol="HTTP/1.0"):
  81     server_address = ('', port)
  82
  83     HandlerClass.protocol_version = protocol
  84     httpd = ServerClass(server_address, HandlerClass)
  85
  86     sa = httpd.socket.getsockname()
  87     print "Serving HTTP on", sa[0], "port", sa[1], "..."
  88     httpd.serve_forever()
  89
  90 if __name__ == '__main__':
  91     _main_(DMaaPHandler, 3904)