|
@@ 2292-2316 (lines=25) @@
|
| 2289 |
|
self.napp.handle_flow_delete(event) |
| 2290 |
|
evc.set_flow_removed_at.assert_called_once() |
| 2291 |
|
|
| 2292 |
|
def test_add_bulk_metadata(self): |
| 2293 |
|
"""Test add_bulk_metadata method""" |
| 2294 |
|
evc_mock = create_autospec(EVC) |
| 2295 |
|
evc_mock.id = 1234 |
| 2296 |
|
self.napp.circuits = {"1234": evc_mock} |
| 2297 |
|
api = self.get_app_test_client(self.napp) |
| 2298 |
|
payload = { |
| 2299 |
|
"circuit_ids": ["1234"], |
| 2300 |
|
"metadata1": 1, |
| 2301 |
|
"metadata2": 2 |
| 2302 |
|
} |
| 2303 |
|
response = api.post( |
| 2304 |
|
f"{self.server_name_url}/v2/evc/metadata", |
| 2305 |
|
data=json.dumps(payload), |
| 2306 |
|
content_type="application/json", |
| 2307 |
|
) |
| 2308 |
|
self.assertEqual(response.status_code, 201) |
| 2309 |
|
args = self.napp.mongo_controller.update_evcs.call_args[0] |
| 2310 |
|
ids = payload.pop("circuit_ids") |
| 2311 |
|
self.assertEqual(args[0], ids) |
| 2312 |
|
self.assertEqual(args[1], payload) |
| 2313 |
|
self.assertEqual(args[2], "add") |
| 2314 |
|
calls = self.napp.mongo_controller.update_evcs.call_count |
| 2315 |
|
self.assertEqual(calls, 1) |
| 2316 |
|
evc_mock.extend_metadata.assert_called_with(payload) |
| 2317 |
|
|
| 2318 |
|
def test_add_bulk_metadata_no_id(self): |
| 2319 |
|
"""Test add_bulk_metadata with unknown evc id""" |
|
@@ 2350-2371 (lines=22) @@
|
| 2347 |
|
) |
| 2348 |
|
self.assertEqual(response.status_code, 400) |
| 2349 |
|
|
| 2350 |
|
def test_delete_bulk_metadata(self): |
| 2351 |
|
"""Test delete_metadata method""" |
| 2352 |
|
evc_mock = create_autospec(EVC) |
| 2353 |
|
evc_mock.id = 1234 |
| 2354 |
|
self.napp.circuits = {"1234": evc_mock} |
| 2355 |
|
api = self.get_app_test_client(self.napp) |
| 2356 |
|
payload = { |
| 2357 |
|
"circuit_ids": ["1234"] |
| 2358 |
|
} |
| 2359 |
|
response = api.delete( |
| 2360 |
|
f"{self.server_name_url}/v2/evc/metadata/metadata1", |
| 2361 |
|
data=json.dumps(payload), |
| 2362 |
|
content_type="application/json", |
| 2363 |
|
) |
| 2364 |
|
self.assertEqual(response.status_code, 200) |
| 2365 |
|
args = self.napp.mongo_controller.update_evcs.call_args[0] |
| 2366 |
|
self.assertEqual(args[0], payload["circuit_ids"]) |
| 2367 |
|
self.assertEqual(args[1], {"metadata1": ""}) |
| 2368 |
|
self.assertEqual(args[2], "del") |
| 2369 |
|
calls = self.napp.mongo_controller.update_evcs.call_count |
| 2370 |
|
self.assertEqual(calls, 1) |
| 2371 |
|
self.assertEqual(evc_mock.remove_metadata.call_count, 1) |
| 2372 |
|
|