@Test public void assertUpdateExisted() { when(configService.load("transient_test_job")).thenReturn(Optional.of(transientJobConfig)); when(runningService.getRunningTasks("transient_test_job")) .thenReturn( Arrays.asList( TaskContext.from("transient_test_job@-@0@-@READY@-@SLAVE-S0@-@UUID"), TaskContext.from("transient_test_job@-@1@-@READY@-@SLAVE-S0@-@UUID"))); producerManager.update(transientJobConfig); verify(configService).update(transientJobConfig); verify(lifecycleService).killJob("transient_test_job"); verify(runningService).remove("transient_test_job"); verify(readyService).remove(Lists.newArrayList("transient_test_job")); }
@Test(expected = JobConfigurationException.class) public void assertUpdateNotExisted() { when(configService.load("transient_test_job")) .thenReturn(Optional.<CloudJobConfiguration>absent()); producerManager.update(transientJobConfig); }