diff --git a/test/emqttd_trie_SUITE.erl b/test/emqttd_trie_SUITE.erl index 7ee9d6616..ceda3abcb 100644 --- a/test/emqttd_trie_SUITE.erl +++ b/test/emqttd_trie_SUITE.erl @@ -22,6 +22,8 @@ -define(TRIE, emqttd_trie). +-include_lib("eunit/include/eunit.hrl"). + all() -> [t_insert, t_match, t_match2, t_match3, t_delete, t_delete2, t_delete3]. @@ -81,9 +83,9 @@ t_match3(_) -> Topics = [<<"d/#">>, <<"a/b/c">>, <<"a/b/+">>, <<"a/#">>, <<"#">>, <<"$SYS/#">>], mnesia:transaction(fun() -> [emqttd_trie:insert(Topic) || Topic <- Topics] end), Matched = mnesia:async_dirty(fun emqttd_trie:match/1, [<<"a/b/c">>]), - 4 = length(Matched), + ?assertEqual(4, length(Matched)), SysMatched = mnesia:async_dirty(fun emqttd_trie:match/1, [<<"$SYS/a/b/c">>]), - [<<"$SYS/#">>] = SysMatched. + ?assertEqual([<<"$SYS/#">>], SysMatched). t_delete(_) -> TN = #trie_node{node_id = <<"sensor/1">>, @@ -129,5 +131,5 @@ t_delete3(_) -> end). clear_tables() -> - lists:foreach(fun mnesia:clear_table/1, [trie, trie_node]). + lists:foreach(fun mnesia:clear_table/1, [mqtt_trie, mqtt_trie_node]).