[UnitTest] Add some functions, testing.

tests/test.py

@@ -33,11 +33,12 @@ class TestBasic(unittest.TestCase):
     def new_event(self):
         event = self.misp.new_event(0, 1, 0, "This is a test")
         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'ShadowAttribute': [], u'published': 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)
 
     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'ShadowAttribute': [], u'distribution': u'2', u'type': u'filename|sha256'}],
                                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):
         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'ShadowAttribute': [], u'distribution': u'2', u'type': u'filename|sha256'}],
                                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):
         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()
         time.sleep(1)
-        self.add_hashes(eventid)
-        time.sleep(1)
-        self.publish(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__':
     unittest.main()