Просмотр исходного кода

chore(rebalance test): fix API tests

Ilya Averyanov 2 лет назад
Родитель
Сommit
aba52c6a25
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      apps/emqx_node_rebalance/test/emqx_node_rebalance_api_SUITE.erl

+ 2 - 2
apps/emqx_node_rebalance/test/emqx_node_rebalance_api_SUITE.erl

@@ -192,7 +192,7 @@ t_start_stop_evacuation(Config) ->
     [{DonorNode, _}, {RecipientNode, _}] = ?config(cluster_nodes, Config),
 
     StartOpts = maps:merge(
-        emqx_node_rebalance_api:rebalance_evacuation_example(),
+        maps:get(evacuation, emqx_node_rebalance_api:rebalance_evacuation_example()),
         #{migrate_to => [atom_to_binary(RecipientNode)]}
     ),
 
@@ -295,7 +295,7 @@ t_start_stop_rebalance(Config) ->
 
     StartOpts = maps:without(
         [nodes],
-        emqx_node_rebalance_api:rebalance_example()
+        maps:get(rebalance, emqx_node_rebalance_api:rebalance_example())
     ),
 
     ?assertMatch(