MyriadDriver { public SchedulerDriver getDriver() { return driver; } @Inject  MyriadDriver(SchedulerDriver driver); Status stop(boolean failover); Status start(); Status kill(final TaskID taskId); Status abort(); SchedulerDriver getDriver();  }
@Test public void testGetDriver() throws Exception { MyriadDriver driver = new MyriadDriver(new MockSchedulerDriver()); SchedulerDriver sDriver = driver.getDriver(); assertTrue(sDriver instanceof MockSchedulerDriver); }