fix error in cleanup, bump version

README.md

@@ -42,7 +42,7 @@ Efx follows the following principles:
 To use `Efx` in your project, add this to your dependencies in `mix.ex`:
 
 ```elixir
-{:efx, "~> 0.2.6"}
+{:efx, "~> 0.2.7"}
 ```
 
 If you want to have proper formatting of the `Efx.defeffect/2` macro, you can add

lib/efx_case.ex

@@ -208,11 +208,16 @@ defmodule EfxCase do
         Internal.init(pid)
 
         on_exit(fn ->
-          Internal.verify_mocks!(pid)
+          res = Internal.verify_mocks(pid)
 
           unless unquote(async?) do
             MockState.clean_after_test()
           end
+
+          case res do
+            :ok -> :ok
+            {:error, error} -> raise error
+          end
         end)
       end
 

lib/efx_case/internal.ex

@@ -1,6 +1,13 @@
 defmodule EfxCase.Internal do
   alias EfxCase.MockState
 
+  def verify_mocks(pid) do
+    verify_mocks!(pid)
+    :ok
+  rescue
+    e -> {:error, e}
+  end
+
   def verify_mocks!(pid) do
     MockState.verify_called!(pid)
   end

lib/efx_case/mock_state.ex

@@ -13,8 +13,12 @@ defmodule EfxCase.MockState do
   invocations as well as store mocks as global when a test is flagged as
   async.
   """
+  alias ExUnit.Assertions
   alias EfxCase.Mock
   alias EfxCase.Mock.MockedFun
+
+  require ExUnit.Assertions
+
   use Agent
 
   @typedoc """
@@ -44,7 +48,9 @@ defmodule EfxCase.MockState do
             mock
 
           {:error, :function_not_in_mock} ->
-            raise "Not matching function found for #{behaviour}: #{fun_identifier}/#{arity}"
+            Assertions.flunk(
+              "Not matching function found for #{behaviour}: #{fun_identifier}/#{arity}"
+            )
         end
 
       new_pid_state = Map.put(pid_state, behaviour, new_mock)
@@ -70,7 +76,9 @@ defmodule EfxCase.MockState do
           {ret, Map.put(state, scope, new_pid_state)}
         else
           {:error, _err} ->
-            raise "no mock found for #{inspect(behaviour)}/#{inspect(fun_identifier)} in scope #{inspect(pid)} with state #{inspect(state)}"
+            Assertions.flunk(
+              "no mock found for #{inspect(behaviour)}/#{inspect(fun_identifier)} in scope #{inspect(pid)} with state #{inspect(state)}"
+            )
         end
       end)
 
@@ -115,7 +123,9 @@ defmodule EfxCase.MockState do
         unsatisfied = Mock.get_unsatisfied(mock)
 
         unless Enum.empty?(unsatisfied) do
-          raise "expectations for #{inspect(behaviour)} were not meat: \n #{parse(unsatisfied)}"
+          Assertions.flunk(
+            "expectations for #{inspect(behaviour)} were not met: \n #{parse(unsatisfied)}"
+          )
         end
       end)
     end

mix.exs

@@ -1,7 +1,7 @@
 defmodule Efx.MixProject do
   use Mix.Project
 
-  @version "0.2.6"
+  @version "0.2.7"
   @github_page "https://github.com/bravobike/efx"
 
   def project do
@@ -56,7 +56,7 @@ defmodule Efx.MixProject do
 
   defp package do
     [
-      files: ~w(mix.exs README.md lib),
+      files: ~w(mix.exs README.md lib .formatter.exs),
       licenses: ["Apache-2.0"],
       links: %{
         "GitHub" => @github_page

test/efx_case/mock_state_test.exs

@@ -42,7 +42,7 @@ defmodule EfxCase.MockStateTest do
     end
 
     test "raises if if unsatisfied" do
-      assert_raise(RuntimeError, fn ->
+      assert_raise(ExUnit.AssertionError, fn ->
         add_fun(num_expected_calls: 1)
         MockState.verify_called!(self())
       end)