|
@@ 2326-2350 (lines=25) @@
|
| 2323 |
|
self.napp.handle_flow_delete(event) |
| 2324 |
|
evc.set_flow_removed_at.assert_called_once() |
| 2325 |
|
|
| 2326 |
|
def test_add_bulk_metadata(self): |
| 2327 |
|
"""Test add_bulk_metadata method""" |
| 2328 |
|
evc_mock = create_autospec(EVC) |
| 2329 |
|
evc_mock.id = 1234 |
| 2330 |
|
self.napp.circuits = {"1234": evc_mock} |
| 2331 |
|
api = self.get_app_test_client(self.napp) |
| 2332 |
|
payload = { |
| 2333 |
|
"circuit_ids": ["1234"], |
| 2334 |
|
"metadata1": 1, |
| 2335 |
|
"metadata2": 2 |
| 2336 |
|
} |
| 2337 |
|
response = api.post( |
| 2338 |
|
f"{self.server_name_url}/v2/evc/metadata", |
| 2339 |
|
data=json.dumps(payload), |
| 2340 |
|
content_type="application/json", |
| 2341 |
|
) |
| 2342 |
|
self.assertEqual(response.status_code, 201) |
| 2343 |
|
args = self.napp.mongo_controller.update_evcs.call_args[0] |
| 2344 |
|
ids = payload.pop("circuit_ids") |
| 2345 |
|
self.assertEqual(args[0], ids) |
| 2346 |
|
self.assertEqual(args[1], payload) |
| 2347 |
|
self.assertEqual(args[2], "add") |
| 2348 |
|
calls = self.napp.mongo_controller.update_evcs.call_count |
| 2349 |
|
self.assertEqual(calls, 1) |
| 2350 |
|
evc_mock.extend_metadata.assert_called_with(payload) |
| 2351 |
|
|
| 2352 |
|
def test_add_bulk_metadata_no_id(self): |
| 2353 |
|
"""Test add_bulk_metadata with unknown evc id""" |
|
@@ 2384-2405 (lines=22) @@
|
| 2381 |
|
) |
| 2382 |
|
self.assertEqual(response.status_code, 400) |
| 2383 |
|
|
| 2384 |
|
def test_delete_bulk_metadata(self): |
| 2385 |
|
"""Test delete_metadata method""" |
| 2386 |
|
evc_mock = create_autospec(EVC) |
| 2387 |
|
evc_mock.id = 1234 |
| 2388 |
|
self.napp.circuits = {"1234": evc_mock} |
| 2389 |
|
api = self.get_app_test_client(self.napp) |
| 2390 |
|
payload = { |
| 2391 |
|
"circuit_ids": ["1234"] |
| 2392 |
|
} |
| 2393 |
|
response = api.delete( |
| 2394 |
|
f"{self.server_name_url}/v2/evc/metadata/metadata1", |
| 2395 |
|
data=json.dumps(payload), |
| 2396 |
|
content_type="application/json", |
| 2397 |
|
) |
| 2398 |
|
self.assertEqual(response.status_code, 200) |
| 2399 |
|
args = self.napp.mongo_controller.update_evcs.call_args[0] |
| 2400 |
|
self.assertEqual(args[0], payload["circuit_ids"]) |
| 2401 |
|
self.assertEqual(args[1], {"metadata1": ""}) |
| 2402 |
|
self.assertEqual(args[2], "del") |
| 2403 |
|
calls = self.napp.mongo_controller.update_evcs.call_count |
| 2404 |
|
self.assertEqual(calls, 1) |
| 2405 |
|
self.assertEqual(evc_mock.remove_metadata.call_count, 1) |
| 2406 |
|
|