github/rebar
1
0
Fork 0
mirror of https://github.com/correl/rebar.git synced 2024-11-27 11:09:55 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix up code path for eunit so that parse transforms in eunit tests work properly

Browse source
This commit is contained in:
Dave Smith 2010-06-28 15:19:19 -06:00
parent f75bcdc7d5
commit ef175652e4
1 changed files with 24 additions and 29 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
src/rebar_eunit.erl
View file

@ -77,6 +77,13 @@ eunit(Config, AppFile) ->
%% Make sure ?EUNIT_DIR/ directory exists (tack on dummy module)
ok = filelib:ensure_dir(?EUNIT_DIR ++ "/foo"),
%% Setup code path prior to compilation so that parse_transforms and the like
%% work properly. Also, be sure to add ebin_dir() to the END of the code path
%% so that we don't have to jump through hoops to access the .app file
CodePath = code:get_path(),
true = code:add_patha(eunit_dir()),
true = code:add_pathz(ebin_dir()),
%% Obtain all the test modules for inclusion in the compile stage.
%% Notice: this could also be achieved with the following rebar.config option:
%% {eunit_compile_opts, [{src_dirs, ["test"]}]}
@ -112,9 +119,11 @@ eunit(Config, AppFile) ->
ok ->
ok;
_ ->
?CONSOLE("One or more eunit tests failed.~n", []),
?FAIL
?ABORT("One or more eunit tests failed.~n", [])
end,
%% Restore code path
code:set_path(CodePath),
ok.
clean(_Config, _File) ->
@ -127,15 +136,25 @@ clean(_Config, _File) ->
eunit_dir() ->
filename:join(rebar_utils:get_cwd(), ?EUNIT_DIR).
ebin_dir() ->
filename:join(rebar_utils:get_cwd(), "ebin").
perform_eunit(Config, Modules) ->
%% suite defined, so only specify the module that relates to the
%% suite (if any)
Suite = rebar_config:get_global(suite, undefined),
EunitOpts = get_eunit_opts(Config),
OrigEnv = set_proc_env(),
%% Move down into ?EUNIT_DIR while we run tests so any generated files
%% are created there (versus in the source dir)
Cwd = rebar_utils:get_cwd(),
file:set_cwd(?EUNIT_DIR),
EunitResult = perform_eunit(EunitOpts, Modules, Suite),
restore_proc_env(OrigEnv),
%% Return to original working dir
file:set_cwd(Cwd),
EunitResult.
perform_eunit(EunitOpts, Modules, undefined) ->
@ -143,24 +162,6 @@ perform_eunit(EunitOpts, Modules, undefined) ->
perform_eunit(EunitOpts, _Modules, Suite) ->
(catch eunit:test(list_to_atom(Suite), EunitOpts)).
set_proc_env() ->
%% Save current code path and then prefix ?EUNIT_DIR on it so that our modules
%% are found there
CodePath = code:get_path(),
true = code:add_patha(eunit_dir()),
%% Move down into ?EUNIT_DIR while we run tests so any generated files
%% are created there (versus in the source dir)
Cwd = rebar_utils:get_cwd(),
file:set_cwd(?EUNIT_DIR),
{CodePath, Cwd}.
restore_proc_env({CodePath, Cwd}) ->
%% Return to original working dir
file:set_cwd(Cwd),
%% Restore code path
true = code:set_path(CodePath).
get_eunit_opts(Config) ->
%% Enable verbose in eunit if so requested..
BaseOpts = case rebar_config:is_verbose() of
@ -271,19 +272,13 @@ cover_analyze_mod(Module) ->
is_eunitized(Mod) ->
has_eunit_test_fun(Mod) andalso
has_eunit_header(Mod).
has_header(Mod, "include/eunit.hrl").
has_eunit_test_fun(Mod) ->
length([F || {exports, Funs} <- Mod:module_info(),
{F, 0} <- Funs,
F == test]) =/= 0.
has_eunit_header(Mod) ->
OrigEnv = set_proc_env(),
try has_header(Mod, "include/eunit.hrl")
after restore_proc_env(OrigEnv)
end.
has_header(Mod, Header) ->
{ok, {_, [{abstract_code, {_, AC}}]}} = beam_lib:chunks(Mod, [abstract_code]),
length([F || {attribute, 1, file, {F, 1}} <- AC,