Merge pull request #141 from radcortez/master

Added test for chunk-checkpoint project.

Author: arun-gupta

batch/chunk-checkpoint/pom.xml

@@ -1,5 +1,6 @@
 <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
          xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+
     <modelVersion>4.0.0</modelVersion>
     <parent>
         <groupId>org.javaee7.batch</groupId>
@@ -8,10 +9,13 @@
         <relativePath>../pom.xml</relativePath>
     </parent>
 
-    <groupId>org.javaee7.batch</groupId>
     <artifactId>chunk-checkpoint</artifactId>
-    <version>1.0-SNAPSHOT</version>
     <packaging>war</packaging>
-    <name>${project.artifactId}</name>
 
+    <dependencies>
+        <dependency>
+            <groupId>org.javaee7</groupId>
+            <artifactId>util-samples</artifactId>
+        </dependency>
+    </dependencies>
 </project>

batch/chunk-checkpoint/src/main/java/org/javaee7/batch/chunk/checkpoint/MyCheckpointAlgorithm.java

@@ -48,13 +48,15 @@
  */
 @Named
 public class MyCheckpointAlgorithm extends AbstractCheckpointAlgorithm {
+    private static boolean checkpointAlgorithmExecuted;
 
     @Override
     public boolean isReadyToCheckpoint() throws Exception {
-        if (MyItemReader.COUNT % 5 == 0)
-            return true;
-        else
-            return false;
+        checkpointAlgorithmExecuted = true;
+        return MyItemReader.COUNT % 5 == 0;
     }
 
+    public static boolean isCheckpointAlgorithmExecuted() {
+        return checkpointAlgorithmExecuted;
+    }
 }

batch/chunk-checkpoint/src/main/resources/META-INF/batch-jobs/myJob.xml

@@ -43,9 +43,9 @@
 <job id="myJob" xmlns="http://xmlns.jcp.org/xml/ns/javaee" version="1.0">
     <step id="myStep" >
         <chunk item-count="3" checkpoint-policy="custom">
-            <reader ref="myItemReader"></reader> 
-            <processor ref="myItemProcessor"></processor>
-            <writer ref="myItemWriter"></writer> 
+            <reader ref="myItemReader"/>
+            <processor ref="myItemProcessor"/>
+            <writer ref="myItemWriter"/>
             <checkpoint-algorithm ref="myCheckpointAlgorithm"/>
         </chunk>	
     </step>

batch/chunk-checkpoint/src/test/java/org/javaee7/batch/chunk/checkpoint/BatchChunkCheckpointTest.java

@@ -0,0 +1,65 @@
+package org.javaee7.batch.chunk.checkpoint;
+
+import org.javaee7.util.BatchTestHelper;
+import org.jboss.arquillian.container.test.api.Deployment;
+import org.jboss.arquillian.junit.Arquillian;
+import org.jboss.shrinkwrap.api.ArchivePaths;
+import org.jboss.shrinkwrap.api.ShrinkWrap;
+import org.jboss.shrinkwrap.api.asset.EmptyAsset;
+import org.jboss.shrinkwrap.api.spec.WebArchive;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+import javax.batch.operations.JobOperator;
+import javax.batch.runtime.BatchRuntime;
+import javax.batch.runtime.BatchStatus;
+import javax.batch.runtime.JobExecution;
+import javax.batch.runtime.Metric;
+import javax.batch.runtime.StepExecution;
+import java.util.List;
+import java.util.Map;
+import java.util.Properties;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author Roberto Cortez
+ */
+@RunWith(Arquillian.class)
+public class BatchChunkCheckpointTest {
+
+    @Deployment
+    public static WebArchive createDeployment() {
+        WebArchive war = ShrinkWrap.create(WebArchive.class)
+                                   .addClass(BatchTestHelper.class)
+                                   .addPackage("org.javaee7.batch.chunk.checkpoint")
+                                   .addAsWebInfResource(EmptyAsset.INSTANCE, ArchivePaths.create("beans.xml"))
+                                   .addAsResource("META-INF/batch-jobs/myJob.xml");
+        System.out.println(war.toString(true));
+        return war;
+    }
+
+    @Test
+    public void testBatchChunkCheckpoint() throws Exception {
+        JobOperator jobOperator = BatchRuntime.getJobOperator();
+        Long executionId = jobOperator.start("myJob", new Properties());
+        JobExecution jobExecution = jobOperator.getJobExecution(executionId);
+
+        BatchTestHelper.keepTestAlive(jobExecution);
+
+        List<StepExecution> stepExecutions = jobOperator.getStepExecutions(executionId);
+        for (StepExecution stepExecution : stepExecutions) {
+            if (stepExecution.getStepName().equals("myStep")) {
+                Map<Metric.MetricType, Long> metricsMap = BatchTestHelper.getMetricsMap(stepExecution.getMetrics());
+
+                assertEquals(10L, (long) metricsMap.get(Metric.MetricType.READ_COUNT));
+                assertEquals(10L / 2L, (long) metricsMap.get(Metric.MetricType.WRITE_COUNT));
+                assertEquals(10L / 5L + 1, (long) metricsMap.get(Metric.MetricType.COMMIT_COUNT));
+            }
+        }
+
+        assertTrue(MyCheckpointAlgorithm.isCheckpointAlgorithmExecuted());
+        assertEquals(jobExecution.getBatchStatus(), BatchStatus.COMPLETED);
+    }
+}