fix a bug of branchGraph cache with test_case

Author: lvcangquan.lcq

scripts/bash-buddy

@@ -539,7 +539,8 @@ private List<TransitionNode> buildGatewayFollowingNodes(TransitionNode flowNode)
     }
 
     private List<TransitionNode> buildBranchNodes(TransitionNode branchNode) {
-        if (branchGraph.containsKey(branchNode.getId())) {
+        if (branchGraph.containsKey(branchNode.getId()) &&
+            !branchNode.getIncomingNodes().stream().anyMatch(incomingNode -> incomingNode instanceof GatewayElement)) {
             return branchGraph.get(branchNode.getId());
         }
 

src/main/java/com/alibaba/compileflow/engine/runtime/impl/AbstractProcessRuntime.java

@@ -42,6 +42,19 @@ public void testProcessEngine() {
         System.out.println(processEngine.execute(code, context));
     }
 
+    @Test
+    public void testBranchGraph() {
+        final String code = "bpm.om.branchGraph";
+        Map<String, Object> context = new HashMap<String, Object>();
+        Integer input = 16;
+        Integer compareNumber = 18;
+        context.put("input", input);
+        final ProcessEngine processEngine = ProcessEngineFactory.getProcessEngine();
+        Map<String, Object> result = processEngine.execute(code, context);
+        Integer finalResult = (Integer) result.get("finalResult");
+        assert (finalResult == compareNumber);
+    }
+
     @Test
     public void testProcessEngineBpmn20() {
         final String code = "bpmn20.ktv.ktvExample";

src/test/java/com/allibaba/compileflow/test/ProcessEngineTest.java

@@ -0,0 +1,15 @@
+package com.allibaba.compileflow.test.om.branch_graph.calculate;
+
+public class InputCheckOp {
+    public InputCheckOp() {
+    }
+
+    public Boolean process(Integer input) {
+        Integer compareNumber = 20;
+        if (input > compareNumber) {
+            return false;
+        } else {
+            return true;
+        }
+    }
+}

src/test/java/com/allibaba/compileflow/test/om/branch_graph/calculate/InputCheckOp.java

@@ -0,0 +1,12 @@
+package com.allibaba.compileflow.test.om.branch_graph.calculate;
+
+public class OnlinePreProcessOp {
+    public OnlinePreProcessOp() {
+    }
+
+    public Integer process(Integer sqrtResult) {
+        Integer addNumber = 2;
+        Integer onlineInput = sqrtResult + addNumber;
+        return onlineInput;
+    }
+}

src/test/java/com/allibaba/compileflow/test/om/branch_graph/calculate/OnlinePreProcessOp.java

@@ -0,0 +1,12 @@
+package com.allibaba.compileflow.test.om.branch_graph.calculate;
+
+public class OnlineSolveOp {
+    public OnlineSolveOp() {
+    }
+
+    public Integer process(Integer onlineInput) {
+        Integer addNumber = 2;
+        Integer onlineResult = onlineInput + addNumber;
+        return onlineResult;
+    }
+}

src/test/java/com/allibaba/compileflow/test/om/branch_graph/calculate/OnlineSolveOp.java

@@ -0,0 +1,15 @@
+package com.allibaba.compileflow.test.om.branch_graph.calculate;
+
+import java.math.BigInteger;
+
+public class PreProcessOp {
+    public PreProcessOp() {
+    }
+
+    public void process(Integer input) {
+        int compareNumber = 10;
+        if (input < compareNumber) {
+            System.out.println("input is smaller than 10");
+        }
+    }
+}

src/test/java/com/allibaba/compileflow/test/om/branch_graph/calculate/PreProcessOp.java

@@ -0,0 +1,10 @@
+package com.allibaba.compileflow.test.om.branch_graph.calculate;
+
+public class RandomOp {
+    public RandomOp() {
+    }
+
+    public void process(Integer result) {
+        System.out.println("Random process is finished " + result);
+    }
+}

src/test/java/com/allibaba/compileflow/test/om/branch_graph/calculate/RandomOp.java

@@ -0,0 +1,14 @@
+package com.allibaba.compileflow.test.om.branch_graph.calculate;
+
+public class ResultCheckOp {
+    public ResultCheckOp() {
+    }
+
+    public Boolean process(Integer result) {
+        Integer compareNumber = 10;
+        if (result < compareNumber) {
+            return true;
+        }
+        return false;
+    }
+}

src/test/java/com/allibaba/compileflow/test/om/branch_graph/calculate/ResultCheckOp.java

@@ -0,0 +1,12 @@
+package com.allibaba.compileflow.test.om.branch_graph.calculate;
+
+public class ResultPostProcessOp {
+    public ResultPostProcessOp() {
+    }
+
+    public Integer process(Integer result) {
+        Integer addNumber = 10;
+        Integer finalResult = result + addNumber;
+        return finalResult;
+    }
+}