add regression test

Author: csegarragonz

tests/test/scheduler/test_mpi_exec_graph.cpp

@@ -4,8 +4,12 @@
 
 #include <faabric/scheduler/MpiWorld.h>
 #include <faabric/util/exec_graph.h>
+#include <faabric/util/json.h>
 #include <faabric/util/macros.h>
 
+// DELEETEEE ME
+#include <faabric/util/logging.h>
+
 namespace tests {
 TEST_CASE_METHOD(MpiTestFixture,
                  "Test tracing the number of MPI messages",
@@ -74,4 +78,59 @@ TEST_CASE_METHOD(MpiTestFixture,
     REQUIRE(msg.intexecgraphdetails_size() == 0);
     REQUIRE(msg.execgraphdetails_size() == 0);
 }
+
+TEST_CASE_METHOD(MpiBaseTestFixture,
+                 "Test different threads populate the graph",
+                 "[util][exec-graph]")
+{
+    int rank = 0;
+    int otherRank = 1;
+    int worldSize = 2;
+    int worldId = 123;
+
+    faabric::Message msg = faabric::util::messageFactory("mpi", "hellompi");
+    msg.set_ismpi(true);
+    msg.set_recordexecgraph(true);
+    msg.set_mpiworldsize(worldSize);
+    msg.set_mpiworldid(worldId);
+
+    faabric::Message otherMsg = msg;
+    otherMsg.set_mpirank(otherRank);
+    msg.set_mpirank(rank);
+
+    faabric::scheduler::MpiWorld& thisWorld =
+      faabric::scheduler::getMpiWorldRegistry().createWorld(msg, worldId);
+
+    std::vector<int> messageData = { 0, 1, 2 };
+    auto buffer = new int[messageData.size()];
+    std::thread otherWorldThread([&messageData, &otherMsg, rank, otherRank] {
+        faabric::scheduler::MpiWorld& otherWorld =
+          faabric::scheduler::getMpiWorldRegistry().getOrInitialiseWorld(
+            otherMsg);
+
+        otherWorld.send(otherRank,
+                        rank,
+                        BYTES(messageData.data()),
+                        MPI_INT,
+                        messageData.size());
+
+        otherWorld.destroy();
+    });
+
+    thisWorld.recv(
+      otherRank, rank, BYTES(buffer), MPI_INT, messageData.size(), nullptr);
+
+    thisWorld.destroy();
+
+    if (otherWorldThread.joinable()) {
+        otherWorldThread.join();
+    }
+
+    std::string expectedKey =
+      faabric::util::exec_graph::mpiMsgCountPrefix + std::to_string(rank);
+    REQUIRE(otherMsg.mpirank() == otherRank);
+    REQUIRE(otherMsg.intexecgraphdetails_size() == 1);
+    REQUIRE(otherMsg.intexecgraphdetails().count(expectedKey) == 1);
+    REQUIRE(otherMsg.intexecgraphdetails().at(expectedKey) == 1);
+}
 }