From 588eee751fafc754e7332818d867894b216fe1a5 Mon Sep 17 00:00:00 2001
From: Janne Valkealahti <janne.valkealahti@gmail.com>
Date: Thu, 19 May 2016 11:13:15 +0100
Subject: [PATCH] Make tests more reliable

---
 .../src/test/java/demo/ordershipping/StateMachineTests.java     | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/spring-statemachine-samples/ordershipping/src/test/java/demo/ordershipping/StateMachineTests.java b/spring-statemachine-samples/ordershipping/src/test/java/demo/ordershipping/StateMachineTests.java
index 1559dd70..959f373b 100644
--- 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();