added graceful error handling to workflow tests

akto-api-security · Apr 16, 2024 · 6b76963 · 6b76963
1 parent c62e46b
commit 6b76963
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/apps/testing/src/main/java/com/akto/testing/workflow_node_executor/YamlNodeExecutor.java b/apps/testing/src/main/java/com/akto/testing/workflow_node_executor/YamlNodeExecutor.java
@@ -74,10 +74,18 @@ public NodeResult processNode(Node node, Map<String, Object> varMap, Boolean all
                 String apiType = firstChildNode.getValues().toString();
                 if (apiType.equalsIgnoreCase("get_asset_api")) {
                     rawApi = memory.findAssetGetterRequest(apiInfoKey);
+                    if (rawApi == null)  {
+                        testErrors.add("Couldn't find corresponding getter api");
+                        new WorkflowTestResult.NodeResult("[]",false, testErrors);
+                    }
                 }
                 childNodes.remove(0);
             } else {
                 OriginalHttpRequest request = memory.run(apiInfoKey.getApiCollectionId(), apiInfoKey.getUrl(), apiInfoKey.getMethod().name());
+                if (request == null) {
+                    testErrors.add("Failed getting request from dependency graph");
+                    new WorkflowTestResult.NodeResult("[]",false, testErrors);
+                }
                 rawApi.setRequest(request);
             }
         }