Added rebar_ct to support common_test targets test, int_test and perf_test.

1 files changed, 196 insertions, 0 deletions

diff --git a/src/rebar_ct.erl b/src/rebar_ct.erl
new file mode 100644
index 0000000..2cb2f8f
--- /dev/null
+++ b/src/rebar_ct.erl
@@ -0,0 +1,196 @@
+%% -------------------------------------------------------------------
+%%
+%% rebar: Erlang Build Tools
+%%
+%% Copyright (c) 2009 Dave Smith (dizzyd@dizzyd.com)
+%%
+%% Permission is hereby granted, free of charge, to any person obtaining a copy
+%% of this software and associated documentation files (the "Software"), to deal
+%% in the Software without restriction, including without limitation the rights
+%% to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+%% copies of the Software, and to permit persons to whom the Software is
+%% furnished to do so, subject to the following conditions:
+%%
+%% The above copyright notice and this permission notice shall be included in
+%% all copies or substantial portions of the Software.
+%%
+%% THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+%% IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+%% FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+%% AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+%% LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+%% OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+%% THE SOFTWARE.
+%% -------------------------------------------------------------------
+%% 
+%% Targets:
+%% test - runs common test suites in ./test
+%% int_test - runs suites in ./int_test
+%% perf_test - runs suites inm ./perf_test
+%%
+%% Global options:
+%% verbose=1 - show output from the common_test run as it goes
+%% suite="foo"" - runs <test>/foo_SUITE
+%% case="mycase" - runs individual test case foo_SUITE:mycase
+%% -------------------------------------------------------------------
+-module(rebar_ct).
+
+-export([test/2,
+         int_test/2,
+         perf_test/2]).
+
+-compile([export_all]).
+
+-include("rebar.hrl").
+
+%% ===================================================================
+%% Public API
+%% ===================================================================
+
+test(Config, File) ->
+    run_test_if_present("test", Config, File).
+
+int_test(Config, File) ->
+    run_test_if_present("int_test", Config, File).
+
+perf_test(Config, File) ->
+    run_test_if_present("perf_test", Config, File).
+
+%% ===================================================================
+%% Internal functions
+%% ===================================================================
+run_test_if_present(TestDir, Config, File) ->
+    case filelib:is_dir(TestDir) of
+        false ->
+	    ?WARN("~s directory not present - skipping\n", [TestDir]),
+            ok;
+        true ->
+	    run_test(TestDir, Config, File)
+    end.
+
+run_test(TestDir, Config, _File) ->
+    {Cmd, RawLog} = make_cmd(TestDir, Config),
+    clear_log(RawLog),
+    case rebar_config:get_global(verbose, "0") of
+	"0" ->
+	    Output = " >> " ++ RawLog ++ " 2>&1";
+	_ ->
+	    Output = " 2>&1 | tee -a " ++ RawLog
+    end,
+    
+    case rebar_utils:sh(Cmd ++ Output, [{"TESTDIR", TestDir}]) of
+	ok ->
+	    check_log(RawLog);
+	{error, _Rc} ->
+	    show_log(RawLog),
+	    ?ERROR("Executing tests failed.\n", []),
+	    ?FAIL
+    end.
+
+
+clear_log(RawLog) ->
+    case filelib:ensure_dir("logs/index.html") of
+	ok ->
+	    NowStr = rebar_utils:now_str(),
+	    LogHeader = "--- Test run on " ++ NowStr ++ " ---\n",
+	    ok = file:write_file(RawLog, LogHeader);
+	{error, Reason} ->
+	    ?ERROR("Could not create log dir - ~p\n", [Reason]),
+	    ?FAIL
+    end.    
+
+%% calling ct with erl does not return non-zero on failure - have to check
+%% log results
+check_log(RawLog) ->
+    Msg = os:cmd("grep 'TEST COMPLETE' " ++ RawLog),
+    case string:str(Msg, ", 0 failed") of
+	0 ->
+	    show_log(RawLog),
+	    ?ERROR("One or more tests failed\n",[]),
+	    ?FAIL;
+	_ ->
+	    ?CONSOLE("DONE. ~s\n", [Msg])
+    end.
+		
+%% Show the log if it hasn't already been shown because verbose was on
+show_log(RawLog) ->
+    ?CONSOLE("Showing log\n", []),
+    case rebar_config:get_global(verbose, "0") of
+	"0" ->
+	    {ok, Contents} = file:read_file(RawLog),
+	    ?CONSOLE("~s", [Contents]);
+	_ ->
+	    ok
+    end.
+
+make_cmd(TestDir, Config) ->
+    {ok, Cwd} = file:get_cwd(),
+    LogDir = filename:join(Cwd, "logs"),
+    Ebin = filename:join(Cwd, "ebin"),
+    IncludeDir = filename:join(Cwd, "include"),
+    case filelib:is_dir(IncludeDir) of
+	true ->
+	    Include = " -I \"" ++ IncludeDir ++ "\"";
+	false ->
+	    Include = ""
+    end,
+
+    Cmd = lists:flatten(io_lib:format("erl " % should we expand ERL_PATH?
+				      " -noshell -pa \"~s\" ~s"
+				      " -s ct_run script_start -s erlang halt"
+				      " -name test@~s"
+				      " -logdir \"~s\""
+				      " -env TEST_DIR \"~s\"",
+				      [Ebin,
+				       Include,
+				       net_adm:localhost(), 
+				       LogDir,
+				       filename:join(Cwd, TestDir)])) ++
+	get_ct_config_file(TestDir) ++
+	get_config_file(TestDir) ++
+	get_suite(TestDir) ++
+	get_case(),
+    RawLog = filename:join(LogDir, "raw.log"),
+    {Cmd, RawLog}.
+     
+
+get_ct_config_file(TestDir) ->
+    Config = filename:join(TestDir, "test.config"),
+    case filelib:is_regular(Config) of
+        false ->
+            " ";
+        true ->
+            " -ct_config " ++ Config
+    end.
+
+get_config_file(TestDir) ->
+    Config = filename:join(TestDir, "app.config"),
+    case filelib:is_regular(Config) of
+        false ->
+            " ";
+        true ->
+            " -config " ++ Config
+    end.
+
+get_suite(TestDir) ->
+    case rebar_config:get_global(suite, undefined) of
+	undefined ->
+	    " -dir " ++ TestDir;
+	Suite ->
+	    Filename = filename:join(TestDir, Suite ++ "_SUITE.erl"),
+	    case filelib:is_regular(Filename) of
+		false ->
+		    ?ERROR("Suite ~s not found\n", [Suite]),
+		    ?FAIL;
+		true ->
+		    " -suite " ++ Filename
+	    end
+    end.
+
+get_case() ->
+    case rebar_config:get_global('case', undefined) of
+	undefined ->
+	    "";
+	Case ->
+	    " -case " ++ Case
+    end.