1 files changed, 64 insertions, 0 deletions
diff --git a/app/test/event_based_scan/test_event_delete_base.py b/app/test/event_based_scan/test_event_delete_base.py
new file mode 100644
index 0000000..1ccabb3
--- /dev/null
+++ b/app/test/event_based_scan/test_event_delete_base.py
@@ -0,0 +1,64 @@
+###############################################################################
+# 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 bson import ObjectId
+
+from discover.clique_finder import CliqueFinder
+from discover.events.event_base import EventBase
+from test.event_based_scan.test_event import TestEvent
+
+
+class TestEventDeleteBase(TestEvent):
+
+    def setUp(self):
+        super().setUp()
+        self.values = {}
+
+    def set_item_for_deletion(self, object_type, document):
+
+        payload = self.values['payload']
+        self.item_id = payload['{}_id'.format(object_type)]
+        if object_type == 'router':
+            host_id = self.values['publisher_id'].replace("network.", "", 1)
+            self.item_id = "-".join([host_id, "qrouter", self.item_id])
+
+        self.assertEqual(document['id'], self.item_id, msg="Document id and payload id are different")
+
+        item = self.inv.get_by_id(self.env, self.item_id)
+        if not item:
+            self.log.info('{} document is not found, add document for deleting.'.format(object_type))
+
+            # add network document for deleting.
+            self.set_item(document)
+            item = self.inv.get_by_id(self.env, self.item_id)
+            self.assertIsNotNone(item)
+
+    def handle_delete(self, handler: EventBase):
+
+        item = self.inv.get_by_id(self.env, self.item_id)
+        db_id = ObjectId(item['_id'])
+        clique_finder = CliqueFinder()
+
+        # delete item
+        event_result = handler.handle(self.env, self.values)
+        self.assertTrue(event_result.result)
+
+        # check instance delete result.
+        item = self.inv.get_by_id(self.env, self.item_id)
+        self.assertIsNone(item)
+
+        # check links
+        matched_links_source = clique_finder.find_links_by_source(db_id)
+        matched_links_target = clique_finder.find_links_by_target(db_id)
+        self.assertEqual(matched_links_source.count(), 0)
+        self.assertEqual(matched_links_target.count(), 0)
+
+        # check children
+        matched_children = self.inv.get_children(self.env, None, self.item_id)
+        self.assertEqual(len(matched_children), 0)