add tests for StateExecutionImpl.getVariable methods

NitorCreations · Apr 23, 2016 · ab99cdd · ab99cdd
1 parent d561075
commit ab99cdd
Showing 1 changed file with 59 additions and 0 deletions.
diff --git a/...c/test/java/com/nitorcreations/nflow/engine/internal/workflow/StateExecutionImplTest.java b/...c/test/java/com/nitorcreations/nflow/engine/internal/workflow/StateExecutionImplTest.java
@@ -139,6 +139,65 @@ public void workflowInstanceBuilder() {
     verify(objectStringMapper).convertFromObject("foo", data);
   }
 
+  @Test
+  public void getStringVariableWorks() {
+    execution.setVariable("foo", "bar");
+
+    assertThat(execution.getVariable("foo"), is("bar"));
+  }
+
+  @Test
+  public void getMissingStringVariableReturnsNull() {
+    assertThat(execution.getVariable("foo"), is(nullValue()));
+  }
+
+  @Test
+  public void getStringVariableWithDefaultWorks() {
+    execution.setVariable("foo", "bar");
+
+    assertThat(execution.getVariable("foo", "default"), is("bar"));
+  }
+
+  @Test
+  public void getMissingStringVariableWithDefaultReturnDefaultValue() {
+    assertThat(execution.getVariable("foo", "default"), is("default"));
+  }
+
+  @Test
+  public void getVariableWorks() {
+    Data data = new Data(47, "bar");
+    String serializedData = "data in serialized form";
+    when(objectStringMapper.convertFromObject("foo", data)).thenReturn(serializedData);
+    when(objectStringMapper.convertToObject(Data.class, "foo", serializedData)).thenReturn(data);
+    execution.setVariable("foo", data);
+
+    assertThat(execution.getVariable("foo", Data.class), is(data));
+  }
+
+  @Test
+  public void getMissingVariableReturnsNull() {
+    assertThat(execution.getVariable("foo", Data.class), is(nullValue()));
+  }
+
+  @Test
+  public void getVariableWithDefaultWorks() {
+    Data data = new Data(47, "bar");
+    Data defaultData = new Data(42, "foobar");
+    String serializedData = "data in serialized form";
+    when(objectStringMapper.convertFromObject("foo", data)).thenReturn(serializedData);
+    when(objectStringMapper.convertToObject(Data.class, "foo", serializedData)).thenReturn(data);
+    execution.setVariable("foo", data);
+
+    assertThat(execution.getVariable("foo", Data.class, defaultData), is(data));
+  }
+
+  @Test
+  public void getMissingVariableWithDefaultReturnsDefaultValue() {
+    Data defaultData = new Data(42, "foobar");
+
+    assertThat(execution.getVariable("foo", Data.class, defaultData), is(defaultData));
+  }
+
   static class Data {
     public final int number;
     public final String text;