expect(api.createDrive(anyObject(DriveInfo.class))).andReturn(mockDrive);
api.imageDrive(image.getId(), mockDrive.getUuid(), ImageConversionType.GUNZIP);
// Set a custom exception message to make sure the exception is thrown at the right point
expectLastCall().andThrow(new IllegalStateException("imageDrive call failed"));
api.destroyDrive(mockDrive.getUuid());
expectLastCall();
replay(api);
ElasticStackComputeServiceAdapter adapter = new ElasticStackComputeServiceAdapter(api, driveNotClaimed,