#BAEL-8695: add unit test

Author: hmdrzsharifi

spring-batch-2/src/main/java/com/baeldung/restartjob/BatchConfig.java

@@ -48,16 +48,8 @@ public FlatFileItemReader<String> flatFileItemReader() {
     }
 
     @Bean
-    public ItemProcessor<String, String> itemProcessor() {
-        return item -> {
-          System.out.println("Processing: " + item);
-
-          if (item.equals("Item3")) {
-              throw new RuntimeException("Simulated failure on Item3");
-          }
-
-          return "PROCESSED " + item;
-        };
+    public RestartItemProcessor itemProcessor() {
+        return new RestartItemProcessor();
     }
 
     @Bean
@@ -69,4 +61,21 @@ public ItemWriter<String> itemWriter() {
           }
         };
     }
+
+    static class RestartItemProcessor implements ItemProcessor<String, String> {
+        private boolean failOnItem3 = true;
+
+        public void setFailOnItem3(boolean failOnItem3) {
+            this.failOnItem3 = failOnItem3;
+        }
+
+        @Override
+        public String process(String item) throws Exception {
+            System.out.println("Processing: " + item + " (failOnItem3=" + failOnItem3 + ")");
+            if (failOnItem3 && item.equals("Item3")) {
+                throw new RuntimeException("Simulated failure on Item3");
+            }
+            return "PROCESSED " + item;
+        }
+    }
 }

spring-batch-2/src/main/java/com/baeldung/restartjob/RestartJobBatchApp.java

@@ -14,6 +14,7 @@
 import org.springframework.boot.CommandLineRunner;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
 import org.springframework.context.annotation.Bean;
 
 @SpringBootApplication
@@ -26,7 +27,9 @@ public static void main(String[] args) {
     }
 
     @Bean
-    CommandLineRunner run(JobLauncher jobLauncher, Job job, JobExplorer jobExplorer, JobOperator jobOperator) {
+    @ConditionalOnProperty(prefix = "job.autorun", name = "enabled", havingValue = "true", matchIfMissing = true)
+    CommandLineRunner run(JobLauncher jobLauncher, Job job, JobExplorer jobExplorer,
+        JobOperator jobOperator, BatchConfig.RestartItemProcessor itemProcessor) {
         return args -> {
           JobParameters jobParameters = new JobParametersBuilder()
               .addString("jobId", "test-job-" + System.currentTimeMillis())
@@ -40,6 +43,7 @@ CommandLineRunner run(JobLauncher jobLauncher, Job job, JobExplorer jobExplorer,
                   JobExecution lastExecution = executions.get(0);
                   if (lastExecution.getStatus() == BatchStatus.FAILED) {
                       System.out.println("Restarting failed job execution with ID: " + lastExecution.getId());
+                      itemProcessor.setFailOnItem3(false);
 
                       JobExecution restartedExecution = jobLauncher.run(job, jobParameters);
 

spring-batch-2/src/test/java/com/baeldung/restartjob/RestartJobBatchAppIntegrationTest.java

@@ -0,0 +1,92 @@
+package com.baeldung.restartjob;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.StandardCopyOption;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.batch.core.BatchStatus;
+import org.springframework.batch.core.Job;
+import org.springframework.batch.core.JobExecution;
+import org.springframework.batch.core.JobParameters;
+import org.springframework.batch.core.JobParametersBuilder;
+import org.springframework.batch.core.launch.JobLauncher;
+import org.springframework.batch.test.JobLauncherTestUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.boot.test.context.TestConfiguration;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Import;
+import org.springframework.core.io.ClassPathResource;
+import org.springframework.core.io.Resource;
+
+@SpringBootTest(classes = {RestartJobBatchApp.class, BatchConfig.class},
+    properties = {"job.autorun.enabled=false"})
+@Import(RestartJobBatchAppIntegrationTest.TestConfig.class)
+public class RestartJobBatchAppIntegrationTest {
+
+    @Autowired
+    private JobLauncherTestUtils jobLauncherTestUtils;
+
+    @Autowired
+    private BatchConfig.RestartItemProcessor itemProcessor;
+
+    @TestConfiguration
+    static class TestConfig {
+        @Autowired
+        private JobLauncher jobLauncher;
+
+        @Autowired
+        private Job job;
+
+        @Bean
+        public JobLauncherTestUtils jobLauncherTestUtils() {
+            JobLauncherTestUtils jobLauncherTestUtils = new JobLauncherTestUtils();
+            jobLauncherTestUtils.setJobLauncher(jobLauncher);
+            jobLauncherTestUtils.setJob(job);
+            return jobLauncherTestUtils;
+        }
+    }
+
+    private final Resource inputFile = new ClassPathResource("data.csv");
+
+    @Test
+    public void givenItems_whenFailed_thenRestartFromFailure() throws Exception {
+        // Given
+        createTestFile("Item1\nItem2\nItem3\nItem4");
+
+        JobParameters jobParameters = new JobParametersBuilder()
+          .addLong("time", System.currentTimeMillis())
+          .toJobParameters();
+
+        // When
+        JobExecution firstExecution = jobLauncherTestUtils.launchJob(jobParameters);
+        assertEquals(BatchStatus.FAILED, firstExecution.getStatus());
+
+        Long executionId = firstExecution.getId();
+
+        itemProcessor.setFailOnItem3(false);
+
+        // Then
+        JobExecution restartedExecution = jobLauncherTestUtils.launchJob(jobParameters);
+
+        assertEquals(BatchStatus.COMPLETED, restartedExecution.getStatus());
+
+        assertEquals(
+          firstExecution.getJobInstance().getInstanceId(),
+          restartedExecution.getJobInstance().getInstanceId()
+        );
+    }
+    
+    private void createTestFile(String content) throws IOException {
+        Path tempFile = Files.createTempFile("test-data", ".csv");
+        Files.write(tempFile, content.getBytes());
+        Files.copy(tempFile, inputFile.getFile().toPath(), StandardCopyOption.REPLACE_EXISTING);
+    }
+
+}
+
+