aboutsummaryrefslogtreecommitdiffstats
path: root/app/test/event_based_scan/test_network_update.py
diff options
context:
space:
mode:
Diffstat (limited to 'app/test/event_based_scan/test_network_update.py')
-rw-r--r--app/test/event_based_scan/test_network_update.py27
1 files changed, 15 insertions, 12 deletions
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()))