.setMode(MaintenanceMode.DRAINED));
builder.add(createTransaction(Op.saveHostAttributes(hostAttributes2)));
expect(storageUtil.attributeStore.saveHostAttributes(
IHostAttributes.build(hostAttributes2.getHostAttributes()))).andReturn(true);
SaveLock saveLock = new SaveLock(new Lock().setKey(LockKey.job(JOB_KEY.newBuilder())));
builder.add(createTransaction(Op.saveLock(saveLock)));
storageUtil.lockStore.saveLock(ILock.build(saveLock.getLock()));
RemoveLock removeLock = new RemoveLock(LockKey.job(JOB_KEY.newBuilder()));
builder.add(createTransaction(Op.removeLock(removeLock)));