chg: [tests] Add custom error message on upload_sample - fix last commit.

pull/418/head
Raphaël Vinot 2019-07-05 10:34:48 +02:00
parent 5aa94d277b
commit 10bd88cb06
1 changed files with 2 additions and 2 deletions

View File

@ -879,8 +879,8 @@ class TestComprehensive(unittest.TestCase):
with open('tests/testlive_comprehensive.py', 'rb') as f:
response = self.user_misp_connector.upload_sample(filename='testfile.py', filepath_or_bytes=f.read(),
event_id=first.id)
self.assertEqual(response['message'], 'Success, saved all attributes.', "Content of response: {}".format(response))
self.assertTrue('message' in response, "Content of response: {}".format(response))
self.assertEqual(response['message'], 'Success, saved all attributes.')
first = self.user_misp_connector.get_event(first.id)
self.assertEqual(len(first.objects), 1)
self.assertEqual(first.objects[0].name, 'file')