Injector injector = getInjector(memStorage);
scheduler = injector.getInstance(TaskScheduler.class);
eventSink = PubsubTestUtil.startPubsub(injector);
ScheduledTask builder = TASK_A.newBuilder();
final IScheduledTask taskA = IScheduledTask.build(builder.setStatus(PENDING));
builder.getAssignedTask().setTaskId("b");
final IScheduledTask taskB = IScheduledTask.build(builder.setStatus(THROTTLED));
memStorage.write(new MutateWork.NoResult.Quiet() {
@Override
protected void execute(MutableStoreProvider store) {
store.getUnsafeTaskStore().saveTasks(ImmutableSet.of(taskA, taskB));