fix: add dashboard test suite & bug fix
This commit is contained in:
parent
3d47e28fa9
commit
6db01bdd85
|
@ -139,6 +139,7 @@ emqx_dashboard_api {
|
|||
zh: """新密码"""
|
||||
}
|
||||
}
|
||||
|
||||
login_failed_response400 {
|
||||
desc {
|
||||
en: """Login failed. Bad username or password"""
|
||||
|
|
|
@ -106,14 +106,14 @@ add_user_(Username, Password, Desc) ->
|
|||
mnesia:write(Admin),
|
||||
#{username => Username, description => Desc};
|
||||
[_] ->
|
||||
mnesia:abort(<<"Username Already Exist">>)
|
||||
mnesia:abort(<<"username_already_exist">>)
|
||||
end.
|
||||
|
||||
-spec(remove_user(binary()) -> {ok, any()} | {error, any()}).
|
||||
remove_user(Username) when is_binary(Username) ->
|
||||
Trans = fun() ->
|
||||
case lookup_user(Username) of
|
||||
[] -> mnesia:abort(<<"Username Not Found">>);
|
||||
[] -> mnesia:abort(<<"username_not_found">>);
|
||||
_ -> mnesia:delete({?ADMIN, Username})
|
||||
end
|
||||
end,
|
||||
|
@ -150,7 +150,7 @@ sha256(SaltBin, Password) ->
|
|||
update_user_(Username, Desc) ->
|
||||
case mnesia:wread({?ADMIN, Username}) of
|
||||
[] ->
|
||||
mnesia:abort(<<"Username Not Found">>);
|
||||
mnesia:abort(<<"username_not_found">>);
|
||||
[Admin] ->
|
||||
mnesia:write(Admin#?ADMIN{description = Desc}),
|
||||
#{username => Username, description => Desc}
|
||||
|
@ -184,7 +184,7 @@ update_pwd(Username, Fun) ->
|
|||
case lookup_user(Username) of
|
||||
[Admin] -> Admin;
|
||||
[] ->
|
||||
mnesia:abort(<<"Username Not Found">>)
|
||||
mnesia:abort(<<"username_not_found">>)
|
||||
end,
|
||||
mnesia:write(Fun(User))
|
||||
end,
|
||||
|
|
|
@ -192,10 +192,10 @@ field(version) ->
|
|||
{version, mk(string(), #{desc => ?DESC(version), example => <<"5.0.0">>})};
|
||||
|
||||
field(old_pwd) ->
|
||||
{password, mk(binary(), #{desc => ?DESC(old_pwd)})};
|
||||
{old_pwd, mk(binary(), #{desc => ?DESC(old_pwd)})};
|
||||
|
||||
field(new_pwd) ->
|
||||
{password, mk(binary(), #{desc => ?DESC(new_pwd)})}.
|
||||
{new_pwd, mk(binary(), #{desc => ?DESC(new_pwd)})}.
|
||||
|
||||
%% -------------------------------------------------------------------------------------------------
|
||||
%% API
|
||||
|
@ -275,20 +275,26 @@ user(delete, #{bindings := #{username := Username}}) ->
|
|||
end.
|
||||
|
||||
change_pwd(put, #{bindings := #{username := Username}, body := Params}) ->
|
||||
LogMeta = #{msg => "Dashboard change password", username => Username},
|
||||
OldPwd = maps:get(<<"old_pwd">>, Params),
|
||||
NewPwd = maps:get(<<"new_pwd">>, Params),
|
||||
case ?EMPTY(OldPwd) orelse ?EMPTY(NewPwd) of
|
||||
true ->
|
||||
?SLOG(error, LogMeta#{result => failed, reason => "password undefined or empty"}),
|
||||
{400, ?BAD_REQUEST, <<"Old password or new password undefined">>};
|
||||
false ->
|
||||
case emqx_dashboard_admin:change_password(Username, OldPwd, NewPwd) of
|
||||
{ok, _} ->
|
||||
?SLOG(info, LogMeta#{result => success}),
|
||||
{204};
|
||||
{error, <<"username_not_found">>} ->
|
||||
?SLOG(error, LogMeta#{result => failed, reason => "username not found"}),
|
||||
{404, ?USER_NOT_FOUND, <<"User not found">>};
|
||||
{error, <<"password_error">>} ->
|
||||
?SLOG(error, LogMeta#{result => failed, reason => "error old pwd"}),
|
||||
{401, ?ERROR_PWD_NOT_MATCH, <<"Old password not match">>};
|
||||
{error, Reason} ->
|
||||
?SLOG(error, LogMeta#{result => failed, reason => Reason}),
|
||||
{400, ?BAD_REQUEST, Reason}
|
||||
end
|
||||
end.
|
||||
|
|
|
@ -0,0 +1,188 @@
|
|||
%%--------------------------------------------------------------------
|
||||
%% Copyright (c) 2020-2022 EMQ Technologies Co., Ltd. All Rights Reserved.
|
||||
%%
|
||||
%% Licensed under the Apache License, Version 2.0 (the "License");
|
||||
%% you may not use this file except in compliance with the License.
|
||||
%% You may obtain a copy of the License at
|
||||
%%
|
||||
%% http://www.apache.org/licenses/LICENSE-2.0
|
||||
%%
|
||||
%% Unless required by applicable law or agreed to in writing, software
|
||||
%% distributed under the License is distributed on an "AS IS" BASIS,
|
||||
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
%% See the License for the specific language governing permissions and
|
||||
%% limitations under the License.
|
||||
%%--------------------------------------------------------------------
|
||||
-module(emqx_dashboard_admin_SUITE).
|
||||
|
||||
-compile(nowarn_export_all).
|
||||
-compile(export_all).
|
||||
|
||||
-include("emqx_dashboard.hrl").
|
||||
-include_lib("emqx/include/http_api.hrl").
|
||||
-include_lib("eunit/include/eunit.hrl").
|
||||
|
||||
all() ->
|
||||
emqx_common_test_helpers:all(?MODULE).
|
||||
|
||||
init_per_suite(Config) ->
|
||||
mria:start(),
|
||||
application:load(emqx_dashboard),
|
||||
emqx_common_test_helpers:start_apps([emqx_conf, emqx_dashboard], fun set_special_configs/1),
|
||||
Config.
|
||||
|
||||
set_special_configs(emqx_dashboard) ->
|
||||
emqx_dashboard_api_test_helpers:set_default_config(),
|
||||
ok;
|
||||
set_special_configs(_) ->
|
||||
ok.
|
||||
|
||||
end_per_suite(Config) ->
|
||||
end_suite(),
|
||||
Config.
|
||||
|
||||
end_per_testcase(_, _Config) ->
|
||||
All = emqx_dashboard_admin:all_users(),
|
||||
[emqx_dashboard_admin:remove_user(Name) || #{username := Name} <- All].
|
||||
|
||||
end_suite() ->
|
||||
application:unload(emqx_management),
|
||||
emqx_common_test_helpers:stop_apps([emqx_dashboard]).
|
||||
|
||||
t_check_user(_) ->
|
||||
Username = <<"admin1">>,
|
||||
Password = <<"public">>,
|
||||
BadUsername = <<"admin_bad">>,
|
||||
BadPassword = <<"public_bad">>,
|
||||
EmptyUsername = <<>>,
|
||||
EmptyPassword = <<>>,
|
||||
{ok, _} = emqx_dashboard_admin:add_user(Username, Password, <<"desc">>),
|
||||
ok = emqx_dashboard_admin:check(Username, Password),
|
||||
{error, <<"password_error">>} = emqx_dashboard_admin:check(Username, BadPassword),
|
||||
{error, <<"username_not_found">>} = emqx_dashboard_admin:check(BadUsername, Password),
|
||||
{error, <<"username_not_found">>} = emqx_dashboard_admin:check(BadUsername, BadPassword),
|
||||
{error, <<"username_not_found">>} = emqx_dashboard_admin:check(EmptyUsername, Password),
|
||||
{error, <<"password_error">>} = emqx_dashboard_admin:check(Username, EmptyPassword),
|
||||
{error, <<"username_not_provided">>} = emqx_dashboard_admin:check(undefined, Password),
|
||||
{error, <<"password_not_provided">>} = emqx_dashboard_admin:check(Username, undefined),
|
||||
ok.
|
||||
|
||||
t_add_user(_) ->
|
||||
AddUser = <<"add_user">>,
|
||||
AddPassword = <<"add_password">>,
|
||||
AddDescription = <<"add_description">>,
|
||||
|
||||
BadAddUser = <<"***add_user_bad">>,
|
||||
|
||||
%% add success. not return password
|
||||
{ok, NewUser} = emqx_dashboard_admin:add_user(AddUser, AddPassword, AddDescription),
|
||||
AddUser = maps:get(username, NewUser),
|
||||
AddDescription = maps:get(description, NewUser),
|
||||
false = maps:is_key(password, NewUser),
|
||||
|
||||
%% add again
|
||||
{error, <<"username_already_exist">>} =
|
||||
emqx_dashboard_admin:add_user(AddUser, AddPassword, AddDescription),
|
||||
|
||||
%% add bad username
|
||||
BadNameError =
|
||||
<<"Bad Username. Only upper and lower case letters, numbers and underscores are supported">>,
|
||||
{error, BadNameError} = emqx_dashboard_admin:add_user(BadAddUser, AddPassword, AddDescription),
|
||||
ok.
|
||||
|
||||
t_lookup_user(_) ->
|
||||
LookupUser = <<"lookup_user">>,
|
||||
LookupPassword = <<"lookup_password">>,
|
||||
LookupDescription = <<"lookup_description">>,
|
||||
|
||||
BadLookupUser = <<"***lookup_user_bad">>,
|
||||
|
||||
{ok, _} =
|
||||
emqx_dashboard_admin:add_user(LookupUser, LookupPassword, LookupDescription),
|
||||
%% lookup success. not return password
|
||||
[#emqx_admin{username = LookupUser, description = LookupDescription}] =
|
||||
emqx_dashboard_admin:lookup_user(LookupUser),
|
||||
|
||||
[] = emqx_dashboard_admin:lookup_user(BadLookupUser),
|
||||
ok.
|
||||
|
||||
t_all_users(_) ->
|
||||
Username = <<"admin_all">>,
|
||||
Password = <<"public">>,
|
||||
{ok, _} = emqx_dashboard_admin:add_user(Username, Password, <<"desc">>),
|
||||
All = emqx_dashboard_admin:all_users(),
|
||||
?assert(erlang:length(All) >= 1),
|
||||
ok.
|
||||
|
||||
t_delete_user(_) ->
|
||||
DeleteUser = <<"delete_user">>,
|
||||
DeletePassword = <<"delete_password">>,
|
||||
DeleteDescription = <<"delete_description">>,
|
||||
|
||||
DeleteBadUser = <<"delete_user_bad">>,
|
||||
|
||||
{ok, _NewUser} =
|
||||
emqx_dashboard_admin:add_user(DeleteUser, DeletePassword, DeleteDescription),
|
||||
{ok, ok} = emqx_dashboard_admin:remove_user(DeleteUser),
|
||||
%% remove again
|
||||
{error, <<"username_not_found">>} = emqx_dashboard_admin:remove_user(DeleteUser),
|
||||
{error, <<"username_not_found">>} = emqx_dashboard_admin:remove_user(DeleteBadUser),
|
||||
ok.
|
||||
|
||||
t_update_user(_) ->
|
||||
UpdateUser = <<"update_user">>,
|
||||
UpdatePassword = <<"update_password">>,
|
||||
UpdateDescription = <<"update_description">>,
|
||||
|
||||
NewDesc = <<"new_description">>,
|
||||
|
||||
BadUpdateUser = <<"update_user_bad">>,
|
||||
|
||||
{ok, _} = emqx_dashboard_admin:add_user(UpdateUser, UpdatePassword, UpdateDescription),
|
||||
{ok, NewUserInfo} =
|
||||
emqx_dashboard_admin:update_user(UpdateUser, NewDesc),
|
||||
UpdateUser = maps:get(username, NewUserInfo),
|
||||
NewDesc = maps:get(description, NewUserInfo),
|
||||
|
||||
{error,<<"username_not_found">>} = emqx_dashboard_admin:update_user(BadUpdateUser, NewDesc),
|
||||
ok.
|
||||
|
||||
t_change_password(_) ->
|
||||
User = <<"change_user">>,
|
||||
OldPassword = <<"change_password">>,
|
||||
Description = <<"change_description">>,
|
||||
|
||||
NewPassword = <<"new_password">>,
|
||||
|
||||
BadChangeUser = <<"change_user_bad">>,
|
||||
|
||||
{ok, _} = emqx_dashboard_admin:add_user(User, OldPassword, Description),
|
||||
|
||||
{ok, ok} = emqx_dashboard_admin:change_password(User, OldPassword, NewPassword),
|
||||
%% change pwd again
|
||||
{error,<<"password_error">>} =
|
||||
emqx_dashboard_admin:change_password(User, OldPassword, NewPassword),
|
||||
|
||||
{error, <<"username_not_found">>} =
|
||||
emqx_dashboard_admin:change_password(BadChangeUser, OldPassword, NewPassword),
|
||||
ok.
|
||||
|
||||
t_clean_token(_) ->
|
||||
Username = <<"admin_token">>,
|
||||
Password = <<"public">>,
|
||||
NewPassword = <<"public1">>,
|
||||
{ok, _} = emqx_dashboard_admin:add_user(Username, Password, <<"desc">>),
|
||||
{ok, Token} = emqx_dashboard_admin:sign_token(Username, Password),
|
||||
ok = emqx_dashboard_admin:verify_token(Token),
|
||||
%% change password
|
||||
{ok, _} = emqx_dashboard_admin:change_password(Username, Password, NewPassword),
|
||||
timer:sleep(5),
|
||||
{error, not_found} = emqx_dashboard_admin:verify_token(Token),
|
||||
%% remove user
|
||||
{ok, Token2} = emqx_dashboard_admin:sign_token(Username, NewPassword),
|
||||
ok = emqx_dashboard_admin:verify_token(Token2),
|
||||
{ok, _} = emqx_dashboard_admin:remove_user(Username),
|
||||
timer:sleep(5),
|
||||
{error, not_found} = emqx_dashboard_admin:verify_token(Token2),
|
||||
ok.
|
||||
|
Loading…
Reference in New Issue