@Test
public void testUnexpectedRollback() throws Exception {
taskletStep.setTransactionManager(
new ResourcelessTransactionManager() {
@Override
protected void doCommit(DefaultTransactionStatus status) throws TransactionException {
super.doRollback(status);
throw new UnexpectedRollbackException("bar");
}
});
taskletStep.setTasklet(
new Tasklet() {
public RepeatStatus execute(StepContribution contribution, ChunkContext attributes)
throws Exception {
attributes
.getStepContext()
.getStepExecution()
.getExecutionContext()
.putString("foo", "bar");
return RepeatStatus.FINISHED;
}
});
taskletStep.execute(stepExecution);
assertEquals(FAILED, stepExecution.getStatus());
Throwable e = stepExecution.getFailureExceptions().get(0);
assertEquals("bar", e.getMessage());
assertEquals(0, stepExecution.getCommitCount());
assertEquals(1, stepExecution.getRollbackCount()); // Failed transaction counts as rollback
assertEquals(0, stepExecution.getExecutionContext().size());
}
@Test
public void testPropertyFileLoaderJob() {
try {
logger.debug("Begin Testing File Loader");
SimpleJobLauncher jobLauncher = new SimpleJobLauncher();
jobLauncher.setTaskExecutor(new SyncTaskExecutor());
jobLauncher.setJobRepository(jobRepository);
jobLauncherTestUtils.setJobLauncher(jobLauncher);
SimpleJob job = importProductsJob();
logJobDetails(job);
jobLauncherTestUtils.setJob(job);
JobExecution exec = jobLauncherTestUtils.launchJob(new JobParameters());
Collection<String> stepNames = job.getStepNames();
for (String stepName : stepNames) {
TaskletStep step = (TaskletStep) job.getStep(stepName);
step.setTransactionManager(transactionManager);
}
assertTrue("Steps MUST exist", stepNames != null);
assertTrue("At least ONE step MUST exist", stepNames.size() > 0);
Assert.assertTrue(
"Product Repository MUST have records",
jobRepository.isJobInstanceExists(job.getName(), new JobParameters()));
logger.debug("Job Execution Status {}", exec.getExitStatus());
logJobRepository(exec);
Assert.assertEquals(BatchStatus.COMPLETED, exec.getStatus());
} catch (Exception e) {
logger.error(e.getMessage(), e);
Assert.fail(e.getMessage());
}
}
@Before
public void init() {
taskletStep = new TaskletStep();
taskletStep.setTasklet(new ExceptionTasklet());
jobRepository = new UpdateCountingJobRepository();
taskletStep.setJobRepository(jobRepository);
taskletStep.setTransactionManager(new ResourcelessTransactionManager());
JobInstance jobInstance = new JobInstance(1L, new JobParameters(), "testJob");
JobExecution jobExecution = new JobExecution(jobInstance);
stepExecution = new StepExecution("testStep", jobExecution);
}
private SimpleJob importProductsJob() throws Exception {
// Create Job
SimpleJob bean = new SimpleJob();
bean.setName("importProductsJob");
bean.setJobRepository(jobRepository);
bean.afterPropertiesSet();
// Create Steps
List<Step> steps = new ArrayList<Step>();
// Create Tasklet Step
TaskletStep step = new TaskletStep();
step.setName("importProducts");
step.setTransactionManager(transactionManager);
step.setJobRepository(jobRepository);
step.setStartLimit(100);
// Create repeat template for Chunk Size
RepeatTemplate repeatTemplate = new TaskExecutorRepeatTemplate();
repeatTemplate.setCompletionPolicy(new SimpleCompletionPolicy(5));
// TaskletStep <- RepeatTemplate
step.setStepOperations(repeatTemplate);
// Create Chunk Tasklet with Provider (reader) and Chunk Processor
// Tasklet <- ChunkProvider <- ItemReader, RepeatTemplate
// Tasklet <- ChunkProcessor <- ItemProcessor, ItemWriter
ChunkOrientedTasklet<Product> tasklet =
new ChunkOrientedTasklet<Product>(
new SimpleChunkProvider<Product>(productReader(), repeatTemplate),
new SimpleChunkProcessor<Product, Product>(
new PassThroughItemProcessor<Product>(), new ProductItemLoggerWriter()));
// Job <- Steps <- TaskletStep <- Tasklet
step.setTasklet(tasklet);
// Job <- Steps <- TaskletStep
steps.add(step);
// Job <- Steps
bean.setSteps(steps);
return bean;
}
@Deprecated
public void scheduleLoanArrearsAndPortfolioAtRisk(
Date initialTime,
long delay,
JobRegistry jobRegistry,
final JobRepository jobRepository,
Map<String, Object> jobData,
ResourcelessTransactionManager transactionManager)
throws TaskSystemException {
final String jobName = "LoanArrearsAndPortfolioAtRiskTask";
try {
final TaskletStep step1 = new TaskletStep();
step1.setName("LoanArrearsAndPortfolioAtRiskTask-step-1");
step1.setTasklet(
(Tasklet)
Class.forName(BATCH_JOB_CLASS_PATH_PREFIX + getHelperName("LoanArrearsTask"))
.newInstance());
step1.setJobRepository(jobRepository);
step1.setTransactionManager(transactionManager);
step1.afterPropertiesSet();
final TaskletStep step2 = new TaskletStep();
step2.setName("LoanArrearsAndPortfolioAtRiskTask-step-2");
step2.setTasklet(
(Tasklet)
Class.forName(BATCH_JOB_CLASS_PATH_PREFIX + getHelperName("PortfolioAtRiskTask"))
.newInstance());
step2.setJobRepository(jobRepository);
step2.setTransactionManager(transactionManager);
step2.afterPropertiesSet();
jobRegistry.register(
new JobFactory() {
@Override
public Job createJob() {
SimpleJob job = new SimpleJob(jobName + "Job");
job.setJobRepository(jobRepository);
job.setRestartable(true);
job.registerJobExecutionListener(new BatchJobListener());
job.addStep(step1);
job.addStep(step2);
return job;
}
@Override
public String getJobName() {
return jobName + "Job";
}
});
} catch (Exception e) {
throw new TaskSystemException(e);
}
JobDetailBean jobDetailBean = new JobDetailBean();
jobDetailBean.setJobDataAsMap(jobData);
try {
jobDetailBean.setJobClass(Class.forName(BATCH_JOB_CLASS_PATH_PREFIX + "PortfolioAtRiskTask"));
} catch (ClassNotFoundException cnfe) {
throw new TaskSystemException(cnfe);
}
jobDetailBean.setName(jobName + "Job");
jobDetailBean.setGroup(Scheduler.DEFAULT_GROUP);
jobDetailBean.afterPropertiesSet();
SimpleTrigger trigger = new SimpleTrigger();
trigger.setName(jobName + "Job");
trigger.setGroup(Scheduler.DEFAULT_GROUP);
trigger.setStartTime(initialTime);
trigger.setRepeatInterval(delay);
trigger.setRepeatCount(SimpleTrigger.REPEAT_INDEFINITELY);
try {
scheduler.scheduleJob(jobDetailBean, trigger);
} catch (SchedulerException se) {
throw new TaskSystemException(se);
}
}
