MISP
/
PyMISP
mirror of https://github.com/MISP/PyMISP
2
0
Fork 0
Code Issues Releases Wiki Activity

new: Test for manual calls to add_object and add_object_reference

pull/382/head
Raphaël Vinot 2019-03-22 12:51:20 +01:00
parent cc35d81fa3
commit aeb0f913e4
1 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/testlive_comprehensive.py
View File

@ -935,6 +935,33 @@ class TestComprehensive(unittest.TestCase):
r = self.admin_misp_connector.enable_tag(tag['id'])
self.assertFalse(r['Tag']['hide_tag'])
def test_add_event_with_attachment_object_controller(self):
first = self.create_simple_event()
try:
first = self.user_misp_connector.add_event(first)
fo, peo, seos = make_binary_objects('tests/viper-test-files/test_files/whoami.exe')
for s in seos:
template_id = self.user_misp_connector.get_object_template_id(s.template_uuid)
r = self.user_misp_connector.add_object(first.id, template_id, s)
self.assertEqual(r['Object']['name'], 'pe-section')
template_id = self.user_misp_connector.get_object_template_id(peo.template_uuid)
r = self.user_misp_connector.add_object(first.id, template_id, peo)
self.assertEqual(r['Object']['name'], 'pe')
for ref in peo.ObjectReference:
r = self.user_misp_connector.add_object_reference(ref)
self.assertTrue('ObjectReference' in r)
template_id = self.user_misp_connector.get_object_template_id(fo.template_uuid)
r = self.user_misp_connector.add_object(first.id, template_id, fo)
self.assertEqual(r['Object']['name'], 'file')
for ref in fo.ObjectReference:
r = self.user_misp_connector.add_object_reference(ref)
self.assertTrue('ObjectReference' in r)
finally:
# Delete event
self.admin_misp_connector.delete_event(first.id)
def test_add_event_with_attachment(self):
first = self.create_simple_event()
try: