CLOSED - issue BATCH-1024: FlowJob's start state should be the first state listed in the config
This commit is contained in:
@@ -20,8 +20,8 @@ import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.fail;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collection;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
@@ -72,7 +72,7 @@ public class FlowJobTests {
|
||||
@Test
|
||||
public void testTwoSteps() throws Exception {
|
||||
SimpleFlow flow = new SimpleFlow("job");
|
||||
Collection<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
List<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "step2"));
|
||||
transitions.add(StateTransition.createEndStateTransition(new StepState(new StubStep("step2"))));
|
||||
flow.setStateTransitions(transitions);
|
||||
@@ -86,7 +86,7 @@ public class FlowJobTests {
|
||||
@Test
|
||||
public void testFailedStep() throws Exception {
|
||||
SimpleFlow flow = new SimpleFlow("job");
|
||||
Collection<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
List<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
transitions.add(StateTransition.createStateTransition(new StepState(new StepSupport("step1") {
|
||||
@Override
|
||||
public void execute(StepExecution stepExecution) throws JobInterruptedException,
|
||||
@@ -108,7 +108,7 @@ public class FlowJobTests {
|
||||
@Test
|
||||
public void testFailedStepRestarted() throws Exception {
|
||||
SimpleFlow flow = new SimpleFlow("job");
|
||||
Collection<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
List<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
transitions.add(StateTransition.createStateTransition(new StepState(new StepSupport("step1") {
|
||||
@Override
|
||||
public void execute(StepExecution stepExecution) throws JobInterruptedException,
|
||||
@@ -150,7 +150,7 @@ public class FlowJobTests {
|
||||
@Test
|
||||
public void testStoppingStep() throws Exception {
|
||||
SimpleFlow flow = new SimpleFlow("job");
|
||||
Collection<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
List<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
transitions.add(StateTransition.createStateTransition(new StepState(new StepSupport("step1") {
|
||||
@Override
|
||||
public void execute(StepExecution stepExecution) throws JobInterruptedException,
|
||||
@@ -175,7 +175,7 @@ public class FlowJobTests {
|
||||
@Test
|
||||
public void testEndStateStopped() throws Exception {
|
||||
SimpleFlow flow = new SimpleFlow("job");
|
||||
Collection<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
List<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "end"));
|
||||
transitions.add(StateTransition.createStateTransition(new EndState(BatchStatus.STOPPED, "end"), "step2"));
|
||||
transitions.add(StateTransition.createEndStateTransition(new StepState(new StubStep("step2"))));
|
||||
@@ -194,7 +194,7 @@ public class FlowJobTests {
|
||||
|
||||
public void testEndStateFailed() throws Exception {
|
||||
SimpleFlow flow = new SimpleFlow("job");
|
||||
Collection<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
List<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "end"));
|
||||
transitions.add(StateTransition.createStateTransition(new EndState(BatchStatus.FAILED, "end"), "step2"));
|
||||
transitions.add(StateTransition.createEndStateTransition(new StepState(new StubStep("step2"))));
|
||||
@@ -209,7 +209,7 @@ public class FlowJobTests {
|
||||
@Test
|
||||
public void testEndStateStoppedWithRestart() throws Exception {
|
||||
SimpleFlow flow = new SimpleFlow("job");
|
||||
Collection<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
List<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "end"));
|
||||
transitions.add(StateTransition.createStateTransition(new EndState(BatchStatus.STOPPED, "end"), "step2"));
|
||||
transitions.add(StateTransition.createEndStateTransition(new StepState(new StubStep("step2"))));
|
||||
@@ -232,7 +232,7 @@ public class FlowJobTests {
|
||||
@Test
|
||||
public void testBranching() throws Exception {
|
||||
SimpleFlow flow = new SimpleFlow("job");
|
||||
Collection<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
List<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "step2"));
|
||||
transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "COMPLETED",
|
||||
"step3"));
|
||||
@@ -251,8 +251,8 @@ public class FlowJobTests {
|
||||
public void testBasicFlow() throws Throwable {
|
||||
SimpleFlow flow = new SimpleFlow("job");
|
||||
Step step = new StubStep("step");
|
||||
flow.setStateTransitions(Collections.singleton(StateTransition.createEndStateTransition(new StepState(step),
|
||||
"*")));
|
||||
flow.setStateTransitions(Collections.singletonList(StateTransition.createEndStateTransition(
|
||||
new StepState(step), "*")));
|
||||
job.setFlow(flow);
|
||||
job.execute(jobExecution);
|
||||
if (!jobExecution.getAllFailureExceptions().isEmpty()) {
|
||||
@@ -272,7 +272,7 @@ public class FlowJobTests {
|
||||
}
|
||||
};
|
||||
|
||||
Collection<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
List<StateTransition> transitions = new ArrayList<StateTransition>();
|
||||
transitions.add(StateTransition.createStateTransition(new StepState(new StubStep("step1")), "*", "decision"));
|
||||
transitions.add(StateTransition.createStateTransition(new DecisionState(decider, "decision"), "*", "step2"));
|
||||
transitions.add(StateTransition
|
||||
|
||||
@@ -1,243 +1,241 @@
|
||||
/*
|
||||
* Copyright 2006-2007 the original author or authors.
|
||||
*
|
||||
* 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.
|
||||
*/
|
||||
package org.springframework.batch.core.job.flow.support;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
import static org.junit.Assert.fail;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collection;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
|
||||
import org.junit.Test;
|
||||
import org.springframework.batch.core.job.flow.FlowExecution;
|
||||
import org.springframework.batch.core.job.flow.FlowExecutionException;
|
||||
import org.springframework.batch.core.job.flow.FlowExecutor;
|
||||
import org.springframework.batch.core.job.flow.support.SimpleFlow;
|
||||
import org.springframework.batch.core.job.flow.support.StateTransition;
|
||||
|
||||
/**
|
||||
* @author Dave Syer
|
||||
*
|
||||
*/
|
||||
public class SimpleFlowTests {
|
||||
|
||||
private SimpleFlow flow = new SimpleFlow("job");
|
||||
|
||||
private FlowExecutor executor = new JobFlowExecutorSupport();
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void testEmptySteps() throws Exception {
|
||||
flow.setStateTransitions(Collections.<StateTransition> emptySet());
|
||||
flow.afterPropertiesSet();
|
||||
}
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void testNoNextStepSpecified() throws Exception {
|
||||
flow.setStateTransitions(Collections.singleton(StateTransition.createStateTransition(new StateSupport(
|
||||
"step"), "foo")));
|
||||
flow.afterPropertiesSet();
|
||||
}
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void testNoStartStep() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StateSupport("step"),
|
||||
FlowExecution.FAILED, "step"), StateTransition
|
||||
.createEndStateTransition(new StateSupport("step"))));
|
||||
flow.afterPropertiesSet();
|
||||
}
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void testNoEndStep() throws Exception {
|
||||
flow.setStateTransitions(Collections.singleton(StateTransition.createStateTransition(new StateSupport(
|
||||
"step"), FlowExecution.FAILED, "step")));
|
||||
flow.setStartStateName("step");
|
||||
flow.afterPropertiesSet();
|
||||
}
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void testMultipleStartSteps() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createEndStateTransition(new StubState("step1")),
|
||||
StateTransition.createEndStateTransition(new StubState("step2"))));
|
||||
flow.afterPropertiesSet();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testNoMatchForNextStep() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1"), "FOO", "step2"),
|
||||
StateTransition.createEndStateTransition(new StubState("step2"))));
|
||||
flow.afterPropertiesSet();
|
||||
try {
|
||||
flow.start(executor);
|
||||
fail("Expected JobExecutionException");
|
||||
}
|
||||
catch (FlowExecutionException e) {
|
||||
// expected
|
||||
String message = e.getMessage();
|
||||
assertTrue("Wrong message: " + message, message.toLowerCase().contains("next state not found"));
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testOneStep() throws Exception {
|
||||
flow.setStateTransitions(Collections
|
||||
.singleton(StateTransition.createEndStateTransition(new StubState("step1"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step1", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testOneStepWithListenerCallsClose() throws Exception {
|
||||
flow.setStateTransitions(Collections
|
||||
.singleton(StateTransition.createEndStateTransition(new StubState("step1"))));
|
||||
flow.afterPropertiesSet();
|
||||
final List<FlowExecution> list = new ArrayList<FlowExecution>();
|
||||
executor = new JobFlowExecutorSupport() {
|
||||
@Override
|
||||
public void close(FlowExecution result) {
|
||||
list.add(result);
|
||||
}
|
||||
};
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(1, list.size());
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step1", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testExplicitStartStep() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step"),
|
||||
FlowExecution.FAILED, "step"), StateTransition.createEndStateTransition(new StubState("step"))));
|
||||
flow.setStartStateName("step");
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testTwoSteps() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1"), "step2"),
|
||||
StateTransition.createEndStateTransition(new StubState("step2"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step2", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testResume() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1"), "step2"),
|
||||
StateTransition.createEndStateTransition(new StubState("step2"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.resume("step2", executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step2", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testFailedStep() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1") {
|
||||
@Override
|
||||
public String handle(FlowExecutor executor) {
|
||||
return FlowExecution.FAILED;
|
||||
}
|
||||
}, "step2"), StateTransition.createEndStateTransition(new StubState("step2"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step2", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testBranching() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1"), "step2"),
|
||||
StateTransition.createStateTransition(new StubState("step1"), FlowExecution.COMPLETED, "step3"),
|
||||
StateTransition.createEndStateTransition(new StubState("step2")), StateTransition
|
||||
.createEndStateTransition(new StubState("step3"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step3", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testPause() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1"), "step2"),
|
||||
StateTransition.createStateTransition(new StubState("step2") {
|
||||
private boolean paused = false;
|
||||
|
||||
@Override
|
||||
public String handle(FlowExecutor executor) throws Exception {
|
||||
if (!paused) {
|
||||
paused = true;
|
||||
return FlowExecution.PAUSED;
|
||||
}
|
||||
paused = false;
|
||||
return FlowExecution.COMPLETED;
|
||||
}
|
||||
|
||||
}, "step3"), StateTransition.createEndStateTransition(new StubState("step3"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.PAUSED, execution.getStatus());
|
||||
assertEquals("step2", execution.getName());
|
||||
execution = flow.resume(execution.getName(), executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step3", execution.getName());
|
||||
}
|
||||
|
||||
private Collection<StateTransition> collect(StateTransition s1, StateTransition s2) {
|
||||
Collection<StateTransition> list = new ArrayList<StateTransition>();
|
||||
list.add(s1);
|
||||
list.add(s2);
|
||||
return list;
|
||||
}
|
||||
|
||||
private Collection<StateTransition> collect(StateTransition s1, StateTransition s2,
|
||||
StateTransition s3) {
|
||||
Collection<StateTransition> list = collect(s1, s2);
|
||||
list.add(s3);
|
||||
return list;
|
||||
}
|
||||
|
||||
private Collection<StateTransition> collect(StateTransition s1, StateTransition s2,
|
||||
StateTransition s3, StateTransition s4) {
|
||||
Collection<StateTransition> list = collect(s1, s2, s3);
|
||||
list.add(s4);
|
||||
return list;
|
||||
}
|
||||
|
||||
/**
|
||||
* @author Dave Syer
|
||||
*
|
||||
*/
|
||||
private static class StubState extends StateSupport {
|
||||
|
||||
/**
|
||||
* @param string
|
||||
*/
|
||||
public StubState(String string) {
|
||||
super(string);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
/*
|
||||
* Copyright 2006-2007 the original author or authors.
|
||||
*
|
||||
* 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.
|
||||
*/
|
||||
package org.springframework.batch.core.job.flow.support;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
import static org.junit.Assert.fail;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
|
||||
import org.junit.Test;
|
||||
import org.springframework.batch.core.job.flow.FlowExecution;
|
||||
import org.springframework.batch.core.job.flow.FlowExecutionException;
|
||||
import org.springframework.batch.core.job.flow.FlowExecutor;
|
||||
|
||||
/**
|
||||
* @author Dave Syer
|
||||
*
|
||||
*/
|
||||
public class SimpleFlowTests {
|
||||
|
||||
private SimpleFlow flow = new SimpleFlow("job");
|
||||
|
||||
private FlowExecutor executor = new JobFlowExecutorSupport();
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void testEmptySteps() throws Exception {
|
||||
flow.setStateTransitions(Collections.<StateTransition> emptyList());
|
||||
flow.afterPropertiesSet();
|
||||
}
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void testNoNextStepSpecified() throws Exception {
|
||||
flow.setStateTransitions(Collections.singletonList(StateTransition.createStateTransition(new StateSupport(
|
||||
"step"), "foo")));
|
||||
flow.afterPropertiesSet();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testStepLoop() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StateSupport("step"),
|
||||
FlowExecution.FAILED, "step"), StateTransition.createEndStateTransition(new StateSupport("step"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step", execution.getName());
|
||||
}
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void testNoEndStep() throws Exception {
|
||||
flow.setStateTransitions(Collections.singletonList(StateTransition.createStateTransition(new StateSupport(
|
||||
"step"), FlowExecution.FAILED, "step")));
|
||||
flow.afterPropertiesSet();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUnconnectedSteps() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createEndStateTransition(new StubState("step1")),
|
||||
StateTransition.createEndStateTransition(new StubState("step2"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step1", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testNoMatchForNextStep() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1"), "FOO", "step2"),
|
||||
StateTransition.createEndStateTransition(new StubState("step2"))));
|
||||
flow.afterPropertiesSet();
|
||||
try {
|
||||
flow.start(executor);
|
||||
fail("Expected JobExecutionException");
|
||||
}
|
||||
catch (FlowExecutionException e) {
|
||||
// expected
|
||||
String message = e.getMessage();
|
||||
assertTrue("Wrong message: " + message, message.toLowerCase().contains("next state not found"));
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testOneStep() throws Exception {
|
||||
flow.setStateTransitions(Collections.singletonList(StateTransition.createEndStateTransition(new StubState(
|
||||
"step1"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step1", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testOneStepWithListenerCallsClose() throws Exception {
|
||||
flow.setStateTransitions(Collections.singletonList(StateTransition.createEndStateTransition(new StubState(
|
||||
"step1"))));
|
||||
flow.afterPropertiesSet();
|
||||
final List<FlowExecution> list = new ArrayList<FlowExecution>();
|
||||
executor = new JobFlowExecutorSupport() {
|
||||
@Override
|
||||
public void close(FlowExecution result) {
|
||||
list.add(result);
|
||||
}
|
||||
};
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(1, list.size());
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step1", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testExplicitStartStep() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step"),
|
||||
FlowExecution.FAILED, "step"), StateTransition.createEndStateTransition(new StubState("step"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testTwoSteps() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1"), "step2"),
|
||||
StateTransition.createEndStateTransition(new StubState("step2"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step2", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testResume() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1"), "step2"),
|
||||
StateTransition.createEndStateTransition(new StubState("step2"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.resume("step2", executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step2", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testFailedStep() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1") {
|
||||
@Override
|
||||
public String handle(FlowExecutor executor) {
|
||||
return FlowExecution.FAILED;
|
||||
}
|
||||
}, "step2"), StateTransition.createEndStateTransition(new StubState("step2"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step2", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testBranching() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1"), "step2"),
|
||||
StateTransition.createStateTransition(new StubState("step1"), FlowExecution.COMPLETED, "step3"),
|
||||
StateTransition.createEndStateTransition(new StubState("step2")), StateTransition
|
||||
.createEndStateTransition(new StubState("step3"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step3", execution.getName());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testPause() throws Exception {
|
||||
flow.setStateTransitions(collect(StateTransition.createStateTransition(new StubState("step1"), "step2"),
|
||||
StateTransition.createStateTransition(new StubState("step2") {
|
||||
private boolean paused = false;
|
||||
|
||||
@Override
|
||||
public String handle(FlowExecutor executor) throws Exception {
|
||||
if (!paused) {
|
||||
paused = true;
|
||||
return FlowExecution.PAUSED;
|
||||
}
|
||||
paused = false;
|
||||
return FlowExecution.COMPLETED;
|
||||
}
|
||||
|
||||
}, "step3"), StateTransition.createEndStateTransition(new StubState("step3"))));
|
||||
flow.afterPropertiesSet();
|
||||
FlowExecution execution = flow.start(executor);
|
||||
assertEquals(FlowExecution.PAUSED, execution.getStatus());
|
||||
assertEquals("step2", execution.getName());
|
||||
execution = flow.resume(execution.getName(), executor);
|
||||
assertEquals(FlowExecution.COMPLETED, execution.getStatus());
|
||||
assertEquals("step3", execution.getName());
|
||||
}
|
||||
|
||||
private List<StateTransition> collect(StateTransition s1, StateTransition s2) {
|
||||
List<StateTransition> list = new ArrayList<StateTransition>();
|
||||
list.add(s1);
|
||||
list.add(s2);
|
||||
return list;
|
||||
}
|
||||
|
||||
private List<StateTransition> collect(StateTransition s1, StateTransition s2, StateTransition s3) {
|
||||
List<StateTransition> list = collect(s1, s2);
|
||||
list.add(s3);
|
||||
return list;
|
||||
}
|
||||
|
||||
private List<StateTransition> collect(StateTransition s1, StateTransition s2, StateTransition s3, StateTransition s4) {
|
||||
List<StateTransition> list = collect(s1, s2, s3);
|
||||
list.add(s4);
|
||||
return list;
|
||||
}
|
||||
|
||||
/**
|
||||
* @author Dave Syer
|
||||
*
|
||||
*/
|
||||
private static class StubState extends StateSupport {
|
||||
|
||||
/**
|
||||
* @param string
|
||||
*/
|
||||
public StubState(String string) {
|
||||
super(string);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user