private Capture<Runnable> expectJobAccepted(IJobConfiguration savedJob) throws Exception {
expectJobValidated(savedJob);
storageUtil.jobStore.saveAcceptedJob(MANAGER_KEY, savedJob);
Capture<Runnable> jobTriggerCapture = createCapture();
expect(cronScheduler.schedule(eq(savedJob.getCronSchedule()), capture(jobTriggerCapture)))
.andReturn(DEFAULT_JOB_KEY);
return jobTriggerCapture;
}
@Test(expected = ScheduleException.class)
public void testScheduleFails() throws Exception {
expectJobValidated(job);
storageUtil.jobStore.saveAcceptedJob(MANAGER_KEY, sanitizedConfiguration.getJobConfig());
expect(cronScheduler.schedule(eq(job.getCronSchedule()), EasyMock.<Runnable>anyObject()))
.andThrow(new CronException("injected"));
control.replay();
cron.receiveJob(sanitizedConfiguration);
}
@Test
public void testRunOverlapLoadedSuccessfully() throws Exception {
// Existing RUN_OVERLAP jobs should still load and map.
expect(cronScheduler.startAsync()).andReturn(cronScheduler);
cronScheduler.awaitRunning();
shutdownRegistry.addAction(EasyMock.<ExceptionalCommand<?>>anyObject());
IJobConfiguration jobA =
IJobConfiguration.build(
makeJob().newBuilder().setCronCollisionPolicy(CronCollisionPolicy.RUN_OVERLAP));
expect(storageUtil.jobStore.fetchJobs(MANAGER_KEY)).andReturn(ImmutableList.of(jobA));
expect(cronScheduler.isValidSchedule(jobA.getCronSchedule())).andReturn(true);
expect(cronScheduler.schedule(eq(jobA.getCronSchedule()), EasyMock.<Runnable>anyObject()))
.andReturn("keyA");
control.replay();
cron.schedulerActive(new SchedulerActive());
}
@Test(expected = IllegalStateException.class)
public void testJobStoredTwice() throws Exception {
// Simulate an inconsistent storage that contains two cron jobs under the same key.
expect(cronScheduler.startAsync()).andReturn(cronScheduler);
cronScheduler.awaitRunning();
shutdownRegistry.addAction(EasyMock.<ExceptionalCommand<?>>anyObject());
IJobConfiguration jobA =
IJobConfiguration.build(makeJob().newBuilder().setCronSchedule("1 2 3 4 5"));
IJobConfiguration jobB =
IJobConfiguration.build(makeJob().newBuilder().setCronSchedule("* * * * *"));
expect(storageUtil.jobStore.fetchJobs(MANAGER_KEY)).andReturn(ImmutableList.of(jobA, jobB));
expectJobValidated(jobA);
expect(cronScheduler.schedule(eq(jobA.getCronSchedule()), EasyMock.<Runnable>anyObject()))
.andReturn("keyA");
expectJobValidated(jobB);
control.replay();
cron.schedulerActive(new SchedulerActive());
}
