1 files changed, 68 insertions, 0 deletions
diff --git a/app/test/event_based_scan/test_subnet_add.py b/app/test/event_based_scan/test_subnet_add.py
new file mode 100644
index 0000000..a8794ef
--- /dev/null
+++ b/app/test/event_based_scan/test_subnet_add.py
@@ -0,0 +1,68 @@
+###############################################################################
+# 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 unittest.mock import MagicMock
+
+from discover.events.event_subnet_add import EventSubnetAdd
+from discover.fetchers.api.api_access import ApiAccess
+from discover.find_links_for_pnics import FindLinksForPnics
+from discover.find_links_for_vservice_vnics import FindLinksForVserviceVnics
+from test.event_based_scan.test_data.event_payload_subnet_add import EVENT_PAYLOAD_SUBNET_ADD,\
+    EVENT_PAYLOAD_REGION, NETWORK_DOC
+from test.event_based_scan.test_event import TestEvent
+
+
+class TestSubnetAdd(TestEvent):
+
+    def test_handle_subnet_add(self):
+        self.values = EVENT_PAYLOAD_SUBNET_ADD
+        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)
+
+        network_document = self.inv.get_by_id(self.env, self.network_id)
+        if network_document:
+            # check subnet in network first.
+            self.assertNotIn(self.subnet['cidr'], network_document['cidrs'])
+        else:
+            self.log.info("network document is not found, add it first.")
+            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
+
+        # Mock function instead of get children data. They should be test in their unit test.
+        # add subnet document for updating network
+        handler = EventSubnetAdd()
+        handler.add_children_documents = MagicMock()
+
+        original_add_pnic_links = FindLinksForPnics.add_links
+        FindLinksForPnics.add_links = MagicMock()
+
+        original_add_vservice_links = FindLinksForVserviceVnics.add_links
+        FindLinksForVserviceVnics.add_links = MagicMock()
+
+        handler.handle(self.env, self.values)
+
+        # reset the methods back
+        FindLinksForPnics.add_links = original_add_pnic_links
+        FindLinksForVserviceVnics.add_links = original_add_vservice_links
+
+        # check network document
+        network_document = self.inv.get_by_id(self.env, self.network_id)
+        self.assertIn(self.subnet['cidr'], network_document['cidrs'])
+        self.assertIn(self.subnet['name'], network_document['subnets'])
+
+        #tearDown method has been implemented in class testEvent.