From 3ee6c6bf8644a94ff2d8bfade97c9417b5510f31 Mon Sep 17 00:00:00 2001
From: Jon Meredith <jon@jonmeredith.com>
Date: Wed, 1 Sep 2010 21:54:34 -0600
Subject: [PATCH] Changed restore_original to check if cover compiled files
 were .beam rather than .erl and restore them appropriately.

---
 src/meck.erl | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/meck.erl b/src/meck.erl
index d3e7b3e9..4b29f53f 100644
--- a/src/meck.erl
+++ b/src/meck.erl
@@ -454,7 +454,12 @@ backup_original(Module) ->
 restore_original(_Mod, false) ->
     ok;
 restore_original(Mod, {File, Data, Options}) ->
-    {ok, Mod} = cover:compile(File, Options),
+    case filename:extension(File) of
+        ".erl" ->
+            {ok, Mod} = cover:compile_module(File, Options);
+        ".beam" ->
+            cover:compile_beam(File)
+    end,
     ok = cover:import(Data),
     file:delete(Data),
     ok.