MyriadDriver { public Status stop(boolean failover) { LOGGER.info("Stopping driver"); Status status = driver.stop(failover); LOGGER.info("Driver stopped with status: {}", status); return status; } @Inject  MyriadDriver(SchedulerDriver driver); Status stop(boolean failover); Status start(); Status kill(final TaskID taskId); Status abort(); SchedulerDriver getDriver();  }
@Test public void testStop() throws Exception { MyriadDriver driver = new MyriadDriver(new MockSchedulerDriver()); Status status = driver.stop(true); assertEquals(Protos.Status.DRIVER_STOPPED_VALUE, status.getNumber()); }