Check if Mix is present before using Mix.env()

appunite · Oct 13, 2017 · 8d87a8a · 8d87a8a
1 parent 4b6901a
commit 8d87a8a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/mockery.ex b/lib/mockery.ex
@@ -35,7 +35,7 @@ defmodule Mockery do
     module | proxy
   def of(mod, opts \\ []) when is_atom(mod)
                           when is_binary(mod) do
-    env = opts[:env] || Mix.env
+    env = opts[:env] || mix_env()
 
     if env != :test do
       to_mod(mod)
@@ -107,4 +107,8 @@ defmodule Mockery do
 
     mod
   end
+
+  defp mix_env do
+    if Kernel.function_exported?(Mix, :env, 0), do: Mix.env(), else: :prod
+  end
 end