From 0063895706cdfbc8923a1e7f3edf87c5894623b3 Mon Sep 17 00:00:00 2001 From: Feng Date: Fri, 23 Sep 2016 15:29:39 +0800 Subject: [PATCH] test emqttd:run_hooks/2 --- test/emqttd_SUITE.erl | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/test/emqttd_SUITE.erl b/test/emqttd_SUITE.erl index 41ad9f60e..0c20b9bd5 100644 --- a/test/emqttd_SUITE.erl +++ b/test/emqttd_SUITE.erl @@ -332,11 +332,16 @@ add_delete_hook(_) -> [] = emqttd_hook:lookup(emqttd_hook). run_hooks(_) -> - emqttd:hook(test_hook, fun ?MODULE:hook_fun3/4, [init]), - emqttd:hook(test_hook, fun ?MODULE:hook_fun4/4, [init]), - emqttd:hook(test_hook, fun ?MODULE:hook_fun5/4, [init]), - {stop, [r3, r2]} = emqttd:run_hooks(test_hook, [arg1, arg2], []), - {ok, []} = emqttd:run_hooks(unknown_hook, [], []). + emqttd:hook(foldl_hook, fun ?MODULE:hook_fun3/4, [init]), + emqttd:hook(foldl_hook, fun ?MODULE:hook_fun4/4, [init]), + emqttd:hook(foldl_hook, fun ?MODULE:hook_fun5/4, [init]), + {stop, [r3, r2]} = emqttd:run_hooks(foldl_hook, [arg1, arg2], []), + {ok, []} = emqttd:run_hooks(unknown_hook, [], []), + + emqttd:hook(foreach_hook, fun ?MODULE:hook_fun6/2, [initArg]), + emqttd:hook(foreach_hook, fun ?MODULE:hook_fun7/2, [initArg]), + emqttd:hook(foreach_hook, fun ?MODULE:hook_fun8/2, [initArg]), + stop = emqttd:run_hooks(foreach_hook, [arg]). hook_fun1([]) -> ok. hook_fun2([]) -> {ok, []}. @@ -345,6 +350,10 @@ hook_fun3(arg1, arg2, _Acc, init) -> ok. hook_fun4(arg1, arg2, Acc, init) -> {ok, [r2 | Acc]}. hook_fun5(arg1, arg2, Acc, init) -> {stop, [r3 | Acc]}. +hook_fun6(arg, initArg) -> ok. +hook_fun7(arg, initArg) -> any. +hook_fun8(arg, initArg) -> stop. + %%-------------------------------------------------------------------- %% HTTP Request Test %%--------------------------------------------------------------------