Mock mongoose_dist_blocker in mongoose_cleanup_SUITE

esl · Mar 5, 2024 · 0777597 · 0777597
1 parent 665c903
commit 0777597
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/test/mongoose_cleanup_SUITE.erl b/test/mongoose_cleanup_SUITE.erl
@@ -124,10 +124,10 @@ opts() ->
       {auth, ?HOST} => config_parser_helper:extra_auth(),
       {modules, ?HOST} => #{}}.
 
-meck_mods(bosh) -> [exometer, mod_bosh_socket];
-meck_mods(s2s) -> [exometer, mongoose_bin];
-meck_mods(component) -> [exometer];
-meck_mods(_) -> [exometer, ejabberd_sm, ejabberd_local].
+meck_mods(bosh) -> [exometer, mod_bosh_socket, mongoose_dist_blocker];
+meck_mods(s2s) -> [exometer, mongoose_bin, mongoose_dist_blocker];
+meck_mods(component) -> [exometer, mongoose_dist_blocker];
+meck_mods(_) -> [exometer, ejabberd_sm, ejabberd_local, mongoose_dist_blocker].
 
 %% -----------------------------------------------------
 %% Tests
@@ -306,6 +306,11 @@ setup_meck([mod_bosh_socket | R]) ->
     meck:new(mod_bosh_socket, [passthrough]),
     meck:expect(mod_bosh_socket, start_supervisor, fun() -> {ok, self()} end),
     setup_meck(R);
+setup_meck([mongoose_dist_blocker | R]) ->
+    meck:new(mongoose_dist_blocker, [passthrough]),
+    meck:expect(mongoose_dist_blocker, add_cleaner, fun(_Pid) -> ok end),
+    meck:expect(mongoose_dist_blocker, cleaning_done, fun(_Pid, _Node) -> ok end),
+    setup_meck(R);
 setup_meck([]) ->
     ok.