Assert all mod_call(s)

+

Author: MarcialRosales

deps/rabbit/test/rabbit_fifo_SUITE.erl

@@ -1750,13 +1750,28 @@ single_active_consumer_priority_test(Config) ->
                when map_size(Ch) == 0)
 
     ],
-    {#rabbit_fifo{ cfg = #cfg{resource = Resource}}, ModCalls} = run_log(Config, S0, Entries, fun single_active_invariant/1),
+    {#rabbit_fifo{ cfg = #cfg{resource = Resource}}, StateMachineEvents} = run_log(Config, S0, Entries, fun single_active_invariant/1),
+    ModCalls = [ S || S = {mod_call, rabbit_quorum_queue, update_consumer_handler, _} <- StateMachineEvents ],
+    ct:log("ModCalls: ~p", [ModCalls]),
+
+    %% At the beginning, C1 should be added as single_active    
+    assert_update_consumer_handler_called(C1, Resource, single_active, lists:nth(1, ModCalls)),
+    %% C1 should transition to waiting
+    assert_update_consumer_handler_called(C1, Resource, waiting, lists:nth(2, ModCalls)),
+    %% And C2 should become single_active
+    assert_update_consumer_handler_called(C2, Resource, single_active, lists:nth(3, ModCalls)),
+    %% Finally C2 should transition as waiting
+    assert_update_consumer_handler_called(C2, Resource, waiting, lists:nth(4, ModCalls)),
+    %% Because C3 is added as single_active
+    assert_update_consumer_handler_called(C3, Resource, single_active, lists:nth(5, ModCalls)),
+    
+    ok.
+
+assert_update_consumer_handler_called(ConsumerId, Resource, UpdatedState, ModCall) ->
     {mod_call,rabbit_quorum_queue,update_consumer_handler,
-            [Resource,
-            _,
-            false,undefined,undefined,false,waiting,[]]} = lists:nth(3, ModCalls),
-    ok.
-
+        [Resource,
+        ConsumerId,
+        _,_,_,_,UpdatedState,[]]} = ModCall.
 
 single_active_consumer_priority_cancel_active_test(Config) ->
     S0 = init(#{name => ?FUNCTION_NAME,