improve recovery test

NitorCreations · Mar 16, 2015 · e4269a4 · e4269a4
1 parent 31cd6fe
commit e4269a4
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/nflow-engine/src/test/java/com/nitorcreations/nflow/engine/internal/dao/ExecutorDaoTest.java b/nflow-engine/src/test/java/com/nitorcreations/nflow/engine/internal/dao/ExecutorDaoTest.java
@@ -63,6 +63,18 @@ public void recoverWorkflowInstancesFromDeadNodesSetsExecutorIdToNullAndInsertsA
     assertThat(workflowInstanceAction.executorId, is(dao.getExecutorId()));
     assertThat(workflowInstanceAction.type, is(recovery));
     assertThat(workflowInstanceAction.stateText, is("Recovered"));
+
+    dao.recoverWorkflowInstancesFromDeadNodes();
+
+    executorId = jdbc.queryForObject("select executor_id from nflow_workflow where id = ?", Integer.class, id);
+    assertThat(executorId, is(nullValue()));
+
+    actions = jdbc.query("select * from nflow_workflow_action where workflow_id = ?", new WorkflowInstanceActionRowMapper(
+        Collections.<Integer, Map<String, String>> emptyMap()), id);
+    assertThat(actions.size(), is(1));
+    assertThat(workflowInstanceAction.executorId, is(dao.getExecutorId()));
+    assertThat(workflowInstanceAction.type, is(recovery));
+    assertThat(workflowInstanceAction.stateText, is("Recovered"));
   }
 
   private void insertCrashedExecutor(int crashedExecutorId) {