yuanbiao
/
emqx
1
0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test: connect MQTT client with a retry

This commit is contained in:
Zaiming (Stone) Shi 2022-11-02 17:25:59 +01:00
parent 18486a7e13
commit f4621d3c71
1 changed files with 19 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
test/emqx_takeover_SUITE.erl
View File

@ -32,11 +32,11 @@
all() -> emqx_ct:all(?MODULE). all() -> emqx_ct:all(?MODULE).
init_per_suite(Config) -> init_per_suite(Config) ->
emqx_ct_helpers:start_apps([]), emqx_ct_helpers:start_apps([emqx]),
Config. Config.
end_per_suite(_Config) -> end_per_suite(_Config) ->
emqx_ct_helpers:stop_apps([]), emqx_ct_helpers:stop_apps([emqx]),
ok. ok.
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
%% Testcases %% Testcases
@ -45,10 +45,14 @@ t_takeover(_) ->
process_flag(trap_exit, true), process_flag(trap_exit, true),
AllMsgs = messages(?CNT), AllMsgs = messages(?CNT),
Pos = rand:uniform(?CNT), Pos = rand:uniform(?CNT),
ClientId = <<"clientid">>, ClientId = <<"clientid">>,
{ok, C1} = emqtt:start_link([{clientid, ClientId}, {clean_start, false}]), C1 =
{ok, _} = emqtt:connect(C1), with_retry(
fun() ->
{ok, C} = emqtt:start_link([{clientid, ClientId}, {clean_start, false}]),
{ok, _} = emqtt:connect(C),
C
end, 5),
emqtt:subscribe(C1, <<"t">>, 1), emqtt:subscribe(C1, <<"t">>, 1),
spawn(fun() -> spawn(fun() ->
@ -78,12 +82,19 @@ t_takeover(_) ->
emqtt:disconnect(C2), emqtt:disconnect(C2),
unload_meck(ClientId). unload_meck(ClientId).
t_takover_in_cluster(_) ->
todo.
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
%% Helpers %% Helpers
with_retry(Fun, 1) -> Fun();
with_retry(Fun, N) when N > 1 ->
try
Fun()
catch
_ : _ ->
ct:sleep(1000),
with_retry(Fun, N - 1)
end.
load_meck(ClientId) -> load_meck(ClientId) ->
meck:new(fake_conn_mod, [non_strict]), meck:new(fake_conn_mod, [non_strict]),
HookTakeover = fun(Pid, Msg = {takeover, 'begin'}) -> HookTakeover = fun(Pid, Msg = {takeover, 'begin'}) ->