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

   1 import logging
   2 import json
   3 import sys
   4 import re
   5 import time
   6 from http.server import BaseHTTPRequestHandler
   7 from os.path import basename
   8 import httpServerLib
   9
  10 ch = logging.StreamHandler(sys.stdout)
  11 handlers = [ch]
  12 logging.basicConfig(
  13     level=logging.DEBUG,
  14     format='[%(asctime)s] {%(filename)s:%(lineno)d} %(levelname)s - %(message)s',
  15     handlers=handlers
  16 )
  17
  18 logger = logging.getLogger('AAI-simulator-logger')
  19
  20 AAI_RESOURCE_NOT_FOUND = b'{}'
  21
  22 pnf_entries = {}
  23 patched_pnf = AAI_RESOURCE_NOT_FOUND
  24 created_logical_link = AAI_RESOURCE_NOT_FOUND
  25 service_instance = AAI_RESOURCE_NOT_FOUND
  26
  27 class AAISetup(BaseHTTPRequestHandler):
  28
  29     def do_GET(self):
  30         try:
  31             if re.search('/setup/patched_pnf', self.path):
  32                 httpServerLib.set_response_200_ok(self, payload = patched_pnf)
  33                 logger.debug('AAISetup GET /setup/patched_pnf -> 200 OK')
  34             elif re.search('/verify/logical-link', self.path):
  35                 httpServerLib.set_response_200_ok(self, payload = created_logical_link)
  36                 logger.debug('AAISetup GET /verify/logical_link -> 200 OK')
  37             else:
  38                 httpServerLib.set_response_404_not_found(self)
  39                 logger.info('AAISetup GET ' + self.path + ' -> 404 Not found')
  40         except Exception as e:
  41             logger.error(e)
  42             httpServerLib.set_response_500_server_error(self)
  43
  44     def do_PUT(self):
  45         try:
  46             if re.search('/setup/add_pnf_entry', self.path):
  47                 pnf_entry = httpServerLib.get_payload(self)
  48                 pnf_name = json.loads(pnf_entry).get("pnf-name")
  49                 if pnf_name == None:
  50                     raise Exception("Invalid PNF entry, could not extract `pnf-name`")
  51
  52                 global pnf_entries
  53                 pnf_entries[pnf_name] = pnf_entry
  54
  55                 httpServerLib.set_response_200_ok(self)
  56                 logger.debug('AAISetup PUT /setup/add_pnf_entry [' + pnf_name + '] -> 200 OK')
  57             elif re.search('/setup/add_service_instance', self.path):
  58                 service_instance_payload = httpServerLib.get_payload(self)
  59                 global service_instance
  60                 service_instance = json.loads(service_instance_payload)
  61                 httpServerLib.set_response_200_ok(self)
  62                 logger.debug('AAISetup PUT /setup/add_service_instance -> 200 OK')
  63             elif re.search('/setup/add_logical_link', self.path):
  64                 logical_link_payload = httpServerLib.get_payload(self)
  65                 logical_link_name = json.loads(logical_link_payload).get("link-name")
  66                 if logical_link_name == None:
  67                     raise Exception("Invalid logical link entry, could not extract `link-name`")
  68
  69                 global created_logical_link
  70                 created_logical_link = logical_link_payload
  71
  72                 httpServerLib.set_response_200_ok(self)
  73                 logger.debug('AAISetup PUT /setup/add_logical_link -> 200 OK')
  74
  75             elif re.search('/set_pnf', self.path):
  76                 pnf_name = httpServerLib.get_payload(self).decode()
  77                 pnf_entries[pnf_name] = AAI_RESOURCE_NOT_FOUND
  78                 httpServerLib.set_response_200_ok(self)
  79             else:
  80                 httpServerLib.set_response_404_not_found(self)
  81                 logger.info('AAISetup PUT ' + self.path + ' -> 404 Not found')
  82         except Exception as e:
  83             logger.error(e)
  84             httpServerLib.set_response_500_server_error(self)
  85
  86     def do_POST(self):
  87         try:
  88             if re.search('/reset', self.path):
  89                 global pnf_entries
  90                 global patched_pnf
  91                 global created_logical_link
  92                 global service_instance
  93                 pnf_entries = {}
  94                 patched_pnf = AAI_RESOURCE_NOT_FOUND
  95                 created_logical_link = AAI_RESOURCE_NOT_FOUND
  96                 service_instance = AAI_RESOURCE_NOT_FOUND
  97
  98                 httpServerLib.set_response_200_ok(self)
  99                 logger.debug('AAISetup POST /reset -> 200 OK')
 100             else:
 101                 httpServerLib.set_response_404_not_found(self)
 102                 logger.info('AAISetup POST ' + self.path + ' -> 404 Not found')
 103         except Exception as e:
 104             logger.error(e)
 105             httpServerLib.set_response_500_server_error(self)
 106
 107 class AAIHandler(BaseHTTPRequestHandler):
 108
 109     def do_GET(self):
 110         try:
 111             if re.search('/aai/v23/network/pnfs/pnf/[^/]*$', self.path):
 112                 pnf_name = basename(self.path)
 113                 if pnf_name in pnf_entries:
 114                     httpServerLib.set_response_200_ok(self, payload = pnf_entries[pnf_name])
 115                     logger.debug('AAIHandler GET /aai/v23/network/pnfs/pnf/' + pnf_name + ' -> 200 OK')
 116                 else:
 117                     httpServerLib.set_response_404_not_found(self)
 118                     logger.info('AAIHandler GET /aai/v23/network/pnfs/pnf/' + pnf_name + ' -> 404 Not found, actual entries: ' + str(pnf_entries.keys()))
 119             elif re.search('/aai/v23/network/logical-links/logical-link/[^/]*$', self.path):
 120                 logical_link_name = basename(self.path)
 121                 if json.loads(created_logical_link).get("link-name") == logical_link_name:
 122                     httpServerLib.set_response_200_ok(self, payload = created_logical_link)
 123                     logger.debug('AAIHandler GET /aai/v23/network/logical-links/logical-link/' + logical_link_name + ' -> 200 OK')
 124                 else:
 125                     httpServerLib.set_response_404_not_found(self)
 126                     logger.info('AAIHandler GET /aai/v23/network/logical-links/logical-link/' + logical_link_name + ' -> 404 Not found, actual link: ' + created_logical_link)
 127             elif re.search('aai/v23/business/customers/customer/Demonstration/service-subscriptions/service-subscription/vFW/service-instances/service-instance/bbs_service', self.path):
 128                 httpServerLib.set_response_200_ok(self, payload = json.dumps(service_instance).encode('utf-8'))
 129                 logger.debug('AAIHandler GET aai/v23/business/customers/customer/Demonstration/service-subscriptions/service-subscription/vFW/service-instances/service-instance/bbs_service -> 200 OK')
 130             else:
 131                 httpServerLib.set_response_404_not_found(self)
 132                 logger.info('AAIHandler GET ' + self.path + ' -> 404 Not found')
 133         except Exception as e:
 134             logger.error(e)
 135             httpServerLib.set_response_500_server_error(self)
 136
 137     def do_PATCH(self):
 138         try:
 139             if re.search('/aai/v23/network/pnfs/pnf/[^/]*$', self.path):
 140                 pnf_name = basename(self.path)
 141                 if pnf_name in pnf_entries:
 142                     global patched_pnf
 143                     patched_pnf = httpServerLib.get_payload(self)
 144
 145                     httpServerLib.set_response_200_ok(self)
 146                     logger.debug('AAIHandler PATCH /aai/v23/network/pnfs/pnf/' + pnf_name + ' -> 200 OK')
 147                 else:
 148                     httpServerLib.set_response_404_not_found(self)
 149                     logger.info('AAIHandler PATCH /aai/v23/network/pnfs/pnf/' + pnf_name + ' -> 404 Not found, actual entries: ' + str(pnf_entries.keys()))
 150             else:
 151                 httpServerLib.set_response_404_not_found(self)
 152                 logger.info('AAIHandler PATCH ' + self.path + ' -> 404 Not found')
 153         except Exception as e:
 154             logger.error(e)
 155             httpServerLib.set_response_500_server_error(self)
 156
 157     def do_PUT(self):
 158         try:
 159             if re.search('/aai/v23/network/logical-links/logical-link/[^/]*$', self.path):
 160                 global created_logical_link
 161                 created_logical_link = httpServerLib.get_payload(self)
 162
 163                 httpServerLib.set_response_200_ok(self)
 164
 165                 logical_link_name = basename(self.path)
 166                 logger.debug('AAIHandler PUT /aai/v23/network/logical-links/logical-link/' + logical_link_name + ' -> 200 OK')
 167             else:
 168                 httpServerLib.set_response_404_not_found(self)
 169                 logger.info('AAIHandler PUT ' + self.path + ' -> 404 Not found')
 170         except Exception as e:
 171             logger.error(e)
 172             httpServerLib.set_response_500_server_error(self)
 173
 174     def do_DELETE(self):
 175         try:
 176             if re.search('/aai/v23/network/logical-links/logical-link/[^/]*\?resource-version=\d+$', self.path):
 177                 httpServerLib.set_response_200_ok(self)
 178                 logical_link_name = re.search('.+?(?=\?)', basename(self.path)).group(0)
 179
 180                 global created_logical_link
 181                 if json.loads(created_logical_link).get("link-name") == logical_link_name:
 182                     created_logical_link = AAI_RESOURCE_NOT_FOUND
 183
 184                 logger.debug('AAIHandler DELETE /aai/v23/network/logical-links/logical-link/' + logical_link_name + ' -> 200 OK')
 185             else:
 186                 httpServerLib.set_response_404_not_found(self)
 187                 logger.info('AAIHandler DELETE ' + self.path + ' -> 404 Not found')
 188         except Exception as e:
 189             logger.error(e)
 190             httpServerLib.set_response_500_server_error(self)
 191
 192 def _main_(handler_class=AAIHandler, protocol="HTTP/1.0"):
 193     handler_class.protocol_version = protocol
 194     httpServerLib.start_http_endpoint(3333, AAIHandler)
 195     httpServerLib.start_https_endpoint(3334, AAIHandler, keyfile="certs/aai.key", certfile="certs/aai.crt", ca_certs="certs/root.crt")
 196     httpServerLib.start_http_endpoint(3335, AAISetup)
 197     while 1:
 198         time.sleep(10)
 199
 200
 201 if __name__ == '__main__':
 202     _main_()