Check if ExUnit is available before setting up after suite calls

Author: agundy

lib/wallaby/session_store.ex

@@ -36,16 +36,18 @@ defmodule Wallaby.SessionStore do
     Process.flag(:trap_exit, true)
     tid = :ets.new(name, opts)
 
-    Application.ensure_all_started(:ex_unit)
-
-    ExUnit.after_suite(fn _ ->
-      try do
-        :ets.tab2list(tid)
-        |> Enum.each(&delete_sessions/1)
-      rescue
-        _ -> nil
-      end
-    end)
+    if Code.ensure_loaded?(ExUnit) do
+      Application.ensure_all_started(:ex_unit)
+
+      ExUnit.after_suite(fn _ ->
+        try do
+          :ets.tab2list(tid)
+          |> Enum.each(&delete_sessions/1)
+        rescue
+          _ -> nil
+        end
+      end)
+    end
 
     {:ok, %{ets_table: tid}}
   end