SchedulerDriver driver = mock(SchedulerDriver.class);
when(driver.killTask(null)).thenReturn(Status.DRIVER_RUNNING);
SchedulerDriverSupplier driverSupplier = new SchedulerDriverSupplier();
driverSupplier.setSchedulerDriver(driver);
binder.bind(SchedulerDriverSupplier.class).toInstance(driverSupplier);
}
}));