#2201: test running LB on first phase

Author: cz4rs

tests/unit/collection/test_lb.extended.cc

@@ -150,6 +150,11 @@ TEST_P(TestLoadBalancerOther, test_load_balancer_other_keep_last_elm) {
   runTest(GetParam(), "test_load_balancer_other_keep_last_elm");
 }
 
+TEST_P(TestLoadBalancerOther, test_load_balancer_other_run_lb_first_phase) {
+  vt::theConfig()->vt_lb_run_lb_first_phase = true;
+  runTest(GetParam(), "test_load_balancer_other_keep_last_elm");
+}
+
 TEST_P(TestLoadBalancerGreedy, test_load_balancer_greedy_2) {
   runTest(GetParam(), "test_load_balancer_greedy_2");
 }
@@ -159,6 +164,11 @@ TEST_P(TestLoadBalancerGreedy, test_load_balancer_greedy_keep_last_elm) {
   runTest(GetParam(), "test_load_balancer_greedy_keep_last_elm");
 }
 
+TEST_P(TestLoadBalancerGreedy, test_load_balancer_greedy_run_lb_first_phase) {
+  vt::theConfig()->vt_lb_run_lb_first_phase = true;
+  runTest(GetParam(), "test_load_balancer_greedy_keep_last_elm");
+}
+
 TEST_F(TestLoadBalancerOther, test_make_graph_symmetric) {
   // setup
   auto const this_node = theContext()->getNode();