tests/dcaegen2/prh-testcases/resources/simulator/DMaaP.py

   1 from http.server import BaseHTTPRequestHandler
   2 from http.server import HTTPServer
   3 import re
   4 import sys
   5
   6 posted_event_from_prh = b'Empty'
   7 received_event_to_get_method = b'Empty'
   8
   9
  10 class DMaaPHandler(BaseHTTPRequestHandler):
  11
  12     def do_PUT(self):
  13         if re.search('/set_get_event', self.path):
  14             global received_event_to_get_method
  15             content_length = int(self.headers['Content-Length'])
  16             received_event_to_get_method = self.rfile.read(content_length)
  17             _header_200_and_json(self)
  18
  19         return
  20
  21     def do_POST(self):
  22         if re.search('/events/unauthenticated.PNF_READY', self.path):
  23             global posted_event_from_prh
  24             content_length = int(self.headers['Content-Length'])
  25             posted_event_from_prh = self.rfile.read(content_length)
  26             _header_200_and_json(self)
  27
  28         return
  29
  30     def do_GET(self):
  31         if re.search('/events/unauthenticated.VES_PNFREG_OUTPUT/OpenDcae-c12/c12', self.path):
  32             _header_200_and_json(self)
  33             self.wfile.write(received_event_to_get_method)
  34         elif re.search('/events/pnfReady', self.path):
  35             _header_200_and_json(self)
  36             self.wfile.write(posted_event_from_prh)
  37
  38         return
  39
  40
  41 def _header_200_and_json(self):
  42     self.send_response(200)
  43     self.send_header('Content-Type', 'application/json')
  44     self.end_headers()
  45
  46
  47 def _main_(handler_class=DMaaPHandler, server_class=HTTPServer, protocol="HTTP/1.0"):
  48
  49     if sys.argv[1:]:
  50         port = int(sys.argv[1])
  51     else:
  52         port = 2222
  53
  54     server_address = ('', port)
  55
  56     handler_class.protocol_version = protocol
  57     httpd = server_class(server_address, handler_class)
  58
  59     sa = httpd.socket.getsockname()
  60     print("Serving HTTP on", sa[0], "port", sa[1], "...")
  61     httpd.serve_forever()
  62
  63
  64 if __name__ == '__main__':
  65     _main_()