emqx/test/emqx_gc_SUITE.erl

58 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_gc_SUITE).
-compile(export_all).
-compile(nowarn_export_all).
-include_lib("eunit/include/eunit.hrl").
all() ->
[t_init, t_run, t_info, t_reset].
t_init(_) ->
?assertEqual(undefined, emqx_gc:init(false)),
GC1 = emqx_gc:init(#{count => 10, bytes => 0}),
?assertEqual(#{cnt => {10, 10}}, emqx_gc:info(GC1)),
GC2 = emqx_gc:init(#{count => 0, bytes => 10}),
?assertEqual(#{oct => {10, 10}}, emqx_gc:info(GC2)),
GC3 = emqx_gc:init(#{count => 10, bytes => 10}),
?assertEqual(#{cnt => {10, 10}, oct => {10, 10}}, emqx_gc:info(GC3)).
t_run(_) ->
GC = emqx_gc:init(#{count => 10, bytes => 10}),
?assertEqual({true, GC}, emqx_gc:run(1, 1000, GC)),
?assertEqual({true, GC}, emqx_gc:run(1000, 1, GC)),
{false, GC1} = emqx_gc:run(1, 1, GC),
?assertEqual(#{cnt => {10, 9}, oct => {10, 9}}, emqx_gc:info(GC1)),
{false, GC2} = emqx_gc:run(2, 2, GC1),
?assertEqual(#{cnt => {10, 7}, oct => {10, 7}}, emqx_gc:info(GC2)),
{false, GC3} = emqx_gc:run(3, 3, GC2),
?assertEqual(#{cnt => {10, 4}, oct => {10, 4}}, emqx_gc:info(GC3)),
?assertEqual({true, GC}, emqx_gc:run(4, 4, GC3)).
t_info(_) ->
?assertEqual(undefined, emqx_gc:info(undefined)),
GC = emqx_gc:init(#{count => 10, bytes => 0}),
?assertEqual(#{cnt => {10, 10}}, emqx_gc:info(GC)).
t_reset(_) ->
?assertEqual(undefined, emqx_gc:reset(undefined)),
GC = emqx_gc:init(#{count => 10, bytes => 10}),
{false, GC1} = emqx_gc:run(5, 5, GC),
?assertEqual(#{cnt => {10, 5}, oct => {10, 5}}, emqx_gc:info(GC1)),
?assertEqual(GC, emqx_gc:reset(GC1)).