test(exhook): update and simplify testsuites setup

This commit is contained in:
Andrew Mayorov 2024-02-13 21:02:39 +01:00
parent ece1c6c6dc
commit 35bf805504
No known key found for this signature in database
GPG Key ID: 2837C62ACFBFED5D
3 changed files with 41 additions and 110 deletions

View File

@ -24,16 +24,11 @@
-include_lib("eunit/include/eunit.hrl"). -include_lib("eunit/include/eunit.hrl").
-include_lib("common_test/include/ct.hrl"). -include_lib("common_test/include/ct.hrl").
-include_lib("emqx/include/emqx_hooks.hrl"). -include_lib("emqx/include/emqx_hooks.hrl").
-include_lib("emqx_conf/include/emqx_conf.hrl").
-include_lib("snabbkaffe/include/snabbkaffe.hrl"). -include_lib("snabbkaffe/include/snabbkaffe.hrl").
-define(DEFAULT_CLUSTER_NAME_ATOM, emqxcl).
-define(OTHER_CLUSTER_NAME_ATOM, test_emqx_cluster).
-define(OTHER_CLUSTER_NAME_STRING, "test_emqx_cluster"). -define(OTHER_CLUSTER_NAME_STRING, "test_emqx_cluster").
-define(CONF_DEFAULT, << -define(CONF_DEFAULT, <<
"\n"
"exhook {\n" "exhook {\n"
" servers = [\n" " servers = [\n"
" { name = default,\n" " { name = default,\n"
@ -54,8 +49,6 @@
"}\n" "}\n"
>>). >>).
-import(emqx_common_test_helpers, [on_exit/1]).
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
%% Setups %% Setups
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
@ -63,47 +56,30 @@
all() -> emqx_common_test_helpers:all(?MODULE). all() -> emqx_common_test_helpers:all(?MODULE).
init_per_suite(Cfg) -> init_per_suite(Cfg) ->
application:load(emqx_conf),
ok = ekka:start(),
application:set_env(ekka, cluster_name, ?DEFAULT_CLUSTER_NAME_ATOM),
ok = mria_rlog:wait_for_shards([?CLUSTER_RPC_SHARD], infinity),
meck:new(emqx_alarm, [non_strict, passthrough, no_link]),
meck:expect(emqx_alarm, activate, 3, ok),
meck:expect(emqx_alarm, deactivate, 3, ok),
_ = emqx_exhook_demo_svr:start(), _ = emqx_exhook_demo_svr:start(),
load_cfg(?CONF_DEFAULT),
emqx_common_test_helpers:start_apps([emqx_exhook]),
Cfg. Cfg.
end_per_suite(_Cfg) -> end_per_suite(_Cfg) ->
application:set_env(ekka, cluster_name, ?DEFAULT_CLUSTER_NAME_ATOM),
ekka:stop(),
mria:stop(),
mria_mnesia:delete_schema(),
meck:unload(emqx_alarm),
emqx_common_test_helpers:stop_apps([emqx_exhook]),
emqx_exhook_demo_svr:stop(). emqx_exhook_demo_svr:stop().
init_per_testcase(_, Config) -> init_per_testcase(TC, Config) ->
{ok, _} = emqx_cluster_rpc:start_link(), Apps = emqx_cth_suite:start(
timer:sleep(200), [
Config. emqx,
{emqx_conf, emqx_conf(TC)},
{emqx_exhook, ?CONF_DEFAULT}
],
#{work_dir => emqx_cth_suite:work_dir(TC, Config)}
),
[{tc_apps, Apps} | Config].
end_per_testcase(_, _Config) -> end_per_testcase(_, Config) ->
case erlang:whereis(node()) of ok = emqx_cth_suite:stop(?config(tc_apps, Config)).
undefined ->
ok;
P ->
erlang:unlink(P),
erlang:exit(P, kill)
end,
emqx_common_test_helpers:call_janitor(),
ok.
load_cfg(Cfg) -> emqx_conf(t_cluster_name) ->
ok = emqx_common_test_helpers:load_config(emqx_exhook_schema, Cfg). io_lib:format("cluster.name = ~p", [?OTHER_CLUSTER_NAME_STRING]);
emqx_conf(_) ->
#{}.
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
%% Test cases %% Test cases
@ -320,23 +296,6 @@ t_misc_test(_) ->
ok. ok.
t_cluster_name(_) -> t_cluster_name(_) ->
SetEnvFun =
fun
(emqx) ->
application:set_env(ekka, cluster_name, ?OTHER_CLUSTER_NAME_ATOM);
(emqx_exhook) ->
ok
end,
stop_apps([emqx, emqx_exhook]),
emqx_common_test_helpers:start_apps([emqx, emqx_exhook], SetEnvFun),
on_exit(fun() ->
stop_apps([emqx, emqx_exhook]),
load_cfg(?CONF_DEFAULT),
emqx_common_test_helpers:start_apps([emqx_exhook]),
mria:wait_for_tables([?CLUSTER_MFA, ?CLUSTER_COMMIT])
end),
?assertEqual(?OTHER_CLUSTER_NAME_STRING, emqx_sys:cluster_name()), ?assertEqual(?OTHER_CLUSTER_NAME_STRING, emqx_sys:cluster_name()),
emqx_exhook_mgr:disable(<<"default">>), emqx_exhook_mgr:disable(<<"default">>),
@ -364,7 +323,7 @@ t_stop_timeout(_) ->
), ),
%% stop application %% stop application
application:stop(emqx_exhook), ok = application:stop(emqx_exhook),
?block_until(#{?snk_kind := exhook_mgr_terminated}, 20000), ?block_until(#{?snk_kind := exhook_mgr_terminated}, 20000),
%% all exhook hooked point should be unloaded %% all exhook hooked point should be unloaded
@ -379,7 +338,7 @@ t_stop_timeout(_) ->
?assertEqual(false, lists:any(fun(M) -> M == emqx_exhook_handler end, Mods)), ?assertEqual(false, lists:any(fun(M) -> M == emqx_exhook_handler end, Mods)),
%% ensure started for other tests %% ensure started for other tests
emqx_common_test_helpers:start_apps([emqx_exhook]), {ok, _} = application:ensure_all_started(emqx_exhook),
snabbkaffe:stop(), snabbkaffe:stop(),
meck:unload(emqx_exhook_demo_svr). meck:unload(emqx_exhook_demo_svr).
@ -510,10 +469,6 @@ data_file(Name) ->
cert_file(Name) -> cert_file(Name) ->
data_file(filename:join(["certs", Name])). data_file(filename:join(["certs", Name])).
%% FIXME: this creates inter-test dependency
stop_apps(Apps) ->
emqx_common_test_helpers:stop_apps(Apps, #{erase_all_configs => false}).
shuffle(List) -> shuffle(List) ->
Sorted = lists:sort(lists:map(fun(L) -> {rand:uniform(), L} end, List)), Sorted = lists:sort(lists:map(fun(L) -> {rand:uniform(), L} end, List)),
lists:map(fun({_, L}) -> L end, Sorted). lists:map(fun({_, L}) -> L end, Sorted).

View File

@ -20,16 +20,13 @@
-compile(nowarn_export_all). -compile(nowarn_export_all).
-include_lib("eunit/include/eunit.hrl"). -include_lib("eunit/include/eunit.hrl").
-include_lib("common_test/include/ct.hrl").
-define(HOST, "http://127.0.0.1:18083/"). -define(HOST, "http://127.0.0.1:18083/").
-define(API_VERSION, "v5"). -define(API_VERSION, "v5").
-define(BASE_PATH, "api"). -define(BASE_PATH, "api").
-define(CLUSTER_RPC_SHARD, emqx_cluster_rpc_shard).
-define(DEFAULT_CLUSTER_NAME_ATOM, emqxcl).
-define(CONF_DEFAULT, << -define(CONF_DEFAULT, <<
"\n"
"exhook {\n" "exhook {\n"
" servers =\n" " servers =\n"
" [ { name = default,\n" " [ { name = default,\n"
@ -56,54 +53,34 @@ all() ->
]. ].
init_per_suite(Config) -> init_per_suite(Config) ->
application:load(emqx_conf),
ok = ekka:start(),
application:set_env(ekka, cluster_name, ?DEFAULT_CLUSTER_NAME_ATOM),
ok = mria_rlog:wait_for_shards([?CLUSTER_RPC_SHARD], infinity),
meck:new(emqx_alarm, [non_strict, passthrough, no_link]),
meck:expect(emqx_alarm, activate, 3, ok),
meck:expect(emqx_alarm, deactivate, 3, ok),
_ = emqx_exhook_demo_svr:start(), _ = emqx_exhook_demo_svr:start(),
load_cfg(?CONF_DEFAULT), Apps = emqx_cth_suite:start(
emqx_mgmt_api_test_util:init_suite([emqx_exhook]), [
emqx,
emqx_conf,
emqx_management,
{emqx_exhook, ?CONF_DEFAULT},
{emqx_dashboard, "dashboard.listeners.http { enable = true, bind = 18083 }"}
],
#{work_dir => emqx_cth_suite:work_dir(Config)}
),
{ok, _} = emqx_common_test_http:create_default_app(),
[Conf] = emqx:get_raw_config([exhook, servers]), [Conf] = emqx:get_raw_config([exhook, servers]),
[{template, Conf} | Config]. [{suite_apps, Apps}, {template, Conf} | Config].
end_per_suite(Config) -> end_per_suite(Config) ->
application:set_env(ekka, cluster_name, ?DEFAULT_CLUSTER_NAME_ATOM),
ekka:stop(),
mria:stop(),
mria_mnesia:delete_schema(),
meck:unload(emqx_alarm),
emqx_mgmt_api_test_util:end_suite([emqx_exhook]),
emqx_exhook_demo_svr:stop(), emqx_exhook_demo_svr:stop(),
emqx_exhook_demo_svr:stop(<<"test1">>), emqx_exhook_demo_svr:stop(<<"test1">>),
Config. ok = emqx_cth_suite:stop(?config(suite_apps, Config)).
init_per_testcase(t_add, Config) -> init_per_testcase(t_add, Config) ->
{ok, _} = emqx_cluster_rpc:start_link(),
_ = emqx_exhook_demo_svr:start(<<"test1">>, 9001), _ = emqx_exhook_demo_svr:start(<<"test1">>, 9001),
timer:sleep(200),
Config; Config;
init_per_testcase(_, Config) -> init_per_testcase(_, Config) ->
{ok, _} = emqx_cluster_rpc:start_link(),
timer:sleep(200),
Config. Config.
end_per_testcase(_, Config) -> end_per_testcase(_, _Config) ->
case erlang:whereis(node()) of ok.
undefined ->
ok;
P ->
erlang:unlink(P),
erlang:exit(P, kill)
end,
Config.
load_cfg(Cfg) ->
ok = emqx_common_test_helpers:load_config(emqx_exhook_schema, Cfg).
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
%% Test cases %% Test cases

View File

@ -27,7 +27,6 @@
-define(TARGET_HOOK, 'message.publish'). -define(TARGET_HOOK, 'message.publish').
-define(CONF, << -define(CONF, <<
"\n"
"exhook {\n" "exhook {\n"
" servers = [\n" " servers = [\n"
" { name = succed,\n" " { name = succed,\n"
@ -48,24 +47,24 @@
all() -> emqx_common_test_helpers:all(?MODULE). all() -> emqx_common_test_helpers:all(?MODULE).
init_per_suite(Cfg) -> init_per_suite(Cfg) ->
application:load(emqx_conf),
meck:new(emqx_exhook_mgr, [non_strict, passthrough, no_link]), meck:new(emqx_exhook_mgr, [non_strict, passthrough, no_link]),
meck:new(emqx_exhook_demo_svr, [non_strict, passthrough, no_link]), meck:new(emqx_exhook_demo_svr, [non_strict, passthrough, no_link]),
meck:expect(emqx_exhook_mgr, refresh_tick, fun() -> ok end), meck:expect(emqx_exhook_mgr, refresh_tick, fun() -> ok end),
init_injections(hook_injects()), init_injections(hook_injects()),
emqx_exhook_SUITE:load_cfg(?CONF),
_ = emqx_exhook_demo_svr:start(), _ = emqx_exhook_demo_svr:start(),
_ = emqx_exhook_demo_svr:start(failed, 9001), _ = emqx_exhook_demo_svr:start(failed, 9001),
emqx_common_test_helpers:start_apps([emqx_exhook]), Apps = emqx_cth_suite:start(
Cfg. [emqx, {emqx_exhook, ?CONF}],
#{work_dir => emqx_cth_suite:work_dir(Cfg)}
),
[{suite_apps, Apps} | Cfg].
end_per_suite(_Cfg) -> end_per_suite(Cfg) ->
meck:unload(emqx_exhook_demo_svr), meck:unload(emqx_exhook_demo_svr),
meck:unload(emqx_exhook_mgr), meck:unload(emqx_exhook_mgr),
emqx_exhook_demo_svr:stop(), emqx_exhook_demo_svr:stop(),
emqx_exhook_demo_svr:stop(failed), emqx_exhook_demo_svr:stop(failed),
emqx_common_test_helpers:stop_apps([emqx_exhook]). ok = emqx_cth_suite:stop(?config(suite_apps, Cfg)).
init_per_testcase(_, Config) -> init_per_testcase(_, Config) ->
clear_metrics(), clear_metrics(),