From 84e4f5ae25396fc3160f56590ad46a8c43019529 Mon Sep 17 00:00:00 2001
From: Edvard Fonsell <edvard.fonsell@nitorcreations.com>
Date: Tue, 2 Dec 2014 09:39:16 +0200
Subject: [PATCH] restore done state method for state workflow

---
 .../com/nitorcreations/nflow/tests/demo/StateWorkflow.java    | 4 ++++
 .../com/nitorcreations/nflow/tests/StateVariablesTest.java    | 3 ++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/nflow-tests/src/main/java/com/nitorcreations/nflow/tests/demo/StateWorkflow.java b/nflow-tests/src/main/java/com/nitorcreations/nflow/tests/demo/StateWorkflow.java
index b0248a0da..99925fc70 100644
--- a/nflow-tests/src/main/java/com/nitorcreations/nflow/tests/demo/StateWorkflow.java
+++ b/nflow-tests/src/main/java/com/nitorcreations/nflow/tests/demo/StateWorkflow.java
@@ -92,6 +92,10 @@ public NextAction state5(StateExecution execution, @StateVar(value = "variable2"
     return moveToState(done, "variable2 is set to " + variable2.value);
   }
 
+  public void done(StateExecution execution) {
+    System.out.println("StateWorkflow done.");
+  }
+
   public static class Variable {
     public String value;
 
diff --git a/nflow-tests/src/test/java/com/nitorcreations/nflow/tests/StateVariablesTest.java b/nflow-tests/src/test/java/com/nitorcreations/nflow/tests/StateVariablesTest.java
index 9ba0d0f53..14881b981 100644
--- a/nflow-tests/src/test/java/com/nitorcreations/nflow/tests/StateVariablesTest.java
+++ b/nflow-tests/src/test/java/com/nitorcreations/nflow/tests/StateVariablesTest.java
@@ -53,12 +53,13 @@ public void t02_checkStateVariables() throws InterruptedException {
     assertEquals(singletonMap("value", "foo1"), listResponse.stateVariables.get("variable1"));
     assertEquals(singletonMap("value", "bar3"), listResponse.stateVariables.get("variable2"));
 
-    assertEquals(5, listResponse.actions.size());
+    assertEquals(6, listResponse.actions.size());
     assertState(listResponse.actions, 0, StateWorkflow.State.state1, "foo1", null);
     assertState(listResponse.actions, 1, StateWorkflow.State.state2, null, "bar1");
     assertState(listResponse.actions, 2, StateWorkflow.State.state3, null, "bar2");
     assertState(listResponse.actions, 3, StateWorkflow.State.state4, null, null);
     assertState(listResponse.actions, 4, StateWorkflow.State.state5, null, "bar3");
+    assertState(listResponse.actions, 5, StateWorkflow.State.done, null, null);
   }
 
   private void assertState(List<Action> actions, int index, State state, String variable1, String variable2) {