[UnitTest] Add some functions, testing.

pull/3/head
Raphaël Vinot 2015-09-26 21:00:33 +02:00
parent cc8400a9df
commit 1ac11fa28b
1 changed files with 56 additions and 12 deletions

View File

@ -33,11 +33,12 @@ class TestBasic(unittest.TestCase):
def new_event(self): def new_event(self):
event = self.misp.new_event(0, 1, 0, "This is a test") event = self.misp.new_event(0, 1, 0, "This is a test")
event_id = self._clean_event(event) event_id = self._clean_event(event)
self.assertEqual(event, {u'Event': {u'info': u'This is a test', u'locked': False, to_check = {u'Event': {u'info': u'This is a test', u'locked': False,
u'attribute_count': u'0', u'analysis': u'0', u'attribute_count': u'0', u'analysis': u'0',
u'ShadowAttribute': [], u'published': False, u'ShadowAttribute': [], u'published': False,
u'distribution': u'0', u'Attribute': [], u'proposal_email_lock': False, u'distribution': u'0', u'Attribute': [], u'proposal_email_lock': False,
u'threat_level_id': u'1'}}) u'threat_level_id': u'1'}},
self.assertEqual(event, to_check, 'Failed at creating a new Event')
return int(event_id) return int(event_id)
def add_hashes(self, eventid): def add_hashes(self, eventid):
@ -59,7 +60,7 @@ class TestBasic(unittest.TestCase):
u'to_ids': False, u'value': u'dll_installer.dll|003315b0aea2fcb9f77d29223dd8947d0e6792b3a0227e054be8eb2a11f443d9', u'to_ids': False, u'value': u'dll_installer.dll|003315b0aea2fcb9f77d29223dd8947d0e6792b3a0227e054be8eb2a11f443d9',
u'ShadowAttribute': [], u'distribution': u'2', u'type': u'filename|sha256'}], u'ShadowAttribute': [], u'distribution': u'2', u'type': u'filename|sha256'}],
u'proposal_email_lock': False, u'threat_level_id': u'1'}} u'proposal_email_lock': False, u'threat_level_id': u'1'}}
self.assertEqual(event, to_check) self.assertEqual(event, to_check, 'Failed at adding hashes')
def publish(self, eventid): def publish(self, eventid):
r = self.misp.get_event(eventid) r = self.misp.get_event(eventid)
@ -80,19 +81,62 @@ class TestBasic(unittest.TestCase):
u'to_ids': False, u'value': u'dll_installer.dll|003315b0aea2fcb9f77d29223dd8947d0e6792b3a0227e054be8eb2a11f443d9', u'to_ids': False, u'value': u'dll_installer.dll|003315b0aea2fcb9f77d29223dd8947d0e6792b3a0227e054be8eb2a11f443d9',
u'ShadowAttribute': [], u'distribution': u'2', u'type': u'filename|sha256'}], u'ShadowAttribute': [], u'distribution': u'2', u'type': u'filename|sha256'}],
u'proposal_email_lock': False, u'threat_level_id': u'1'}} u'proposal_email_lock': False, u'threat_level_id': u'1'}}
self.assertEqual(event, to_check) self.assertEqual(event, to_check, 'Failed at publishing event')
def delete(self, eventid): def delete(self, eventid):
event = self.misp.delete_event(eventid) event = self.misp.delete_event(eventid)
event.json() print event.json()
def test_all(self): def delete_attr(self, attrid):
event = self.misp.delete_attribute(attrid)
print event.json()
def get(self, eventid):
event = self.misp.get_event(eventid)
print event.json()
def add(self):
event = {u'Event': {u'info': u'This is a test', u'locked': False,
u'attribute_count': u'3', u'analysis': u'0',
u'ShadowAttribute': [], u'published': False, u'distribution': u'0',
u'Attribute': [
{u'category': u'Payload installation', u'comment': u'Fanny modules',
u'to_ids': False, u'value': u'dll_installer.dll|0a209ac0de4ac033f31d6ba9191a8f7a',
u'ShadowAttribute': [], u'distribution': u'2', u'type': u'filename|md5'},
{u'category': u'Payload installation', u'comment': u'Fanny modules',
u'to_ids': False, u'value': u'dll_installer.dll|1f0ae54ac3f10d533013f74f48849de4e65817a7',
u'ShadowAttribute': [], u'distribution': u'2', u'type': u'filename|sha1'},
{u'category': u'Payload installation', u'comment': u'Fanny modules',
u'to_ids': False, u'value': u'dll_installer.dll|003315b0aea2fcb9f77d29223dd8947d0e6792b3a0227e054be8eb2a11f443d9',
u'ShadowAttribute': [], u'distribution': u'2', u'type': u'filename|sha256'}],
u'proposal_email_lock': False, u'threat_level_id': u'1'}}
event = self.misp.add_event(event)
print event.json()
def test_create_event(self):
eventid = self.new_event() eventid = self.new_event()
time.sleep(1) time.sleep(1)
self.add_hashes(eventid)
time.sleep(1)
self.publish(eventid)
self.delete(eventid) self.delete(eventid)
def test_get_event(self):
eventid = self.new_event()
time.sleep(1)
self.get(eventid)
time.sleep(1)
self.delete(eventid)
def test_add_event(self):
self.add()
time.sleep(1)
self.delete(1)
def test_del_attr(self):
eventid = self.new_event()
time.sleep(1)
self.delete_attr(1)
time.sleep(1)
self.delete(eventid)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()