fix(dashboard): add more test suite

This commit is contained in:
DDDHuang 2022-06-21 19:09:51 +08:00
parent 0b32bf72f7
commit f502e09f5f
2 changed files with 27 additions and 30 deletions

View File

@ -162,7 +162,7 @@ lookup_by_username(Username) ->
{atomic, List} = mria:ro_transaction(?DASHBOARD_SHARD, Fun), {atomic, List} = mria:ro_transaction(?DASHBOARD_SHARD, Fun),
List. List.
-spec owner(Token :: binary()) -> {ok, Username :: binary} | {error, not_found}. -spec owner(Token :: binary()) -> {ok, Username :: binary()} | {error, not_found}.
owner(Token) -> owner(Token) ->
Fun = fun() -> mnesia:read(?TAB, Token) end, Fun = fun() -> mnesia:read(?TAB, Token) end,
case mria:ro_transaction(?DASHBOARD_SHARD, Fun) of case mria:ro_transaction(?DASHBOARD_SHARD, Fun) of

View File

@ -44,23 +44,17 @@
-define(APP_MANAGEMENT, emqx_management). -define(APP_MANAGEMENT, emqx_management).
-define(OVERVIEWS, [ -define(OVERVIEWS, [
'alarms/activated', "alarms",
'alarms/deactivated', "banned",
banned, "stats",
brokers, "metrics",
stats, "listeners",
metrics, "clients",
listeners, "subscriptions"
clients,
subscriptions,
routes,
plugins
]). ]).
all() -> all() ->
%% TODO: V5 API emqx_common_test_helpers:all(?MODULE).
%% emqx_common_test_helpers:all(?MODULE).
[t_cli, t_lookup_by_username_jwt, t_clean_expired_jwt, t_rest_api].
end_suite() -> end_suite() ->
end_suite([]). end_suite([]).
@ -98,37 +92,40 @@ t_overview(_) ->
mnesia:clear_table(?ADMIN), mnesia:clear_table(?ADMIN),
emqx_dashboard_admin:add_user(<<"admin">>, <<"public">>, <<"simple_description">>), emqx_dashboard_admin:add_user(<<"admin">>, <<"public">>, <<"simple_description">>),
[ [
?assert( {ok, _} = request_dashboard(get, api_path([Overview]), auth_header_())
request_dashboard(
get,
api_path(erlang:atom_to_list(Overview)),
auth_header_()
)
)
|| Overview <- ?OVERVIEWS || Overview <- ?OVERVIEWS
]. ].
t_admins_add_delete(_) -> t_admins_add_delete(_) ->
mnesia:clear_table(?ADMIN), mnesia:clear_table(?ADMIN),
Desc = <<"simple description">>, Desc = <<"simple description">>,
ok = emqx_dashboard_admin:add_user(<<"username">>, <<"password">>, Desc), {ok, _} = emqx_dashboard_admin:add_user(<<"username">>, <<"password">>, Desc),
ok = emqx_dashboard_admin:add_user(<<"username1">>, <<"password1">>, Desc), {ok, _} = emqx_dashboard_admin:add_user(<<"username1">>, <<"password1">>, Desc),
Admins = emqx_dashboard_admin:all_users(), Admins = emqx_dashboard_admin:all_users(),
?assertEqual(2, length(Admins)), ?assertEqual(2, length(Admins)),
ok = emqx_dashboard_admin:remove_user(<<"username1">>), {ok, _} = emqx_dashboard_admin:remove_user(<<"username1">>),
Users = emqx_dashboard_admin:all_users(), Users = emqx_dashboard_admin:all_users(),
?assertEqual(1, length(Users)), ?assertEqual(1, length(Users)),
ok = emqx_dashboard_admin:change_password( {ok, _} = emqx_dashboard_admin:change_password(
<<"username">>, <<"username">>,
<<"password">>, <<"password">>,
<<"pwd">> <<"pwd">>
), ),
timer:sleep(10), timer:sleep(10),
Header = auth_header_(<<"username">>, <<"pwd">>), {ok, _} = emqx_dashboard_admin:remove_user(<<"username">>).
?assert(request_dashboard(get, api_path("brokers"), Header)),
ok = emqx_dashboard_admin:remove_user(<<"username">>), t_admin_delete_self_failed(_) ->
?assertNotEqual(true, request_dashboard(get, api_path("brokers"), Header)). mnesia:clear_table(?ADMIN),
Desc = <<"simple description">>,
_ = emqx_dashboard_admin:add_user(<<"username1">>, <<"password">>, Desc),
Admins = emqx_dashboard_admin:all_users(),
?assertEqual(1, length(Admins)),
Header = auth_header_(<<"username1">>, <<"password">>),
{error, {_, 400, _}} = request_dashboard(delete, api_path(["users", "username1"]), Header),
Token = erlang:iolist_to_binary(["Basic ", base64:encode("username1:password")]),
Header2 = {"Authorization", Token},
{error, {_, 400, _}} = request_dashboard(delete, api_path(["users", "username1"]), Header2),
mnesia:clear_table(?ADMIN).
t_rest_api(_Config) -> t_rest_api(_Config) ->
mnesia:clear_table(?ADMIN), mnesia:clear_table(?ADMIN),