test/csit/tests/dcaegen2/prh_testcases/resources/simulator/AAI.py

   1 import BaseHTTPServer
   2 import re
   3 import sys
   4
   5 from robot.api import logger
   6
   7 pnfs = 'Empty'
   8
   9 class AAIHandler(BaseHTTPServer.BaseHTTPRequestHandler):
  10     def do_PUT(self):
  11         if re.search('/set_pnfs', self.path):
  12             global pnfs
  13             content_length = int(self.headers['Content-Length'])
  14             pnfs = self.rfile.read(content_length)
  15             self.send_response(200)
  16             self.send_header('Content-Type', 'application/json')
  17             self.end_headers()
  18
  19         return
  20
  21     def do_PATCH(self):
  22         pnfs_name = '/aai/v12/network/pnfs/pnf/' + pnfs
  23         if re.search(pnfs_name, self.path):
  24             self.send_response(200)
  25             self.end_headers()
  26
  27         return
  28
  29     def do_GET(self):
  30         self.send_response(200)
  31         self.send_header('Content-Type', 'application/json')
  32         self.end_headers()
  33         self.wfile.write('GET')
  34         self.wfile.close()
  35
  36         return
  37
  38 def _main_ (HandlerClass = AAIHandler,
  39          ServerClass = BaseHTTPServer.HTTPServer, protocol="HTTP/1.0"):
  40
  41     if sys.argv[1:]:
  42         port = int(sys.argv[1])
  43     else:
  44         port = 3333
  45
  46     server_address = ('', port)
  47
  48     HandlerClass.protocol_version = protocol
  49     httpd = ServerClass(server_address, HandlerClass)
  50
  51     sa = httpd.socket.getsockname()
  52     print "Serving HTTP on", sa[0], "port", sa[1], "..."
  53     httpd.serve_forever()
  54
  55 if __name__ == '__main__':
  56     _main_()