Make tests more reliable

2016-05-19 11:13:15 +01:00
parent 6b93760535
commit 588eee751f
1 changed files with 2 additions and 0 deletions
--- a/spring-statemachine-samples/ordershipping/src/test/java/demo/ordershipping/StateMachineTests.java
+++ b/spring-statemachine-samples/ordershipping/src/test/java/demo/ordershipping/StateMachineTests.java
@@ -77,6 +77,7 @@ public class StateMachineTests {
 						.sendEvent("PLACE_ORDER")
 						.expectStates("CUSTOMER_ERROR")
 						.expectStateChanged(2)
+						.expectStateMachineStopped(1)
 						.and()
 					.build();
 		plan.test();
@@ -102,6 +103,7 @@ public class StateMachineTests {
 								.setHeader("payment", "1000").build())
 						.expectStates("ORDER_SHIPPED")
 						.expectStateChanged(4)
+						.expectStateMachineStopped(3)
 						.and()
 					.build();
 		plan.test();