From 80e55215110a9ef468868bfe1eba801719d080bc Mon Sep 17 00:00:00 2001
From: Carlos Segarra <carlos@carlossegarra.com>
Date: Fri, 29 Oct 2021 15:37:16 +0000
Subject: [PATCH] cleanup

---
 src/scheduler/MpiWorld.cpp               | 1 +
 tests/test/scheduler/test_exec_graph.cpp | 1 +
 tests/test/util/test_json.cpp            | 1 -
 3 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/scheduler/MpiWorld.cpp b/src/scheduler/MpiWorld.cpp
index e88ab3386..ab814f4d5 100644
--- a/src/scheduler/MpiWorld.cpp
+++ b/src/scheduler/MpiWorld.cpp
@@ -201,6 +201,7 @@ void MpiWorld::create(faabric::Message& call, int newId, int newSize)
         // Log chained functions to generate execution graphs
         if (thisRankMsg != nullptr && thisRankMsg->recordexecgraph()) {
             sch.logChainedFunction(call.id(), msg.id());
+            msg.set_recordexecgraph(true);
         }
     }
 
diff --git a/tests/test/scheduler/test_exec_graph.cpp b/tests/test/scheduler/test_exec_graph.cpp
index e6a7e5bef..b1619b171 100644
--- a/tests/test/scheduler/test_exec_graph.cpp
+++ b/tests/test/scheduler/test_exec_graph.cpp
@@ -98,6 +98,7 @@ TEST_CASE_METHOD(MpiBaseTestFixture,
         messages.at(rank).set_mpiworldid(worldId);
         messages.at(rank).set_mpirank(rank);
         messages.at(rank).set_mpiworldsize(worldSize);
+        messages.at(rank).set_recordexecgraph(true);
     }
 
     world.create(msg, worldId, worldSize);
diff --git a/tests/test/util/test_json.cpp b/tests/test/util/test_json.cpp
index df3248ae3..f8ac1766d 100644
--- a/tests/test/util/test_json.cpp
+++ b/tests/test/util/test_json.cpp
@@ -61,7 +61,6 @@ TEST_CASE("Test message to JSON round trip", "[util]")
     REQUIRE(msg.timestamp() > 0);
 
     std::string jsonString = faabric::util::messageToJson(msg);
-    SPDLOG_INFO("{}", jsonString);
 
     faabric::Message actual = faabric::util::jsonToMessage(jsonString);