From 2f9d5cf583b4b5dd635f386cb9ac07435bd44cc8 Mon Sep 17 00:00:00 2001 From: Ilia Abashin Date: Fri, 25 Aug 2017 14:24:59 +0300 Subject: Heavily refactored all event based scans Restricted real mongo interaction. Still work to do, but it's a good start. Fixed a bug with old subnets not being properly deleted. Change-Id: I5f260e09f0e11a477a47cb031d397a454465123f Signed-off-by: Ilia Abashin --- app/test/event_based_scan/test_network_update.py | 27 +++++++++++++----------- 1 file changed, 15 insertions(+), 12 deletions(-) (limited to 'app/test/event_based_scan/test_network_update.py') diff --git a/app/test/event_based_scan/test_network_update.py b/app/test/event_based_scan/test_network_update.py index bf9eee4..90330c0 100644 --- a/app/test/event_based_scan/test_network_update.py +++ b/app/test/event_based_scan/test_network_update.py @@ -8,8 +8,9 @@ # http://www.apache.org/licenses/LICENSE-2.0 # ############################################################################### from discover.events.event_network_update import EventNetworkUpdate -from test.event_based_scan.test_data.event_payload_network_update import EVENT_PAYLOAD_NETWORK_UPDATE, \ - NETWORK_DOCUMENT +from test.event_based_scan.test_data.event_payload_network_update import \ + EVENT_PAYLOAD_NETWORK_UPDATE, \ + NETWORK_DOCUMENT, UPDATED_NETWORK_FIELDS from test.event_based_scan.test_event import TestEvent @@ -18,16 +19,18 @@ class TestNetworkUpdate(TestEvent): def test_handle_network_update(self): self.values = EVENT_PAYLOAD_NETWORK_UPDATE self.payload = self.values['payload'] - self.network = self.payload['network'] - name = self.network['name'] - status = self.network['admin_state_up'] - self.network_id = self.network['id'] - self.item_ids.append(self.network_id) - self.set_item(NETWORK_DOCUMENT) + network = NETWORK_DOCUMENT + self.inv.get_by_id.return_value = network - EventNetworkUpdate().handle(self.env, self.values) + res = EventNetworkUpdate().handle(self.env, self.values) - network_document = self.inv.get_by_id(self.env, self.network_id) - self.assertEqual(network_document['name'], name) - self.assertEqual(network_document['admin_state_up'], status) + self.assertTrue(res.result) + self.assertTrue(self.inv.values_replace.called) + self.assertTrue(self.inv.set.called) + + # check that all changed fields are updated + call_args, _ = self.inv.set.call_args + # Assert that all updated fields have been added to db + self.assertTrue(all(item in call_args[0].items() + for item in UPDATED_NETWORK_FIELDS.items())) -- cgit 1.2.3-korg