Decorate all functions inside behaviour-implementation module

lib/hammox.ex

@@ -87,6 +87,34 @@ defmodule Hammox do
   """
   defdelegate verify_on_exit!(context \\ %{}), to: Mox
 
+  @doc since: "0.1.0"
+  @doc """
+  Same as `protect/3`, but decorate all public functions inside the module.
+
+  Example:
+
+  ```elixir
+  defmodule Calculator do
+    @callback add(integer(), integer()) :: integer()
+    def add(a, b), do: a + b
+    @callback add(integer(), integer(), integer()) :: integer()
+    def add(a, b, c), do: a + b + c
+    @callback multiply(integer(), integer()) :: integer()
+    def multiply(a, b), do: a * b
+  end
+
+  %{
+    add_2: add_2,
+    add_3: add_3,
+    multiply_2: multiply_2
+  } = Hammox.protect(Calculator)
+  """
+  @spec protect(module_name :: module()) :: map()
+  def protect(module_name) when is_atom(module_name) do
+    funs = module_name.__info__(:functions)
+    protect(module_name, module_name, funs)
+  end
+
   @doc since: "0.1.0"
   @doc """
   Takes the function provided by a module, function, arity tuple and

test/hammox_test.exs

@@ -5,6 +5,13 @@ defmodule HammoxTest do
 
   defmock(TestMock, for: Hammox.Test.Behaviour)
 
+  describe "protect/1" do
+    test "decorate all functions inside the module" do
+      assert %{other_foo_0: _, other_foo_1: _, foo_0: _} =
+        Hammox.protect(Hammox.Test.BehaviourImplementation)
+    end
+  end
+
   describe "protect/2" do
     test "returns function protected from contract errors" do
       fun = Hammox.protect({Hammox.Test.SmallImplementation, :foo, 0}, Hammox.Test.SmallBehaviour)