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