Remove cover data from original module

edgurgel · Jun 23, 2024 · cc8d943 · cc8d943
1 parent ea12b00
commit cc8d943
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/lib/mimic/module.ex b/lib/mimic/module.ex
@@ -11,6 +11,7 @@ defmodule Mimic.Module do
     :code.delete(module)
     :code.purge(original(module))
     :code.delete(original(module))
+    :cover.reset(original(module))
     :ok
   end
 

diff --git a/test/support/test_cover.ex b/test/support/test_cover.ex
@@ -15,17 +15,16 @@ defmodule Mimic.TestCover do
   defp execute do
     {:result, results, _fail} = :cover.analyse(:calls, :function)
 
-    results =
-      Enum.filter(results, fn
-        {{Calculator, _, _}, _} -> true
-        {{NoStubs, _, _}, _} -> true
+    mimic_module_cover =
+      Enum.any?(results, fn
+        {{Calculator.Mimic.Original.Module, _, _}, _} -> true
         _ -> false
       end)
 
     expected =
       {{Calculator, :add, 2}, 5} in results &&
         {{Calculator, :mult, 2}, 5} in results &&
-        {{NoStubs, :add, 2}, 2} in results
+        {{NoStubs, :add, 2}, 2} in results && !mimic_module_cover
 
     unless expected do
       IO.puts("Cover results are incorrect!")