Simplify stream management suite

big_tests/tests/mongoose_helper.erl

@@ -35,6 +35,7 @@
 -export([make_jid/2]).
 -export([make_jid/3]).
 -export([make_jid_noprep/3]).
+-export([get_session_pid/1]).
 -export([get_session_pid/2]).
 -export([get_session_info/2]).
 -export([wait_for_route_message_count/2]).
@@ -48,6 +49,8 @@
 -export([print_debug_info_for_module/1]).
 -export([backup_and_set_config/2, backup_and_set_config_option/3, change_config_option/3]).
 -export([restore_config/1, restore_config_option/2]).
+-export([wait_for_n_offline_messages/2]).
+-export([wait_for_c2s_state_name/2, get_c2s_state_name/1]).
 
 -import(distributed_helper, [mim/0, rpc/4]).
 
@@ -440,6 +443,9 @@ make_jid(User, Server) ->
 make_jid(User, Server, Resource) ->
     jid:make(User, Server, Resource).
 
+get_session_pid(User) ->
+    get_session_pid(User, mim()).
+
 get_session_pid(User, Node) ->
     Resource = escalus_client:resource(User),
     Username = escalus_client:username(User),
@@ -562,3 +568,22 @@ do_restore_config_option(Option, {ok, Value}) ->
     rpc(mim(), mongoose_config, set_opt, [Option, Value]);
 do_restore_config_option(Option, {error, not_found}) ->
     rpc(mim(), mongoose_config, unset_opt, [Option]).
+
+wait_for_n_offline_messages(Client, N) ->
+    LUser = escalus_utils:jid_to_lower(escalus_client:username(Client)),
+    LServer = escalus_utils:jid_to_lower(escalus_client:server(Client)),
+    WaitFn = fun() -> mongoose_helper:total_offline_messages({LUser, LServer}) end,
+    wait_until(WaitFn, N).
+
+wait_for_c2s_state_name(C2SPid, NewStateName) ->
+    wait_until(fun() -> get_c2s_state_name(C2SPid) end, NewStateName,
+                #{name => get_c2s_state_name, time_left => timer:seconds(5)}).
+
+get_c2s_state_name(C2SPid) when is_pid(C2SPid) ->
+    SysStatus = rpc(mim(), sys, get_status, [C2SPid]),
+    extract_state_name(SysStatus).
+
+extract_state_name(SysStatus) ->
+    {status, _Pid, {module, _},
+     [_, _, _, _, [_, {data, FSMData} | _]]} = SysStatus,
+    proplists:get_value("StateName", FSMData).

big_tests/tests/offline_SUITE.erl

@@ -18,6 +18,7 @@
 -define(NS_FEATURE_MSGOFFLINE,  <<"msgoffline">>).
 
 -import(domain_helper, [host_type/0]).
+-import(mongoose_helper, [wait_for_n_offline_messages/2]).
 
 %%%===================================================================
 %%% Suite configuration
@@ -370,12 +371,6 @@ has_element_with_ns(Stanza, Element, NS) ->
 %%%===================================================================
 %%% Helpers
 %%%===================================================================
-wait_for_n_offline_messages(Client, N) ->
-    LUser = escalus_utils:jid_to_lower(escalus_client:username(Client)),
-    LServer = escalus_utils:jid_to_lower(escalus_client:server(Client)),
-    WaitFn = fun() -> mongoose_helper:total_offline_messages({LUser, LServer}) end,
-    mongoose_helper:wait_until(WaitFn, N).
-
 logout(Config, User) ->
     mongoose_helper:logout_user(Config, User).