53 lines
2.2 KiB
Erlang
53 lines
2.2 KiB
Erlang
%% Copyright (c) 2013-2019 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_banned_SUITE).
|
|
|
|
-compile(export_all).
|
|
-compile(nowarn_export_all).
|
|
|
|
-include("emqx.hrl").
|
|
-include("emqx_mqtt.hrl").
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
|
|
all() -> [t_banned_all].
|
|
|
|
t_banned_all(_) ->
|
|
emqx_ct_helpers:start_apps([]),
|
|
emqx_banned:start_link(),
|
|
TimeNow = erlang:system_time(second),
|
|
Banned = #banned{who = {client_id, <<"TestClient">>},
|
|
reason = <<"test">>,
|
|
by = <<"banned suite">>,
|
|
desc = <<"test">>,
|
|
until = TimeNow + 1},
|
|
ok = emqx_banned:add(Banned),
|
|
% here is not expire banned test because its check interval is greater than 5 mins, but its effect has been confirmed
|
|
?assert(emqx_banned:check(#{client_id => <<"TestClient">>,
|
|
username => undefined,
|
|
peername => {undefined, undefined}})),
|
|
timer:sleep(2500),
|
|
?assertNot(emqx_banned:check(#{client_id => <<"TestClient">>,
|
|
username => undefined,
|
|
peername => {undefined, undefined}})),
|
|
ok = emqx_banned:add(Banned),
|
|
?assert(emqx_banned:check(#{client_id => <<"TestClient">>,
|
|
username => undefined,
|
|
peername => {undefined, undefined}})),
|
|
emqx_banned:delete({client_id, <<"TestClient">>}),
|
|
?assertNot(emqx_banned:check(#{client_id => <<"TestClient">>,
|
|
username => undefined,
|
|
peername => {undefined, undefined}})),
|
|
emqx_ct_helpers:stop_apps([]).
|