67 lines
2.3 KiB
Erlang
67 lines
2.3 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2020-2021 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(prop_emqx_psk).
|
|
|
|
-include_lib("proper/include/proper.hrl").
|
|
|
|
-define(ALL(Vars, Types, Exprs),
|
|
?SETUP(fun() ->
|
|
State = do_setup(),
|
|
fun() -> do_teardown(State) end
|
|
end, ?FORALL(Vars, Types, Exprs))).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Properties
|
|
%%--------------------------------------------------------------------
|
|
|
|
prop_lookup() ->
|
|
?ALL({ClientPSKID, UserState},
|
|
{client_pskid(), user_state()},
|
|
begin
|
|
case emqx_psk:lookup(psk, ClientPSKID, UserState) of
|
|
{ok, _Result} -> true;
|
|
error -> true;
|
|
_Other -> false
|
|
end
|
|
end).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Helper
|
|
%%--------------------------------------------------------------------
|
|
|
|
do_setup() ->
|
|
ok = emqx_logger:set_log_level(emergency),
|
|
ok = meck:new(emqx_hooks, [passthrough, no_history]),
|
|
ok = meck:expect(emqx_hooks, run_fold,
|
|
fun('tls_handshake.psk_lookup', [ClientPSKID], not_found) ->
|
|
unicode:characters_to_binary(ClientPSKID)
|
|
end).
|
|
|
|
do_teardown(_) ->
|
|
ok = emqx_logger:set_log_level(error),
|
|
ok = meck:unload(emqx_hooks).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Generator
|
|
%%--------------------------------------------------------------------
|
|
|
|
client_pskid() -> oneof([string(), integer(), [1, [-1]]]).
|
|
|
|
user_state() -> term().
|
|
|