From 3c334e6b56168d3118d384a9b1529ea7242434b4 Mon Sep 17 00:00:00 2001
From: Danil Kornishev <danil.kornishev@gmail.com>
Date: Fri, 31 Mar 2017 16:01:27 -0400
Subject: [PATCH] Spring State Machine x4 (#1547)

* [Fix] Move stateDo onto a separate state

* Change tests to spring runner

* Add generic params to StateMachine variables
---
 .../baeldung/spring/statemachine/ForkJoinStateMachineTest.java  | 2 +-
 .../spring/statemachine/HierarchicalStateMachineTest.java       | 2 +-
 .../baeldung/spring/statemachine/JunctionStateMachineTest.java  | 2 +-
 .../com/baeldung/spring/statemachine/StateEnumMachineTest.java  | 2 +-
 .../spring/statemachine/StateMachineIntegrationTest.java        | 2 +-
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/ForkJoinStateMachineTest.java b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/ForkJoinStateMachineTest.java
index 03cb101a9d..0de61da3bd 100644
--- a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/ForkJoinStateMachineTest.java
+++ b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/ForkJoinStateMachineTest.java
@@ -22,7 +22,7 @@ import static org.junit.Assert.assertTrue;
 public class ForkJoinStateMachineTest {
 
     @Resource
-    private StateMachine stateMachine;
+    private StateMachine<String, String> stateMachine;
 
     @Before
     public void setUp() {
diff --git a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/HierarchicalStateMachineTest.java b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/HierarchicalStateMachineTest.java
index 950414bb0e..76105368d8 100644
--- a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/HierarchicalStateMachineTest.java
+++ b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/HierarchicalStateMachineTest.java
@@ -21,7 +21,7 @@ import static org.junit.Assert.assertEquals;
 public class HierarchicalStateMachineTest {
 
     @Resource
-    private StateMachine stateMachine;
+    private StateMachine<String, String> stateMachine;
 
     @Before
     public void setUp() {
diff --git a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/JunctionStateMachineTest.java b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/JunctionStateMachineTest.java
index 64930162fd..b91e6af5c5 100644
--- a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/JunctionStateMachineTest.java
+++ b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/JunctionStateMachineTest.java
@@ -19,7 +19,7 @@ import javax.annotation.Resource;
 public class JunctionStateMachineTest {
 
     @Resource
-    private StateMachine stateMachine;
+    private StateMachine<String, String> stateMachine;
 
     @Before
     public void setUp() {
diff --git a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateEnumMachineTest.java b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateEnumMachineTest.java
index b7cbebe145..cd9e58eb8e 100644
--- a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateEnumMachineTest.java
+++ b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateEnumMachineTest.java
@@ -23,7 +23,7 @@ import static org.junit.Assert.assertTrue;
 public class StateEnumMachineTest {
 
     @Resource
-    private StateMachine stateMachine;
+    private StateMachine<ApplicationReviewStates, ApplicationReviewEvents> stateMachine;
 
     @Before
     public void setUp() {
diff --git a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateMachineIntegrationTest.java b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateMachineIntegrationTest.java
index 8f61d93105..1c1ab58917 100644
--- a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateMachineIntegrationTest.java
+++ b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateMachineIntegrationTest.java
@@ -22,7 +22,7 @@ import javax.annotation.Resource;
 public class StateMachineIntegrationTest {
 
     @Resource
-    private StateMachine stateMachine;
+    private StateMachine<String, String> stateMachine;
 
     @Before
     public void setUp() {