1 files changed, 55 insertions, 0 deletions
diff --git a/learn-you-some-erlang/tester.erl b/learn-you-some-erlang/tester.erl
new file mode 100644
index 0000000..f1d1b1c
--- /dev/null
+++ b/learn-you-some-erlang/tester.erl
@@ -0,0 +1,55 @@
+-module(tester).
+-export([dir/0, dir/2]).
+-define(EXT, ".erl"). % file extension to look for
+-define(MODS, "./").
+-define(TESTS, "./tests/").
+
+%% scans both a module directory and a test directory, compiles the
+%% modules inside and then call for tests to be ran.
+%%
+%% usage:
+%%     tester:dir("./","./tests/").
+dir() -> dir(?MODS, ?TESTS).
+dir(ModulePath, TestPath) ->
+    ModuleList = module_list(ModulePath),
+    TestList = module_list(TestPath),
+    [compile(ModulePath++X) || X <- ModuleList],
+    [compile(TestPath++X) || X <- TestList],
+    test_all(TestList),
+    warnings(),
+    cleanup(ModuleList),
+    cleanup(TestList),
+    ok.
+
+%% assumes pre-compiled modules
+test_all(FileList) ->
+    Split = [lists:nth(1, string:tokens(File, ".")) || File <- FileList],
+    [eunit:test(list_to_existing_atom(F), [verbose]) || F <- Split].
+   % [(list_to_existing_atom(F)):test() || F <- Split].
+
+cleanup(Files) ->
+    [file:delete(lists:nth(1, string:tokens(F, "."))++".beam") || F <- Files].
+
+%% get module .erl file names from a directory
+module_list(Path) ->
+    SameExt = fun(File) -> get_ext(File) =:= ?EXT end,
+    {ok, Files} = file:list_dir(Path),
+    lists:filter(SameExt, Files).
+
+%% find the extension of a file (length is taken from the ?EXT macro).
+get_ext(Str) ->
+    lists:reverse(string:sub_string(lists:reverse(Str), 1, length(?EXT))).
+
+compile(FileName) ->
+    compile:file(FileName, [report, verbose, export_all]).
+
+warnings() ->
+    Warns = [{Mod, get_warnings(Mod)} || {Mod,_Path} <- code:all_loaded(),
+                                         has_warnings(Mod)],
+    io:format("These need to be tested better: ~n\t~p~n", [Warns]).
+
+has_warnings(Mod) ->
+    is_list(get_warnings(Mod)).
+
+get_warnings(Mod) ->
+    proplists:get_value(test_warnings, Mod:module_info(attributes)).