faaslet: test potentially flaky test

csegarragonz · csegarragonz · commit a056c5083d9d · 2023-07-04T12:14:41.000Z
diff --git a/tests/test/faaslet/test_exec_graph.cpp b/tests/test/faaslet/test_exec_graph.cpp
@@ -53,8 +53,12 @@ TEST_CASE_METHOD(FunctionExecTestFixture,
     sch.callFunctions(req);
     faabric::Message result = sch.getFunctionResult(call, 5000);
     REQUIRE(result.returnvalue() == 0);
+    for (const int msgId : faabric::util::getChainedFunctions(call)) {
+        auto chainedResult = sch.getFunctionResult(call.appid(), msgId, 1000);
+        REQUIRE(chainedResult.returnvalue() == 0);
+    }
 
-    auto execGraph = faabric::util::getFunctionExecGraph(call);
+    auto execGraph = faabric::util::getFunctionExecGraph(result);
     int numNodes = faabric::util::countExecGraphNodes(execGraph);
     REQUIRE(numNodes == expectedNumNodes);
     REQUIRE(execGraph.rootNode.msg.id() == call.id());
