yuanbiao
/
emqx
1
0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add test case for emqx_trie:empty/0

This commit is contained in:
Feng Lee 2019-01-04 18:09:40 +08:00 committed by Feng Lee
parent 4230a3c535
commit d27c824ed8
1 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
test/emqx_trie_SUITE.erl
View File

@ -24,7 +24,7 @@
-define(TRIE_TABS, [emqx_trie, emqx_trie_node]). -define(TRIE_TABS, [emqx_trie, emqx_trie_node]).
all() -> all() ->
[t_mnesia, t_insert, t_match, t_match2, t_match3, t_delete, t_delete2, t_delete3]. [t_mnesia, t_insert, t_match, t_match2, t_match3, t_empty, t_delete, t_delete2, t_delete3].
init_per_suite(Config) -> init_per_suite(Config) ->
application:load(emqx), application:load(emqx),
@ -58,7 +58,7 @@ t_insert(_) ->
?TRIE:insert(<<"sensor">>), ?TRIE:insert(<<"sensor">>),
?TRIE:lookup(<<"sensor">>) ?TRIE:lookup(<<"sensor">>)
end, end,
?assertEqual({atomic, [TN]}, mnesia:transaction(Fun)). ?assertEqual({atomic, [TN]}, trans(Fun)).
t_match(_) -> t_match(_) ->
Machted = [<<"sensor/+/#">>, <<"sensor/#">>], Machted = [<<"sensor/+/#">>, <<"sensor/#">>],
@ -68,7 +68,7 @@ t_match(_) ->
?TRIE:insert(<<"sensor/#">>), ?TRIE:insert(<<"sensor/#">>),
?TRIE:match(<<"sensor/1">>) ?TRIE:match(<<"sensor/1">>)
end, end,
?assertEqual({atomic, Machted}, mnesia:transaction(Fun)). ?assertEqual({atomic, Machted}, trans(Fun)).
t_match2(_) -> t_match2(_) ->
Matched = {[<<"+/+/#">>, <<"+/#">>, <<"#">>], []}, Matched = {[<<"+/+/#">>, <<"+/#">>, <<"#">>], []},
@ -79,16 +79,23 @@ t_match2(_) ->
{?TRIE:match(<<"a/b/c">>), {?TRIE:match(<<"a/b/c">>),
?TRIE:match(<<"$SYS/broker/zenmq">>)} ?TRIE:match(<<"$SYS/broker/zenmq">>)}
end, end,
?assertEqual({atomic, Matched}, mnesia:transaction(Fun)). ?assertEqual({atomic, Matched}, trans(Fun)).
t_match3(_) -> t_match3(_) ->
Topics = [<<"d/#">>, <<"a/b/c">>, <<"a/b/+">>, <<"a/#">>, <<"#">>, <<"$SYS/#">>], Topics = [<<"d/#">>, <<"a/b/c">>, <<"a/b/+">>, <<"a/#">>, <<"#">>, <<"$SYS/#">>],
mnesia:transaction(fun() -> [emqx_trie:insert(Topic) || Topic <- Topics] end), trans(fun() -> [emqx_trie:insert(Topic) || Topic <- Topics] end),
Matched = mnesia:async_dirty(fun emqx_trie:match/1, [<<"a/b/c">>]), Matched = mnesia:async_dirty(fun emqx_trie:match/1, [<<"a/b/c">>]),
?assertEqual(4, length(Matched)), ?assertEqual(4, length(Matched)),
SysMatched = mnesia:async_dirty(fun emqx_trie:match/1, [<<"$SYS/a/b/c">>]), SysMatched = mnesia:async_dirty(fun emqx_trie:match/1, [<<"$SYS/a/b/c">>]),
?assertEqual([<<"$SYS/#">>], SysMatched). ?assertEqual([<<"$SYS/#">>], SysMatched).
t_empty(_) ->
?assert(?TRIE:empty()),
trans(fun ?TRIE:insert/1, [<<"topic/x/#">>]),
?assertNot(?TRIE:empty()),
trans(fun ?TRIE:delete/1, [<<"topic/x/#">>]),
?assert(?TRIE:empty()).
t_delete(_) -> t_delete(_) ->
TN = #trie_node{node_id = <<"sensor/1">>, TN = #trie_node{node_id = <<"sensor/1">>,
edge_count = 2, edge_count = 2,
@ -103,7 +110,7 @@ t_delete(_) ->
?TRIE:delete(<<"sensor/1/metric">>), ?TRIE:delete(<<"sensor/1/metric">>),
?TRIE:lookup(<<"sensor/1">>) ?TRIE:lookup(<<"sensor/1">>)
end, end,
?assertEqual({atomic, [TN]}, mnesia:transaction(Fun)). ?assertEqual({atomic, [TN]}, trans(Fun)).
t_delete2(_) -> t_delete2(_) ->
Fun = fun() -> Fun = fun() ->
@ -115,7 +122,7 @@ t_delete2(_) ->
?TRIE:delete(<<"sensor/1/metric/3">>), ?TRIE:delete(<<"sensor/1/metric/3">>),
{?TRIE:lookup(<<"sensor">>), ?TRIE:lookup(<<"sensor/1">>)} {?TRIE:lookup(<<"sensor">>), ?TRIE:lookup(<<"sensor/1">>)}
end, end,
?assertEqual({atomic, {[], []}}, mnesia:transaction(Fun)). ?assertEqual({atomic, {[], []}}, trans(Fun)).
t_delete3(_) -> t_delete3(_) ->
Fun = fun() -> Fun = fun() ->
@ -129,8 +136,13 @@ t_delete3(_) ->
?TRIE:delete(<<"sensor/+/unknown">>), ?TRIE:delete(<<"sensor/+/unknown">>),
{?TRIE:lookup(<<"sensor">>), ?TRIE:lookup(<<"sensor/+">>)} {?TRIE:lookup(<<"sensor">>), ?TRIE:lookup(<<"sensor/+">>)}
end, end,
?assertEqual({atomic, {[], []}}, mnesia:transaction(Fun)). ?assertEqual({atomic, {[], []}}, trans(Fun)).
clear_tables() -> clear_tables() ->
lists:foreach(fun mnesia:clear_table/1, ?TRIE_TABS). lists:foreach(fun mnesia:clear_table/1, ?TRIE_TABS).
trans(Fun) ->
mnesia:transaction(Fun).
trans(Fun, Args) ->
mnesia:transaction(Fun, Args).