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')