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

Improve testing

pull/30/head
Raphaël Vinot 2016-08-26 18:23:02 +02:00
parent aaaab590f5
commit 1532bc879f
1 changed files with 29 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
tests/test_offline.py
View File

@ -17,6 +17,7 @@ class TestOffline(unittest.TestCase):
self.domain = 'http://misp.local/' self.domain = 'http://misp.local/'
self.key = 'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa' self.key = 'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'
self.event = {'Event': json.load(open('tests/misp_event.json', 'r'))} self.event = {'Event': json.load(open('tests/misp_event.json', 'r'))}
self.new_misp_event = {'Event': json.load(open('tests/new_misp_event.json', 'r'))}
self.types = json.load(open('tests/describeTypes.json', 'r')) self.types = json.load(open('tests/describeTypes.json', 'r'))
self.sharing_groups = json.load(open('tests/sharing_groups.json', 'r')) self.sharing_groups = json.load(open('tests/sharing_groups.json', 'r'))
self.auth_error_msg = {"name": "Authentication failed. Please make sure you pass the API key of an API enabled user along in the Authorization header.", self.auth_error_msg = {"name": "Authentication failed. Please make sure you pass the API key of an API enabled user along in the Authorization header.",
@ -36,7 +37,7 @@ class TestOffline(unittest.TestCase):
def test_getEvent(self, m): def test_getEvent(self, m):
self.initURI(m) self.initURI(m)
pymisp = PyMISP(self.domain, self.key, debug=True) pymisp = PyMISP(self.domain, self.key)
e1 = pymisp.get_event(2) e1 = pymisp.get_event(2)
e2 = pymisp.get(2) e2 = pymisp.get(2)
self.assertEqual(e1, e2) self.assertEqual(e1, e2)
@ -44,7 +45,7 @@ class TestOffline(unittest.TestCase):
def test_updateEvent(self, m): def test_updateEvent(self, m):
self.initURI(m) self.initURI(m)
pymisp = PyMISP(self.domain, self.key, debug=True) pymisp = PyMISP(self.domain, self.key)
e0 = pymisp.update_event(2, json.dumps(self.event)) e0 = pymisp.update_event(2, json.dumps(self.event))
e1 = pymisp.update_event(2, self.event) e1 = pymisp.update_event(2, self.event)
self.assertEqual(e0, e1) self.assertEqual(e0, e1)
@ -54,7 +55,7 @@ class TestOffline(unittest.TestCase):
def test_deleteEvent(self, m): def test_deleteEvent(self, m):
self.initURI(m) self.initURI(m)
pymisp = PyMISP(self.domain, self.key, debug=True) pymisp = PyMISP(self.domain, self.key)
d = pymisp.delete_event(2) d = pymisp.delete_event(2)
self.assertEqual(d, {'message': 'Event deleted.'}) self.assertEqual(d, {'message': 'Event deleted.'})
d = pymisp.delete_event(3) d = pymisp.delete_event(3)
@ -62,13 +63,13 @@ class TestOffline(unittest.TestCase):
def test_deleteAttribute(self, m): def test_deleteAttribute(self, m):
self.initURI(m) self.initURI(m)
pymisp = PyMISP(self.domain, self.key, debug=True) pymisp = PyMISP(self.domain, self.key)
d = pymisp.delete_attribute(2) d = pymisp.delete_attribute(2)
self.assertEqual(d, {'message': 'Attribute deleted.'}) self.assertEqual(d, {'message': 'Attribute deleted.'})
def test_publish(self, m): def test_publish(self, m):
self.initURI(m) self.initURI(m)
pymisp = PyMISP(self.domain, self.key, debug=True) pymisp = PyMISP(self.domain, self.key)
e = pymisp.publish(self.event) e = pymisp.publish(self.event)
pub = self.event pub = self.event
pub['Event']['published'] = True pub['Event']['published'] = True
@ -78,7 +79,7 @@ class TestOffline(unittest.TestCase):
def test_getVersions(self, m): def test_getVersions(self, m):
self.initURI(m) self.initURI(m)
pymisp = PyMISP(self.domain, self.key, debug=True) pymisp = PyMISP(self.domain, self.key)
api_version = pymisp.get_api_version() api_version = pymisp.get_api_version()
self.assertEqual(api_version, {'version': pm.__version__}) self.assertEqual(api_version, {'version': pm.__version__})
server_version = pymisp.get_version() server_version = pymisp.get_version()
@ -86,17 +87,36 @@ class TestOffline(unittest.TestCase):
def test_getSharingGroups(self, m): def test_getSharingGroups(self, m):
self.initURI(m) self.initURI(m)
pymisp = PyMISP(self.domain, self.key, debug=True) pymisp = PyMISP(self.domain, self.key)
sharing_groups = pymisp.get_sharing_groups() sharing_groups = pymisp.get_sharing_groups()
self.assertEqual(sharing_groups, self.sharing_groups['response'][0]) self.assertEqual(sharing_groups, self.sharing_groups['response'][0])
def test_auth_error(self, m): def test_auth_error(self, m):
self.initURI(m) self.initURI(m)
pymisp = PyMISP(self.domain, self.key, debug=True) pymisp = PyMISP(self.domain, self.key)
error = pymisp.get(1) error = pymisp.get(1)
response = self.auth_error_msg response = self.auth_error_msg
response['errors'] = [response['message']] response['errors'] = [response['message']]
print(error) self.assertEqual(error, response)
def test_newEvent(self, m):
error_empty_info = {'message': 'The event could not be saved.', 'name': 'Add event failed.', 'errors': {'Event': {'info': ['Info cannot be empty.']}}, 'url': '/events/add'}
error_empty_info_flatten = {'message': 'The event could not be saved.', 'name': 'Add event failed.', 'errors': ["Error in info: ['Info cannot be empty.']"], 'url': '/events/add'}
self.initURI(m)
pymisp = PyMISP(self.domain, self.key)
with self.assertRaises(pm.api.NewEventError):
pymisp.new_event()
with self.assertRaises(pm.api.NewEventError):
pymisp.new_event(0)
with self.assertRaises(pm.api.NewEventError):
pymisp.new_event(0, 1)
m.register_uri('POST', self.domain + 'events', json=error_empty_info)
response = pymisp.new_event(0, 1, 0)
self.assertEqual(response, error_empty_info_flatten)
m.register_uri('POST', self.domain + 'events', json=self.new_misp_event)
response = pymisp.new_event(0, 1, 0, "This is a test.", '2016-08-26', False)
self.assertEqual(response, self.new_misp_event)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()