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/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')
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_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`")
69 global created_logical_link
70 created_logical_link = logical_link_payload
72 httpServerLib.set_response_200_ok(self)
73 logger.debug('AAISetup PUT /setup/add_logical_link -> 200 OK')
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)
80 httpServerLib.set_response_404_not_found(self)
81 logger.info('AAISetup PUT ' + self.path + ' -> 404 Not found')
82 except Exception as e:
84 httpServerLib.set_response_500_server_error(self)
88 if re.search('/reset', self.path):
91 global created_logical_link
92 global service_instance
94 patched_pnf = AAI_RESOURCE_NOT_FOUND
95 created_logical_link = AAI_RESOURCE_NOT_FOUND
96 service_instance = AAI_RESOURCE_NOT_FOUND
98 httpServerLib.set_response_200_ok(self)
99 logger.debug('AAISetup POST /reset -> 200 OK')
101 httpServerLib.set_response_404_not_found(self)
102 logger.info('AAISetup POST ' + self.path + ' -> 404 Not found')
103 except Exception as e:
105 httpServerLib.set_response_500_server_error(self)
107 class AAIHandler(BaseHTTPRequestHandler):
111 if re.search('/aai/v12/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/v12/network/pnfs/pnf/' + pnf_name + ' -> 200 OK')
117 httpServerLib.set_response_404_not_found(self)
118 logger.info('AAIHandler GET /aai/v12/network/pnfs/pnf/' + pnf_name + ' -> 404 Not found, actual entries: ' + str(pnf_entries.keys()))
119 elif re.search('/aai/v12/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/v12/network/logical-links/logical-link/' + logical_link_name + ' -> 200 OK')
125 httpServerLib.set_response_404_not_found(self)
126 logger.info('AAIHandler GET /aai/v12/network/logical-links/logical-link/' + logical_link_name + ' -> 404 Not found, actual link: ' + created_logical_link)
127 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):
128 httpServerLib.set_response_200_ok(self, payload = json.dumps(service_instance).encode('utf-8'))
129 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')
131 httpServerLib.set_response_404_not_found(self)
132 logger.info('AAIHandler GET ' + self.path + ' -> 404 Not found')
133 except Exception as e:
135 httpServerLib.set_response_500_server_error(self)
139 if re.search('/aai/v12/network/pnfs/pnf/[^/]*$', self.path):
140 pnf_name = basename(self.path)
141 if pnf_name in pnf_entries:
143 patched_pnf = httpServerLib.get_payload(self)
145 httpServerLib.set_response_200_ok(self)
146 logger.debug('AAIHandler PATCH /aai/v12/network/pnfs/pnf/' + pnf_name + ' -> 200 OK')
148 httpServerLib.set_response_404_not_found(self)
149 logger.info('AAIHandler PATCH /aai/v12/network/pnfs/pnf/' + pnf_name + ' -> 404 Not found, actual entries: ' + str(pnf_entries.keys()))
151 httpServerLib.set_response_404_not_found(self)
152 logger.info('AAIHandler PATCH ' + self.path + ' -> 404 Not found')
153 except Exception as e:
155 httpServerLib.set_response_500_server_error(self)
159 if re.search('/aai/v12/network/logical-links/logical-link/[^/]*$', self.path):
160 global created_logical_link
161 created_logical_link = httpServerLib.get_payload(self)
163 httpServerLib.set_response_200_ok(self)
165 logical_link_name = basename(self.path)
166 logger.debug('AAIHandler PUT /aai/v12/network/logical-links/logical-link/' + logical_link_name + ' -> 200 OK')
168 httpServerLib.set_response_404_not_found(self)
169 logger.info('AAIHandler PUT ' + self.path + ' -> 404 Not found')
170 except Exception as e:
172 httpServerLib.set_response_500_server_error(self)
176 if re.search('/aai/v12/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)
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
184 logger.debug('AAIHandler DELETE /aai/v12/network/logical-links/logical-link/' + logical_link_name + ' -> 200 OK')
186 httpServerLib.set_response_404_not_found(self)
187 logger.info('AAIHandler DELETE ' + self.path + ' -> 404 Not found')
188 except Exception as e:
190 httpServerLib.set_response_500_server_error(self)
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)
201 if __name__ == '__main__':