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

   1 import _thread
   2 import re
   3 import ssl
   4 import time
   5 from http.server import BaseHTTPRequestHandler
   6 from http.server import HTTPServer
   7
   8 pnfs = 'Empty'
   9
  10
  11 class AAISetup(BaseHTTPRequestHandler):
  12
  13     def do_PUT(self):
  14         if re.search('/set_pnfs', self.path):
  15             global pnfs
  16             content_length = int(self.headers['Content-Length'])
  17             pnfs = self.rfile.read(content_length)
  18             _header_200_and_json(self)
  19
  20         return
  21
  22     def do_POST(self):
  23         if re.search('/reset', self.path):
  24             global pnfs
  25             pnfs = 'Empty'
  26             _header_200_and_json(self)
  27
  28         return
  29
  30 class AAIHandler(BaseHTTPRequestHandler):
  31
  32     def do_PATCH(self):
  33         pnfs_name = '/aai/v12/network/pnfs/pnf/' + pnfs.decode()
  34         if re.search('wrong_aai_record', self.path):
  35             self.send_response(400)
  36             self.end_headers()
  37         elif re.search(pnfs_name, self.path):
  38             self.send_response(200)
  39             self.end_headers()
  40
  41         return
  42
  43
  44 def _header_200_and_json(self):
  45     self.send_response(200)
  46     self.send_header('Content-Type', 'application/json')
  47     self.end_headers()
  48
  49
  50 def _main_(handler_class=AAIHandler, protocol="HTTP/1.0"):
  51     handler_class.protocol_version = protocol
  52     _thread.start_new_thread(_init_http_endpoints, (3333, AAIHandler))
  53     _thread.start_new_thread(_init_https_endpoints, (3334, AAIHandler))
  54     _thread.start_new_thread(_init_http_endpoints, (3335, AAISetup))
  55     while 1:
  56         time.sleep(10)
  57
  58
  59 def _init_http_endpoints(port, handler_class, server_class=HTTPServer):
  60     server = server_class(('', port), handler_class)
  61     sa = server.socket.getsockname()
  62     print("Serving HTTP on", sa[0], "port", sa[1], "for", handler_class, "...")
  63     server.serve_forever()
  64
  65
  66 def _init_https_endpoints(port, handler_class, server_class=HTTPServer):
  67     server = server_class(('', port), handler_class)
  68     server.socket = ssl.wrap_socket(server.socket,
  69                                     keyfile="certs/server.key", certfile="certs/server.crt",
  70                                     ca_certs="certs/client.crt", server_side=True)
  71     sa = server.socket.getsockname()
  72     print("Serving HTTPS on", sa[0], "port", sa[1], "for", handler_class, "...")
  73     server.serve_forever()
  74
  75
  76 if __name__ == '__main__':
  77     _main_()