From 7e83d0876ddb84a45e130eeba28bc40ef53c074b Mon Sep 17 00:00:00 2001
From: Yaron Yogev <yaronyogev@gmail.com>
Date: Thu, 27 Jul 2017 09:02:54 +0300
Subject: Calipso initial release for OPNFV

Change-Id: I7210c244b0c10fa80bfa8c77cb86c9d6ddf8bc88
Signed-off-by: Yaron Yogev <yaronyogev@gmail.com>
---
 .../event_based_scan/test_event_delete_base.py     | 64 ++++++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 app/test/event_based_scan/test_event_delete_base.py

(limited to 'app/test/event_based_scan/test_event_delete_base.py')

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)
-- 
cgit 1.2.3-korg