#2033: tests: add another include, sync the phase work

DARMA-tasking · Jan 23, 2023 · e86514c · e86514c
1 parent cdcc394
commit e86514c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/unit/lb/test_offlinelb.cc b/tests/unit/lb/test_offlinelb.cc
@@ -43,6 +43,7 @@
 
 #include <vt/transport.h>
 #include <vt/vrt/collection/balance/lb_data_restart_reader.h>
+#include <vt/utils/json/json_appender.h>
 
 #include <fstream>
 
@@ -137,7 +138,9 @@ TEST_F(TestOfflineLB, test_offlinelb_1) {
     .wait();
 
   for (PhaseType i = 0; i < num_phases; i++) {
-    proxy.broadcastCollective<typename SimCol::Msg, &SimCol::handler>(i);
+    runInEpochCollective("run handler", [&]{
+      proxy.broadcastCollective<typename SimCol::Msg, &SimCol::handler>(i);
+    });
     thePhase()->nextPhaseCollective();
   }
 }