Improve coverage flow/controller/node #4

Issue-ID: APPC-440
Change-Id: Iee74ce27c3978d265fccc79dacaf8df965768a36
Signed-off-by: kurczews <krzysztof.kurczewski@nokia.com>

1 files changed, 54 insertions, 0 deletions

diff --git a/appc-config/appc-flow-controller/provider/src/test/java/org/onap/appc/flow/controller/node/FlowGeneratorTest.java b/appc-config/appc-flow-controller/provider/src/test/java/org/onap/appc/flow/controller/node/FlowGeneratorTest.java
new file mode 100644
index 000000000..5981fc0b6
--- /dev/null
+++ b/appc-config/appc-flow-controller/provider/src/test/java/org/onap/appc/flow/controller/node/FlowGeneratorTest.java
@@ -0,0 +1,54 @@
+package org.onap.appc.flow.controller.node;
+
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+import static org.onap.appc.flow.controller.utils.FlowControllerConstants.ACTION_LEVEL;
+import static org.onap.appc.flow.controller.utils.FlowControllerConstants.PAYLOAD;
+import static org.onap.appc.flow.controller.utils.FlowControllerConstants.REQUEST_ACTION;
+
+import java.util.HashMap;
+import java.util.List;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.onap.appc.flow.controller.data.Response;
+import org.onap.appc.flow.controller.data.ResponseAction;
+import org.onap.appc.flow.controller.data.Transaction;
+import org.onap.ccsdk.sli.core.sli.SvcLogicContext;
+
+public class FlowGeneratorTest {
+
+  private HashMap<String, String> inParams;
+  private SvcLogicContext ctx;
+
+  @Before
+  public void setUp() {
+    inParams = new HashMap<>();
+    ctx = mock(SvcLogicContext.class);
+  }
+
+  @Test
+  public void should_get_proper_transactions() {
+    when(ctx.getAttribute(REQUEST_ACTION)).thenReturn("some-request-action");
+    when(ctx.getAttribute(ACTION_LEVEL)).thenReturn("some-action-level");
+    when(ctx.getAttribute(PAYLOAD)).thenReturn("some-payload");
+
+    FlowGenerator flowGenerator = new FlowGenerator();
+    List<Transaction> transactionsList = flowGenerator.createSingleStepModel(inParams, ctx).getTransactions();
+
+    Assert.assertEquals(1, transactionsList.size());
+
+    Transaction transaction = transactionsList.get(0);
+    Assert.assertEquals(1, transaction.getTransactionId());
+    Assert.assertEquals("some-request-action", transaction.getAction());
+    Assert.assertEquals("some-payload", transaction.getPayload());
+    Assert.assertEquals("some-action-level", transaction.getActionLevel());
+
+    List<Response> responses = transaction.getResponses();
+    Assert.assertEquals(1, responses.size());
+
+    ResponseAction responseAction = responses.get(0).getResponseAction();
+    Assert.assertTrue(responseAction.isStop());
+  }
+
+}
\ No newline at end of file