Filter out nodes being drained test

Author: SimonUnge

deps/rabbit/test/member_evaluation_SUITE.erl

@@ -23,7 +23,7 @@ groups() ->
     [
      {unclustered, [],
       [
-       {quorum_queue_3, [], [auto_grow, auto_shrink]}
+       {quorum_queue_3, [], [auto_grow, auto_grow_drained_node, auto_shrink]}
       ]}
     ].
 
@@ -130,6 +130,43 @@ auto_grow(Config) ->
                        3 =:= length(M)
                end).
 
+auto_grow_drained_node(Config) ->
+    [Server0, Server1, Server2] =
+        rabbit_ct_broker_helpers:get_node_configs(Config, nodename),
+    Ch = rabbit_ct_client_helpers:open_channel(Config, Server0),
+
+    QQ = ?config(queue_name, Config),
+    ?assertEqual({'queue.declare_ok', QQ, 0, 0},
+                 declare(Ch, QQ, [{<<"x-queue-type">>, longstr, <<"quorum">>}])),
+
+    %% There is only one node in the cluster at the moment
+    {ok, Members, _} = ra:members({quorum_queue_utils:ra_name(QQ), Server0}),
+    ?assertEqual(1, length(Members)),
+
+    add_server_to_cluster(Server1, Server0),
+    %% mark server1 as drained, which should mean the node is not a candiate
+    %% for qq membership
+    rabbit_ct_broker_helpers:mark_as_being_drained(Config, Server1),
+    rabbit_ct_helpers:await_condition(
+        fun () -> rabbit_ct_broker_helpers:is_being_drained_local_read(Config, Server1) end,
+        10000),
+    add_server_to_cluster(Server2, Server0),
+    timer:sleep(5000),
+    %% We have 3 nodes, but one is drained, so it will not be concidered.
+    {ok, Members1, _} = ra:members({quorum_queue_utils:ra_name(QQ), Server0}),
+    ?assertEqual(1, length(Members1)),
+
+    rabbit_ct_broker_helpers:unmark_as_being_drained(Config, Server1),
+    rabbit_ct_helpers:await_condition(
+        fun () -> not rabbit_ct_broker_helpers:is_being_drained_local_read(Config, Server1) end,
+        10000),
+    %% We have 3 nodes, none is being drained, so we should grow membership to 3
+    wait_until(fun() ->
+                       {ok, M, _} = ra:members({quorum_queue_utils:ra_name(QQ), Server0}),
+                       3 =:= length(M)
+               end).
+
+
 auto_shrink(Config) ->
     [Server0, Server1, Server2] =
         rabbit_ct_broker_helpers:get_node_configs(Config, nodename),