1 files changed, 45 insertions, 0 deletions
diff --git a/app/test/event_based_scan/test_subnet_update.py b/app/test/event_based_scan/test_subnet_update.py
new file mode 100644
index 0000000..eddfe84
--- /dev/null
+++ b/app/test/event_based_scan/test_subnet_update.py
@@ -0,0 +1,45 @@
+###############################################################################
+# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems)   #
+# and others                                                                  #
+#                                                                             #
+# All rights reserved. This program and the accompanying materials            #
+# are made available under the terms of the Apache License, Version 2.0       #
+# which accompanies this distribution, and is available at                    #
+# http://www.apache.org/licenses/LICENSE-2.0                                  #
+###############################################################################
+from discover.events.event_subnet_update import EventSubnetUpdate
+from discover.fetchers.api.api_access import ApiAccess
+from test.event_based_scan.test_data.event_payload_subnet_add import  \
+    EVENT_PAYLOAD_REGION
+from test.event_based_scan.test_data.event_payload_subnet_update import EVENT_PAYLOAD_SUBNET_UPDATE, NETWORK_DOC
+from test.event_based_scan.test_event import TestEvent
+
+
+class TestSubnetUpdate(TestEvent):
+
+    def test_handle_subnet_add(self):
+        self.values = EVENT_PAYLOAD_SUBNET_UPDATE
+        self.payload = self.values['payload']
+        self.subnet = self.payload['subnet']
+        self.subnet_id = self.subnet['id']
+        self.network_id = self.subnet['network_id']
+        self.item_ids.append(self.network_id)
+
+        #add network document for subnet.
+        self.set_item(NETWORK_DOC)
+
+        # check network document
+        network_document = self.inv.get_by_id(self.env, self.network_id)
+        self.assertIsNotNone(network_document)
+
+        # check region data.
+        if not ApiAccess.regions:
+            ApiAccess.regions = EVENT_PAYLOAD_REGION
+
+        handler = EventSubnetUpdate()
+        handler.handle(self.env, self.values)
+
+        # check network document
+        network_document = self.inv.get_by_id(self.env, self.network_id)
+        self.assertIn(self.subnet['name'], network_document['subnets'])
+        self.assertEqual(self.subnet['gateway_ip'], network_document['subnets'][self.subnet['name']]['gateway_ip'])