pr comments

faasm · Oct 15, 2021 · bb95f65 · bb95f65
1 parent b206a4e
commit bb95f65
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/tests/test/scheduler/test_exec_graph.cpp b/tests/test/scheduler/test_exec_graph.cpp
@@ -77,6 +77,9 @@ TEST_CASE_METHOD(MpiBaseTestFixture,
     faabric::scheduler::MpiWorld world;
     msg.set_ismpi(true);
 
+    // Update the result for the master message
+    sch.setFunctionResult(msg);
+
     // Build the message vector to reconstruct the graph
     std::vector<faabric::Message> messages(worldSize);
     for (int rank = 0; rank < worldSize; rank++) {
@@ -91,11 +94,6 @@ TEST_CASE_METHOD(MpiBaseTestFixture,
 
     world.destroy();
 
-    // Update the result for the master message
-    sch.setFunctionResult(msg);
-    // Wait for the scheduler to set the result on the MPI non-master messages
-    SLEEP_MS(500);
-
     // Build expected graph
     ExecGraphNode nodeB1 = { .msg = messages.at(1) };
     ExecGraphNode nodeB2 = { .msg = messages.at(2) };
@@ -108,6 +106,7 @@ TEST_CASE_METHOD(MpiBaseTestFixture,
     ExecGraph expected{ .rootNode = nodeA };
 
     // Check the execution graph
+    sch.getFunctionResult(msg.id(), 500);
     ExecGraph actual = sch.getFunctionExecGraph(msg.id());
     REQUIRE(countExecGraphNodes(actual) == worldSize);
     REQUIRE(countExecGraphNodes(expected) == worldSize);