Merge pull request #18 from sofakingworld/manual_file_tracking

Manual file tracking function

Author: danhper

lib/temp.ex

@@ -122,6 +122,17 @@ defmodule Temp do
     end
   end
 
+  @doc """
+  Function allows add file to tracker, which will be removed on Temp.cleanup
+  """
+  @spec track_file(any) :: {:error, :tracker_not_found} | {:ok, Path.t}
+  def track_file(path, tracker \\ get_tracker()) do
+    case is_nil(tracker) do
+      true -> {:error, :tracker_not_found}
+      false -> {:ok, register_path(tracker, path)}
+    end
+  end
+
   @doc """
   Same as `open/1`, but raises an exception on failure.
   """

test/temp_test.exs

@@ -120,4 +120,19 @@ defmodule TempTest do
     assert_receive {:cleaned, 1}
     refute File.exists?(dir)
   end
+
+  test :track_file do
+    assert {:ok, tracker} = Temp.track
+
+    path_of_tmp_file = "test/tmp_file_created_by_programmer"
+    File.write!(path_of_tmp_file, "Make Elixir Gr8 Again")
+
+    assert File.exists?(path_of_tmp_file)
+
+    Temp.track_file(path_of_tmp_file, tracker)
+
+    Temp.cleanup(tracker)
+
+    refute File.exists?(path_of_tmp_file)
+  end
 end