Log failures from MPI workers

Author: kmichaelk

Diff for: tasks/mpi/runner.cpp

@@ -2,14 +2,45 @@
 
 #include <boost/mpi/communicator.hpp>
 #include <boost/mpi/environment.hpp>
+#include <memory>
+
+class WorkerTestFailurePrinter : public ::testing::EmptyTestEventListener {
+ public:
+  WorkerTestFailurePrinter(std::shared_ptr<::testing::TestEventListener> base, boost::mpi::communicator world)
+      : base_(std::move(base)), world_(std::move(world)) {}
+
+  void OnTestEnd(const ::testing::TestInfo& test_info) override {
+    if (test_info.result()->Passed()) {
+      return;
+    }
+    PrintProcessRank();
+    base_->OnTestEnd(test_info);
+  }
+
+  void OnTestPartResult(const ::testing::TestPartResult& test_part_result) override {
+    if (test_part_result.passed() || test_part_result.skipped()) {
+      return;
+    }
+    PrintProcessRank();
+    base_->OnTestPartResult(test_part_result);
+  }
+
+ private:
+  void PrintProcessRank() const { printf(" [  PROCESS %d  ] ", world_.rank()); }
+
+  std::shared_ptr<TestEventListener> base_;
+  boost::mpi::communicator world_;
+};
 
 int main(int argc, char** argv) {
   boost::mpi::environment env(argc, argv);
   boost::mpi::communicator world;
+
   ::testing::InitGoogleTest(&argc, argv);
-  ::testing::TestEventListeners& listeners = ::testing::UnitTest::GetInstance()->listeners();
-  if (world.rank() != 0) {
-    delete listeners.Release(listeners.default_result_printer());
+  auto& listeners = ::testing::UnitTest::GetInstance()->listeners();
+  if (world.rank() != 0 && (argc < 2 || argv[1] != std::string("--print-workers"))) {
+    auto* listener = listeners.Release(listeners.default_result_printer());
+    listeners.Append(new WorkerTestFailurePrinter(std::shared_ptr<::testing::TestEventListener>(listener), world));
   }
   return RUN_ALL_TESTS();
 }