BATCHEE-117 added deciderTest

jbatch/src/test/java/org/apache/batchee/test/decider/DeciderTest.java

@@ -0,0 +1,84 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.batchee.test.decider;
+
+import org.apache.batchee.util.Batches;
+import org.testng.annotations.Test;
+
+import javax.batch.api.AbstractBatchlet;
+import javax.batch.api.BatchProperty;
+import javax.batch.api.Decider;
+import javax.batch.operations.JobOperator;
+import javax.batch.runtime.BatchRuntime;
+import javax.batch.runtime.BatchStatus;
+import javax.batch.runtime.StepExecution;
+import javax.inject.Inject;
+import java.util.List;
+import java.util.Properties;
+
+import static org.testng.Assert.assertEquals;
+
+public class DeciderTest {
+
+
+    @Test
+    public void testDeciderRestart() {
+
+        JobOperator jobOperator = BatchRuntime.getJobOperator();
+        long executionId = jobOperator.start("decider-test", new Properties());
+
+        BatchStatus batchStatus = Batches.waitFor(jobOperator, executionId);
+        assertEquals(batchStatus, BatchStatus.STOPPED);
+        assertEquals(jobOperator.getJobExecution(executionId).getExitStatus(), "decider-stop");
+
+        List<StepExecution> stepExecutions = jobOperator.getStepExecutions(executionId);
+        assertEquals(stepExecutions.size(), 1);
+
+        long restartExecutionId = jobOperator.restart(executionId, new Properties());
+
+        BatchStatus restartStatus = Batches.waitFor(jobOperator, restartExecutionId);
+        assertEquals(restartStatus, BatchStatus.COMPLETED);
+
+        String exitStatus = jobOperator.getJobExecution(restartExecutionId).getExitStatus();
+        assertEquals("COMPLETED", exitStatus);
+
+        List<StepExecution> restartExecutions = jobOperator.getStepExecutions(restartExecutionId);
+        assertEquals(restartExecutions.size(), 1);
+        assertEquals(restartExecutions.get(0).getStepName(), "executeOnRestart");
+    }
+
+
+    public static class TheDecider implements Decider {
+
+        @Override
+        public String decide(StepExecution[] executions) throws Exception {
+            return "foobar";
+        }
+    }
+
+    public static class TheBatchlet extends AbstractBatchlet {
+
+        @Inject
+        @BatchProperty
+        private String inStep;
+
+        @Override
+        public String process() throws Exception {
+            return inStep;
+        }
+    }
+}

jbatch/src/test/resources/META-INF/batch-jobs/decider-test.xml

@@ -0,0 +1,32 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!--
+  See the NOTICE file distributed with this work for additional information
+  regarding copyright ownership. Licensed under the Apache License,
+  Version 2.0 (the "License"); you may not use this file except in compliance
+  with the License. You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+  Unless required by applicable law or agreed to in writing, software
+  distributed under the License is distributed on an "AS IS" BASIS,
+  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+  See the License for the specific language governing permissions and
+  limitations under the License.
+-->
+<job id="myJob" xmlns="http://xmlns.jcp.org/xml/ns/javaee" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+    xsi:schemaLocation="http://xmlns.jcp.org/xml/ns/javaee http://xmlns.jcp.org/xml/ns/javaee/jobXML_1_0.xsd" version="1.0">
+    <step id="step1" next="decider1">
+        <batchlet ref="org.apache.batchee.test.decider.DeciderTest$TheBatchlet" />
+    </step>
+    <decision id="decider1" ref="org.apache.batchee.test.decider.DeciderTest$TheDecider">
+        <next on="continue" to="step2" /> <!-- so we can ensure the test fails if we modify the return value of the decider -->
+        <stop on="foobar" exit-status="decider-stop" restart="executeOnRestart"/>
+    </decision>
+    <step id="step2">
+        <batchlet ref="org.apache.batchee.test.decider.DeciderTest$TheBatchlet" />
+    </step>
+    <step id="executeOnRestart">
+        <batchlet ref="org.apache.batchee.test.decider.DeciderTest$TheBatchlet" />
+    </step>
+</job>
+