Merge pull request #10685 from rabbitmq/moar-flakes

rabbit_stream_queue_SUITE: test reliability with khepri
rabbitmq · Mar 6, 2024 · 8dbdaf4 · 8dbdaf4
2 parents ff6cbfa + e7f676d
commit 8dbdaf4
Showing 1 changed file with 18 additions and 8 deletions.
diff --git a/deps/rabbit/test/rabbit_stream_queue_SUITE.erl b/deps/rabbit/test/rabbit_stream_queue_SUITE.erl
@@ -16,6 +16,7 @@
 -compile(nowarn_export_all).
 -compile(export_all).
 
+-import(rabbit_ct_helpers, [await_condition/2]).
 -define(WAIT, 5000).
 
 suite() ->
@@ -2168,8 +2169,10 @@ leader_locator_client_local(Config) ->
                  declare(Config, Server1, Q, [{<<"x-queue-type">>, longstr, <<"stream">>},
                                               {<<"x-queue-leader-locator">>, longstr, <<"client-local">>}])),
 
-    ?assertMatch(Server1, proplists:get_value(leader,
-                                              find_queue_info(Config, [leader]))),
+    await_condition(
+      fun () ->
+              Server1 == proplists:get_value(leader, find_queue_info(Config, [leader]))
+      end, 60),
 
     ?assertMatch(#'queue.delete_ok'{},
                  delete(Config, Server1, Q)),
@@ -2180,8 +2183,13 @@ leader_locator_client_local(Config) ->
                  declare(Config, Server2, Q2, [{<<"x-queue-type">>, longstr, <<"stream">>},
                                               {<<"x-queue-leader-locator">>, longstr, <<"client-local">>}])),
 
-    ?assertMatch(Server2, proplists:get_value(leader,
-                                              find_queue_info(Q2, Config, 0, [leader]))),
+    %% the amqqueue:pid field is updated async for khepri
+    %% so we need to await the condition here
+    await_condition(
+      fun () ->
+              Server2 == proplists:get_value(leader,
+                                             find_queue_info(Q2, Config, 0, [leader]))
+      end, 60),
 
     ?assertMatch(#'queue.delete_ok'{}, delete(Config, Server2, Q2)),
 
@@ -2192,9 +2200,11 @@ leader_locator_client_local(Config) ->
                  declare(Config, Server3, Q3, [{<<"x-queue-type">>, longstr, <<"stream">>},
                                               {<<"x-queue-leader-locator">>, longstr, <<"client-local">>}])),
 
-
-    ?assertEqual(Server3, proplists:get_value(leader,
-                                              find_queue_info(Q3, Config, 0, [leader]))),
+    await_condition(
+      fun () ->
+              Server3 == proplists:get_value(leader,
+                                             find_queue_info(Q3, Config, 0, [leader]))
+      end, 60),
 
     ?assertMatch(#'queue.delete_ok'{}, delete(Config, Server3, Q3)),
     ok.
@@ -2235,7 +2245,7 @@ leader_locator_balanced_maintenance(Config) ->
                  declare(Config, Server1, Q, [{<<"x-queue-type">>, longstr, <<"stream">>},
                                               {<<"x-queue-leader-locator">>, longstr, <<"balanced">>}])),
 
-    rabbit_ct_helpers:await_condition(
+    await_condition(
       fun() ->
               Info = find_queue_info(Config, [leader]),
               Leader = proplists:get_value(leader, Info),