Throw error when mocking sticky modules without unstick

eproxus · Jul 13, 2011 · 8810a71 · 8810a71
1 parent 66bbd7f
commit 8810a71
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/meck.erl b/src/meck.erl
@@ -289,8 +289,14 @@ init([Mod, Options]) ->
     Original = backup_original(Mod),
     process_flag(trap_exit, true),
     Expects = init_expects(Mod, Options),
-    meck_mod:compile_and_load_forms(to_forms(Mod, Expects)),
-    {ok, #state{mod = Mod, expects = Expects, original = Original, was_sticky = WasSticky}}.
+    try
+        meck_mod:compile_and_load_forms(to_forms(Mod, Expects)),
+        {ok, #state{mod = Mod, expects = Expects, original = Original,
+                    was_sticky = WasSticky}}
+    catch
+        exit:{error_loading_module, Mod, sticky_directory} ->
+            {stop, module_is_sticky}
+    end.
 
 %% @hidden
 handle_call({get_expect, Func, Arity}, _From, S) ->

diff --git a/test/meck_tests.erl b/test/meck_tests.erl
@@ -723,6 +723,9 @@ can_mock_sticky_module_not_yet_loaded_({Mod, _}) ->
     ?assertEqual(ok, meck:unload(Mod)),
     ?assert(code:is_sticky(Mod)).
 
+cannot_mock_sticky_module_without_unstick_({Mod, _}) ->
+    ?assertError(module_is_sticky, meck:new(Mod, [no_link])).
+
 can_mock_non_sticky_module_test() ->
     ?assertNot(code:is_sticky(meck_test_module)),
     ?assertEqual(ok, meck:new(meck_test_module, [unstick])),