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

   1 import _thread
   2 import ssl
   3 from http.server import HTTPServer
   4
   5
   6 def header_200_and_json(self):
   7     self.send_response(200)
   8     self.send_header('Content-Type', 'application/json')
   9     self.end_headers()
  10
  11
  12 def start_http_endpoint(port, handler_class):
  13     _thread.start_new_thread(init_http_endpoints, (port, handler_class))
  14
  15
  16 def start_https_endpoint(port, handler_class, keyfile="certs/server.key",
  17                          certfile="certs/server.crt", ca_certs="certs/client.crt"):
  18     _thread.start_new_thread(init_https_endpoints, (port, handler_class, keyfile, certfile, ca_certs))
  19
  20
  21 def init_http_endpoints(port, handler_class, server_class=HTTPServer):
  22     server = server_class(('', port), handler_class)
  23     sa = server.socket.getsockname()
  24     print("Serving HTTP on", sa[0], "port", sa[1], "for", handler_class, "...")
  25     server.serve_forever()
  26
  27
  28 def init_https_endpoints(port, handler_class, keyfile, certfile, ca_certs, server_class=HTTPServer):
  29     server = server_class(('', port), handler_class)
  30     server.socket = ssl.wrap_socket(server.socket, keyfile=keyfile, certfile=certfile,
  31                                     ca_certs=ca_certs, server_side=True)
  32     sa = server.socket.getsockname()
  33     print("Serving HTTPS on", sa[0], "port", sa[1], "for", handler_class, "...")
  34     server.serve_forever()