def testTurbiniaRequestSerializationBadJSON(self):
"""Tests that TurbiniaRequest will raise error on wrong JSON object."""
rawdisk = evidence.RawDisk(name='My Evidence', local_path='/tmp/foo')
rawdisk_json = rawdisk.to_json()
self.assertTrue(isinstance(rawdisk_json, str))
request_new = pubsub.TurbiniaRequest()
# Try to load serialization RawDisk() into a TurbiniaRequest, which should
# error because this is not the correct type.
self.assertRaises(TurbiniaException, request_new.from_json,
rawdisk_json)
def getTurbiniaRequest():
"""Get a Turbinia Request object with valid evidence attached.
Returns:
TurbiniaRequest object.
"""
request = pubsub.TurbiniaRequest(request_id='deadbeef',
context={'kw': [1, 2]})
rawdisk = evidence.RawDisk(name='My Evidence',
local_path='/tmp/foo',
mount_path='/mnt/foo')
request.evidence.append(rawdisk)
return request
def testTurbiniaRequestSerialization(self):
"""Test that TurbiniaRequests serializes/unserializes."""
request = getTurbiniaRequest()
request_json = request.to_json()
self.assertTrue(isinstance(request_json, str))
# Create a new Turbinia Request object to load our results into
request_new = pubsub.TurbiniaRequest()
request_new.from_json(request_json)
self.assertTrue(isinstance(request_new, pubsub.TurbiniaRequest))
self.assertTrue(request_new.context['kw'][1], 2)
self.assertTrue(request_new.request_id, 'deadbeef')
self.assertTrue(isinstance(request_new.evidence[0], evidence.RawDisk))
self.assertEqual(request_new.evidence[0].name, 'My Evidence')
def testTurbiniaRequestSerializationBadData(self):
"""Tests that TurbiniaRequest will raise error on non-json data."""
request_new = pubsub.TurbiniaRequest()
self.assertRaises(TurbiniaException, request_new.from_json,
'non-json-data')