Count the number of rows processed, and better names

Author: jeffgbutler

src/test/java/examples/springbatch/BatchConfiguration.java

@@ -128,8 +128,8 @@ public Step step1(ItemReader<Person> reader, ItemProcessor<Person, Person> proce
     }
 
     @Bean
-    public Job importUserJob(Step step1) {
-        return jobBuilderFactory.get("importUserJob")
+    public Job upperCaseLastName(Step step1) {
+        return jobBuilderFactory.get("upperCaseLastName")
                 .incrementer(new RunIdIncrementer())
                 .flow(step1)
                 .end()

src/test/java/examples/springbatch/PersonProcessor.java

@@ -15,16 +15,33 @@
  */
 package examples.springbatch;
 
+import org.springframework.batch.core.StepExecution;
+import org.springframework.batch.core.annotation.BeforeStep;
+import org.springframework.batch.item.ExecutionContext;
 import org.springframework.batch.item.ItemProcessor;
 
 public class PersonProcessor implements ItemProcessor<Person, Person> {
+    
+    private ExecutionContext executionContext;
 
     @Override
     public Person process(Person person) throws Exception {
+        incrementRowCount();
+        
         Person transformed = new Person();
         transformed.setId(person.getId());
         transformed.setFirstName(person.getFirstName().toUpperCase());
         transformed.setLastName(person.getLastName().toUpperCase());
         return transformed;
     }
+
+    @BeforeStep
+    public void beforeStep(StepExecution stepExecution) {
+        executionContext = stepExecution.getExecutionContext();
+    }
+
+    private void incrementRowCount() {
+        executionContext.putInt("row_count",
+                executionContext.getInt("row_count", 0) + 1);
+    }
 }

src/test/java/examples/springbatch/SpringBatchTest.java

@@ -26,6 +26,9 @@
 import org.mybatis.dynamic.sql.select.SelectDSL;
 import org.mybatis.dynamic.sql.select.render.SelectStatementProvider;
 import org.mybatis.spring.SqlSessionFactoryBean;
+import org.springframework.batch.core.ExitStatus;
+import org.springframework.batch.core.JobExecution;
+import org.springframework.batch.core.StepExecution;
 import org.springframework.batch.test.JobLauncherTestUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.test.context.junit.jupiter.SpringJUnitConfig;
@@ -40,15 +43,26 @@ public class SpringBatchTest {
     private SqlSessionFactoryBean sqlSessionFactory;
 
     @Test
-    public void testJob() throws Exception {
-        assertThat(getCount()).isEqualTo(0);
+    public void testThatRowsAreTransformedToUpperCase() throws Exception {
+        // starting condition
+        assertThat(upperCaseRowCount()).isEqualTo(0);
 
-        jobLauncherTestUtils.launchJob();
+        JobExecution execution = jobLauncherTestUtils.launchJob();
+        assertThat(execution.getExitStatus()).isEqualTo(ExitStatus.COMPLETED);
+        assertThat(numberOfRowsProcessed(execution)).isEqualTo(2);
 
-        assertThat(getCount()).isEqualTo(2);
+        // ending condition
+        assertThat(upperCaseRowCount()).isEqualTo(2);
     }
 
-    private long getCount() throws Exception {
+    private int numberOfRowsProcessed(JobExecution jobExecution) {
+        return jobExecution.getStepExecutions().stream()
+                .map(StepExecution::getExecutionContext)
+                .mapToInt(ec -> ec.getInt("row_count"))
+                .sum();
+    }
+    
+    private long upperCaseRowCount() throws Exception {
         try (SqlSession sqlSession = sqlSessionFactory.getObject().openSession()) {
             PersonMapper personMapper = sqlSession.getMapper(PersonMapper.class);