fix: [test] Fix event index tests

pull/7748/head
Jakub Onderka 2021-10-15 16:27:50 +02:00
parent af6521ab0a
commit 5237b2e10f
1 changed files with 12 additions and 9 deletions

View File

@ -105,27 +105,26 @@ class TestComprehensive(unittest.TestCase):
# No event should exists
index = self.user_misp_connector.search_index(eventinfo=event.info)
self.assertGreater(len(index), 0)
self.assertEqual(len(index), 0, "No event should exists")
event = self.user_misp_connector.add_event(event)
check_response(event)
# One event should exists
index = self.user_misp_connector.search_index(eventinfo=event.info)
self.assertGreater(len(index), 1)
self.assertEqual(len(index), 1)
self.assertEqual(index[0].uuid, event.uuid)
self.user_misp_connector.delete_event(event)
def test_search_index_by_tag(self):
tags = self.user_misp_connector.search_tags("tlp:red", True)
print(tags)
index = self.user_misp_connector.search_index(tags="tlp:red")
self.assertGreater(len(index), 0, "No event should exists")
self.assertEqual(len(index), 0, "No event should exists")
index = self.user_misp_connector.search_index(tags=tags[0].id)
self.assertGreater(len(index), 0, "No event should exists")
self.assertEqual(len(index), 0, "No event should exists")
event = create_simple_event()
event.add_tag("tlp:red")
@ -133,23 +132,23 @@ class TestComprehensive(unittest.TestCase):
check_response(event)
index = self.user_misp_connector.search_index(tags="tlp:red")
self.assertGreater(len(index), 1, "One event should exists")
self.assertEqual(len(index), 1, "One event should exists")
index = self.user_misp_connector.search_index(tags=tags[0].id)
self.assertGreater(len(index), 1, "One event should exists")
self.assertEqual(len(index), 1, "One event should exists")
self.user_misp_connector.delete_event(event)
def test_search_index_by_email(self):
index = self.user_misp_connector.search_index(email=self.test_usr.email)
self.assertGreater(len(index), 0, "No event should exists")
self.assertEqual(len(index), 0, index)
event = create_simple_event()
event = self.user_misp_connector.add_event(event)
check_response(event)
index = self.user_misp_connector.search_index(email=self.test_usr.email)
self.assertGreater(len(index), 1, "One event should exists")
self.assertEqual(len(index), 1, "One event should exists")
self.user_misp_connector.delete_event(event)
@ -172,6 +171,10 @@ class TestComprehensive(unittest.TestCase):
minimal_published = self.user_misp_connector.search_index(minimal=True, published=True)
self.assertEqual(len(minimal_published), 0, "No event should be published.")
def test_search_index_minimal_by_org(self):
# pythonify is not supported for minimal results
self.user_misp_connector.global_pythonify = False
# Create test event
event = create_simple_event()
event = self.user_misp_connector.add_event(event, pythonify=True)