102 lines
3.7 KiB
Erlang
102 lines
3.7 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_stats_tests).
|
|
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
|
|
get_state_test() ->
|
|
with_proc(fun() ->
|
|
SetConnsCount = emqx_stats:statsfun('connections.count'),
|
|
SetConnsCount(1),
|
|
1 = emqx_stats:getstat('connections.count'),
|
|
emqx_stats:setstat('connections.count', 2),
|
|
2 = emqx_stats:getstat('connections.count'),
|
|
emqx_stats:setstat('connections.count', 'connections.max', 3),
|
|
timer:sleep(100),
|
|
3 = emqx_stats:getstat('connections.count'),
|
|
3 = emqx_stats:getstat('connections.max'),
|
|
emqx_stats:setstat('connections.count', 'connections.max', 2),
|
|
timer:sleep(100),
|
|
2 = emqx_stats:getstat('connections.count'),
|
|
3 = emqx_stats:getstat('connections.max'),
|
|
SetConns = emqx_stats:statsfun('connections.count', 'connections.max'),
|
|
SetConns(4),
|
|
timer:sleep(100),
|
|
4 = emqx_stats:getstat('connections.count'),
|
|
4 = emqx_stats:getstat('connections.max'),
|
|
Conns = emqx_stats:getstats(),
|
|
4 = proplists:get_value('connections.count', Conns),
|
|
4 = proplists:get_value('connections.max', Conns)
|
|
end).
|
|
|
|
update_interval_test() ->
|
|
TickMs = 200,
|
|
with_proc(fun() ->
|
|
SleepMs = TickMs * 2 + TickMs div 2, %% sleep for 2.5 ticks
|
|
emqx_stats:cancel_update(cm_stats),
|
|
UpdFun = fun() -> emqx_stats:setstat('connections.count', 1) end,
|
|
ok = emqx_stats:update_interval(stats_test, UpdFun),
|
|
timer:sleep(SleepMs),
|
|
?assertEqual(1, emqx_stats:getstat('connections.count'))
|
|
end, TickMs).
|
|
|
|
helper_test_() ->
|
|
TickMs = 200,
|
|
TestF =
|
|
fun(CbModule, CbFun) ->
|
|
SleepMs = TickMs + TickMs div 2, %% sleep for 1.5 ticks
|
|
Ref = make_ref(),
|
|
Tester = self(),
|
|
UpdFun =
|
|
fun() ->
|
|
CbModule:CbFun(),
|
|
Tester ! Ref,
|
|
ok
|
|
end,
|
|
ok = emqx_stats:update_interval(stats_test, UpdFun),
|
|
timer:sleep(SleepMs),
|
|
receive Ref -> ok after 2000 -> error(timeout) end
|
|
end,
|
|
MkTestFun =
|
|
fun(CbModule, CbFun) ->
|
|
fun() ->
|
|
with_proc(fun() -> TestF(CbModule, CbFun) end, TickMs)
|
|
end
|
|
end,
|
|
[{"emqx_broker", MkTestFun(emqx_broker, stats_fun)},
|
|
{"emqx_sm", MkTestFun(emqx_sm, stats_fun)},
|
|
{"emqx_router_helper", MkTestFun(emqx_router_helper, stats_fun)},
|
|
{"emqx_cm", MkTestFun(emqx_cm, stats_fun)}
|
|
].
|
|
|
|
with_proc(F) ->
|
|
{ok, _Pid} = emqx_stats:start_link(),
|
|
with_stop(F).
|
|
|
|
with_proc(F, TickMs) ->
|
|
{ok, _Pid} = emqx_stats:start_link(#{tick_ms => TickMs}),
|
|
with_stop(F).
|
|
|
|
with_stop(F) ->
|
|
try
|
|
%% make a synced call to the gen_server so we know it has
|
|
%% started running, hence it is safe to continue with less risk of race condition
|
|
?assertEqual(ignored, gen_server:call(emqx_stats, ignored)),
|
|
F()
|
|
after
|
|
ok = emqx_stats:stop()
|
|
end.
|
|
|