Fix tests that check for init exceptions

Author: lucaong

test/cubdb/store/file_test.exs

@@ -1,6 +1,7 @@
 defmodule CubDB.Store.FileTest do
   use ExUnit.Case, async: true
   use CubDB.StoreExamples
+  require TestHelper
 
   import CubDB.Btree, only: [header: 1]
 
@@ -90,7 +91,9 @@ defmodule CubDB.Store.FileTest do
   test "returns error if the same file is already in use by another store", %{
     file_path: file_path
   } do
-    assert {:error, {%ArgumentError{message: message}, _}} = CubDB.Store.File.create(file_path)
-    assert message == "file \"#{file_path}\" is already in use by another CubDB.Store.File"
+    TestHelper.trapping_exit do
+      assert {:error, {%ArgumentError{message: message}, _}} = CubDB.Store.File.create(file_path)
+      assert message == "file \"#{file_path}\" is already in use by another CubDB.Store.File"
+    end
   end
 end

test/cubdb_test.exs

@@ -1,5 +1,6 @@
 defmodule CubDBTest do
   use ExUnit.Case, async: true
+  require TestHelper
   doctest CubDB
 
   setup do
@@ -40,7 +41,9 @@ defmodule CubDBTest do
   end
 
   test "start_link/1 returns error if options are invalid", %{tmp_dir: tmp_dir} do
-    assert {:error, _} = CubDB.start_link(data_dir: tmp_dir, auto_compact: "maybe")
+    TestHelper.trapping_exit do
+      assert {:error, _} = CubDB.start_link(data_dir: tmp_dir, auto_compact: "maybe")
+    end
   end
 
   test "start_link/1 returns error if data_dir is missing" do

test/test_helper.exs

@@ -19,6 +19,17 @@ defmodule TestHelper do
     |> CubDB.Btree.commit()
   end
 
+  defmacro trapping_exit(do: block) do
+    quote do
+      original = Process.flag(:trap_exit, true)
+      try do
+        unquote(block)
+      after
+        Process.flag(:trap_exit, original)
+      end
+    end
+  end
+
   defmodule Btree.Utils do
     @moduledoc false