6 from http.server import BaseHTTPRequestHandler
7 from os.path import basename
10 ch = logging.StreamHandler(sys.stdout)
14 format='[%(asctime)s] {%(filename)s:%(lineno)d} %(levelname)s - %(message)s',
18 logger = logging.getLogger('AAI-simulator-logger')
20 AAI_RESOURCE_NOT_FOUND = b'{}'
23 patched_pnf = AAI_RESOURCE_NOT_FOUND
24 created_logical_link = AAI_RESOURCE_NOT_FOUND
25 service_instance = AAI_RESOURCE_NOT_FOUND
27 class AAISetup(BaseHTTPRequestHandler):
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/created_logical_link', self.path):
35 httpServerLib.set_response_200_ok(self, payload = created_logical_link)
36 logger.debug('AAISetup GET /setup/created_logical_link -> 200 OK')
38 httpServerLib.set_response_404_not_found(self)
39 logger.info('AAISetup GET ' + self.path + ' -> 404 Not found')
40 except Exception as e:
42 httpServerLib.set_response_500_server_error(self)
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")
50 raise Exception("Invalid PNF entry, could not extract `pnf-name`")
53 pnf_entries[pnf_name] = pnf_entry
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_instace', 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_instace -> 200 OK')
64 elif re.search('/set_pnf', self.path):
65 pnf_name = httpServerLib.get_payload(self).decode()
66 pnf_entries[pnf_name] = AAI_RESOURCE_NOT_FOUND
67 httpServerLib.set_response_200_ok(self)
69 httpServerLib.set_response_404_not_found(self)
70 logger.info('AAISetup PUT ' + self.path + ' -> 404 Not found')
71 except Exception as e:
73 httpServerLib.set_response_500_server_error(self)
77 if re.search('/reset', self.path):
80 global created_logical_link
81 global service_instance
83 patched_pnf = AAI_RESOURCE_NOT_FOUND
84 created_logical_link = AAI_RESOURCE_NOT_FOUND
85 service_instance = AAI_RESOURCE_NOT_FOUND
87 httpServerLib.set_response_200_ok(self)
88 logger.debug('AAISetup POST /reset -> 200 OK')
90 httpServerLib.set_response_404_not_found(self)
91 logger.info('AAISetup POST ' + self.path + ' -> 404 Not found')
92 except Exception as e:
94 httpServerLib.set_response_500_server_error(self)
96 class AAIHandler(BaseHTTPRequestHandler):
100 if re.search('/aai/v12/network/pnfs/pnf/[^/]*$', self.path):
101 pnf_name = basename(self.path)
102 if pnf_name in pnf_entries:
103 httpServerLib.set_response_200_ok(self, payload = pnf_entries[pnf_name])
104 logger.debug('AAIHandler GET /aai/v12/network/pnfs/pnf/' + pnf_name + ' -> 200 OK')
106 httpServerLib.set_response_404_not_found(self)
107 logger.info('AAIHandler GET /aai/v12/network/pnfs/pnf/' + pnf_name + ' -> 404 Not found, actual entries: ' + str(pnf_entries.keys()))
108 elif re.search('aai/v12/business/customers/customer/Demonstration/service-subscriptions/service-subscription/vFW/service-instances/service-instance/bbs_service', self.path):
109 httpServerLib.set_response_200_ok(self, payload = service_instance)
110 logger.debug('AAIHandler GET aai/v12/business/customers/customer/Demonstration/service-subscriptions/service-subscription/vFW/service-instances/service-instance/bbs_service -> 200 OK')
112 httpServerLib.set_response_404_not_found(self)
113 logger.info('AAIHandler GET ' + self.path + ' -> 404 Not found')
114 except Exception as e:
116 httpServerLib.set_response_500_server_error(self)
120 if re.search('/aai/v12/network/pnfs/pnf/[^/]*$', self.path):
121 pnf_name = basename(self.path)
122 if pnf_name in pnf_entries:
124 patched_pnf = httpServerLib.get_payload(self)
126 httpServerLib.set_response_200_ok(self)
127 logger.debug('AAIHandler PATCH /aai/v12/network/pnfs/pnf/' + pnf_name + ' -> 200 OK')
129 httpServerLib.set_response_404_not_found(self)
130 logger.info('AAIHandler PATCH /aai/v12/network/pnfs/pnf/' + pnf_name + ' -> 404 Not found, actual entries: ' + str(pnf_entries.keys()))
132 httpServerLib.set_response_404_not_found(self)
133 logger.info('AAIHandler PATCH ' + self.path + ' -> 404 Not found')
134 except Exception as e:
136 httpServerLib.set_response_500_server_error(self)
140 if re.search('/aai/v12/network/logical-links/logical-link/[^/]*$', self.path):
141 global created_logical_link
142 created_logical_link = httpServerLib.get_payload(self)
144 httpServerLib.set_response_200_ok(self)
146 logical_link_name = basename(self.path)
147 logger.debug('AAIHandler PUT /aai/v12/network/logical-links/logical-link/' + logical_link_name + ' -> 200 OK')
149 httpServerLib.set_response_404_not_found(self)
150 logger.info('AAIHandler PUT ' + self.path + ' -> 404 Not found')
151 except Exception as e:
153 httpServerLib.set_response_500_server_error(self)
156 def _main_(handler_class=AAIHandler, protocol="HTTP/1.0"):
157 handler_class.protocol_version = protocol
158 httpServerLib.start_https_endpoint(3334, AAIHandler, keyfile="certs/aai.key", certfile="certs/aai.crt", ca_certs="certs/root.crt")
159 httpServerLib.start_http_endpoint(3335, AAISetup)
164 if __name__ == '__main__':