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'{}'
24 patched_pnf = AAI_RESOURCE_NOT_FOUND
25 created_logical_link = AAI_RESOURCE_NOT_FOUND
26 service_instance = AAI_RESOURCE_NOT_FOUND
28 class AAISetup(BaseHTTPRequestHandler):
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])
45 httpServerLib.set_response_404_not_found(self)
46 logger.info('AAISetup GET ' + self.path + ' -> 404 Not found')
47 except Exception as e:
49 httpServerLib.set_response_500_server_error(self)
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")
57 raise Exception("Invalid PNF entry, could not extract `pnf-name`")
60 pnf_entries[pnf_name] = pnf_entry
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`")
77 logical_links[logical_link_name] = logical_link_payload
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')
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)
88 httpServerLib.set_response_404_not_found(self)
89 logger.info('AAISetup PUT ' + self.path + ' -> 404 Not found')
90 except Exception as e:
92 httpServerLib.set_response_500_server_error(self)
96 if re.search('/reset', self.path):
99 global created_logical_link
100 global service_instance
102 patched_pnf = AAI_RESOURCE_NOT_FOUND
103 created_logical_link = AAI_RESOURCE_NOT_FOUND
104 service_instance = AAI_RESOURCE_NOT_FOUND
106 httpServerLib.set_response_200_ok(self)
107 logger.debug('AAISetup POST /reset -> 200 OK')
109 httpServerLib.set_response_404_not_found(self)
110 logger.info('AAISetup POST ' + self.path + ' -> 404 Not found')
111 except Exception as e:
113 httpServerLib.set_response_500_server_error(self)
115 class AAIHandler(BaseHTTPRequestHandler):
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')
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')
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')
139 httpServerLib.set_response_404_not_found(self)
140 logger.info('AAIHandler GET ' + self.path + ' -> 404 Not found')
141 except Exception as e:
143 httpServerLib.set_response_500_server_error(self)
147 if re.search('/aai/v12/network/pnfs/pnf/[^/]*$', self.path):
148 pnf_name = basename(self.path)
149 if pnf_name in pnf_entries:
151 patched_pnf = httpServerLib.get_payload(self)
153 httpServerLib.set_response_200_ok(self)
154 logger.debug('AAIHandler PATCH /aai/v12/network/pnfs/pnf/' + pnf_name + ' -> 200 OK')
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()))
159 httpServerLib.set_response_404_not_found(self)
160 logger.info('AAIHandler PATCH ' + self.path + ' -> 404 Not found')
161 except Exception as e:
163 httpServerLib.set_response_500_server_error(self)
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)
171 httpServerLib.set_response_200_ok(self)
173 logical_link_name = basename(self.path)
174 logger.debug('AAIHandler PUT /aai/v12/network/logical-links/logical-link/' + logical_link_name + ' -> 200 OK')
176 httpServerLib.set_response_404_not_found(self)
177 logger.info('AAIHandler PUT ' + self.path + ' -> 404 Not found')
178 except Exception as e:
180 httpServerLib.set_response_500_server_error(self)
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)
188 del logical_links[logical_link_name]
189 logger.debug('AAIHandler DELETE /aai/v12/network/logical-links/logical-link/' + logical_link_name + ' -> 200 OK')
191 httpServerLib.set_response_404_not_found(self)
192 logger.info('AAIHandler DELETE ' + self.path + ' -> 404 Not found')
193 except Exception as e:
195 httpServerLib.set_response_500_server_error(self)
197 def _main_(handler_class=AAIHandler, protocol="HTTP/1.0"):
198 handler_class.protocol_version = protocol
199 httpServerLib.start_https_endpoint(3334, AAIHandler, keyfile="certs/aai.key", certfile="certs/aai.crt", ca_certs="certs/root.crt")
200 httpServerLib.start_http_endpoint(3335, AAISetup)
205 if __name__ == '__main__':