From 6b937605358a1634de9bada32e36058033ac7c83 Mon Sep 17 00:00:00 2001
From: Janne Valkealahti <janne.valkealahti@gmail.com>
Date: Wed, 18 May 2016 18:59:05 +0100
Subject: [PATCH] Make tests more reliable

---
 .../java/demo/ordershipping/StateMachineTests.java    | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

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 0d5b25f4..1559dd70 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
@@ -40,8 +40,17 @@ public class StateMachineTests {
 	private StateMachine<String, String> stateMachine;
 
 	@Before
-	public void setup() {
+	public void setup() throws Exception {
 		stateMachine = stateMachineFactory.getStateMachine();
+		// plan don't know how to wait if machine is started
+		// automatically so wait here.
+		for (int i = 0; i < 10; i++) {
+			if (stateMachine.getState() != null) {
+				break;
+			} else {
+				Thread.sleep(200);
+			}
+		}
 	}
 
 	@Test