MyriadDriver { public Status start() { LOGGER.info("Starting driver"); Status status = driver.start(); LOGGER.info("Driver started 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 testStart() throws Exception { MyriadDriver driver = new MyriadDriver(new MockSchedulerDriver()); Status status = driver.start(); assertEquals(Protos.Status.DRIVER_RUNNING_VALUE, status.getNumber()); }