queue_SUITE: use a different upstream for each queue on multi-federat…

…ion tests
rabbitmq · Oct 29, 2024 · 91d8194 · 91d8194
1 parent 385b7e4
commit 91d8194
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 10 deletions.
diff --git a/deps/rabbitmq_federation/test/queue_SUITE.erl b/deps/rabbitmq_federation/test/queue_SUITE.erl
@@ -201,7 +201,7 @@ multiple_downstreams(Config) ->
       fun (Ch) ->
               timer:sleep(?INITIAL_WAIT),
               expect_federation(Config, Ch, <<"upstream">>, <<"fed.downstream">>, ?EXPECT_FEDERATION_TIMEOUT),
-              expect_federation(Config, Ch, <<"upstream">>, <<"fed.downstream2">>, ?EXPECT_FEDERATION_TIMEOUT)
+              expect_federation(Config, Ch, <<"upstream2">>, <<"fed.downstream2">>, ?EXPECT_FEDERATION_TIMEOUT)
       end, upstream_downstream(Config) ++ [q(<<"fed.downstream2">>, Args)]).
 
 message_flow(Config) ->
@@ -258,13 +258,13 @@ federate_unfederate(Config) ->
       fun (Ch) ->
               timer:sleep(?INITIAL_WAIT),
               expect_federation(Config, Ch, <<"upstream">>, <<"fed.downstream">>, ?EXPECT_FEDERATION_TIMEOUT),
-              expect_federation(Config, Ch, <<"upstream">>, <<"fed.downstream2">>, ?EXPECT_FEDERATION_TIMEOUT),
+              expect_federation(Config, Ch, <<"upstream2">>, <<"fed.downstream2">>, ?EXPECT_FEDERATION_TIMEOUT),
 
               %% clear the policy
               rabbit_ct_broker_helpers:clear_policy(Config, 0, <<"fed">>),
 
               expect_no_federation(Ch, <<"upstream">>, <<"fed.downstream">>),
-              expect_no_federation(Ch, <<"upstream">>, <<"fed.downstream2">>),
+              expect_no_federation(Ch, <<"upstream2">>, <<"fed.downstream2">>),
 
               rabbit_ct_broker_helpers:set_policy(Config, 0,
                 <<"fed">>, <<"^fed\.">>, <<"all">>, [
@@ -277,17 +277,18 @@ dynamic_plugin_stop_start(Config) ->
     with_ch(Config,
       fun (Ch) ->
           timer:sleep(?INITIAL_WAIT),
-          UpQ = <<"upstream">>,
+          UpQ1 = <<"upstream">>,
+          UpQ2 = <<"upstream2">>,
           DownQ1 = <<"fed.downstream">>,
-          expect_federation(Config, Ch, UpQ, DownQ1, ?EXPECT_FEDERATION_TIMEOUT),
-          expect_federation(Config, Ch, UpQ, DownQ2, ?EXPECT_FEDERATION_TIMEOUT),
+          expect_federation(Config, Ch, UpQ1, DownQ1, ?EXPECT_FEDERATION_TIMEOUT),
+          expect_federation(Config, Ch, UpQ2, DownQ2, ?EXPECT_FEDERATION_TIMEOUT),
 
           %% Disable the plugin, the link disappears
           ct:pal("Stopping rabbitmq_federation"),
           ok = rabbit_ct_broker_helpers:disable_plugin(Config, 0, "rabbitmq_federation"),
 
-          expect_no_federation(Ch, UpQ, DownQ1),
-          expect_no_federation(Ch, UpQ, DownQ2),
+          expect_no_federation(Ch, UpQ1, DownQ1),
+          expect_no_federation(Ch, UpQ2, DownQ2),
 
           maybe_declare_queue(Config, Ch, q(DownQ1, Args)),
           maybe_declare_queue(Config, Ch, q(DownQ2, Args)),
@@ -305,12 +306,13 @@ dynamic_plugin_stop_start(Config) ->
                         Entry || Entry <- Status,
                         proplists:get_value(queue, Entry) =:= DownQ1 orelse
                         proplists:get_value(queue, Entry) =:= DownQ2,
-                        proplists:get_value(upstream_queue, Entry) =:= UpQ,
+                        proplists:get_value(upstream_queue, Entry) =:= UpQ1 orelse
+                                     proplists:get_value(upstream_queue, Entry) =:= UpQ2,
                         proplists:get_value(status, Entry) =:= running
                     ],
                     length(L) =:= 2
             end),
-          expect_federation(Config, Ch, UpQ, DownQ1, 120000)
+          expect_federation(Config, Ch, UpQ1, DownQ1, 120000)
       end, upstream_downstream(Config) ++ [q(DownQ2, Args)]).
 
 restart_upstream(Config) ->

diff --git a/deps/rabbitmq_federation/test/rabbit_federation_test_util.erl b/deps/rabbitmq_federation/test/rabbit_federation_test_util.erl
@@ -99,6 +99,11 @@ setup_federation_with_upstream_params(Config, ExtraParams) ->
       [<<"/">>, <<"fed">>, <<"^fed\.">>, [{<<"federation-upstream-set">>, <<"upstream">>}],
        0, <<"all">>, <<"acting-user">>]),
 
+    rabbit_ct_broker_helpers:rpc(
+      Config, 0, rabbit_policy, set,
+      [<<"/">>, <<"fed2">>, <<"^fed\.downstream2">>, [{<<"federation-upstream-set">>, <<"upstream2">>}],
+       0, <<"all">>, <<"acting-user">>]),
+
     rabbit_ct_broker_helpers:rpc(
       Config, 0, rabbit_policy, set,
       [<<"/">>, <<"fed12">>, <<"^fed12\.">>, [{<<"federation-upstream-set">>, <<"upstream12">>}],