%% Filter out "*_tests" modules so eunit won't doubly run them and

    %% so cover only calculates coverage on production code.

    BeamFiles = [N || N <- rebar_utils:beams(?EUNIT_DIR), 

                      string:str(N, "_tests.beam") =:= 0],

    perform_cover(Config, Modules),

    cover_analyze(Config, BeamFiles).

cover_analyze(_Config, Modules) ->

    Suite = list_to_atom(rebar_config:get_global(suite, "")),

    FilteredModules = [M || M <- Modules, M =/= Suite],

    Coverage = [cover_analyze_mod(M) || M <- FilteredModules],

%% -*- tab-width: 4;erlang-indent-level: 4;indent-tabs-mode: nil -*-

%% ex: ts=4 sw=4 et

%% -------------------------------------------------------------------

%%

%% rebar: Erlang Build Tools

%%

%% Copyright (c) 2009, 2010 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.

%% -------------------------------------------------------------------

%% @author Chris Bernard <cebernard@gmail.com>

%% @doc This tests functionality provided by the rebar command 'eunit'.

%% @copyright 2009, 2010 Dave Smith

%% -------------------------------------------------------------------

-module(rebar_eunit_tests).

-compile(export_all).

-include_lib("eunit/include/eunit.hrl").

%% Assuming this test is run inside the rebar 'eunit'

%% command, the current working directory will be '.eunit'

-define(REBAR_SCRIPT, "../rebar").

-define(TMP_DIR, "tmp_eunit/").

%% ====================================================================

%% Rebar EUnit and Cover Tests

%% ====================================================================

eunit_test_() ->

    {"Ensure EUnit runs with tests in a 'test' dir and no defined suite",

     setup, fun() -> setup_basic_project(), rebar("-v eunit") end,

     fun teardown/1,

     fun(RebarOut) ->

             [{"Tests in 'test' directory are found and run",

               ?_assert(string:str(RebarOut, "myapp_mymod_tests:") =/= 0)},

              {"Tests in 'src' directory are found and run",

               ?_assert(string:str(RebarOut, "myapp_mymod:") =/= 0)},

              {"Tests are only run once",

               ?_assert(string:str(RebarOut, "All 2 tests passed") =/= 0)}]

     end}. 

cover_test_() ->

    {"Ensure Cover runs with tests in a test dir and no defined suite",

     setup, fun() -> setup_cover_project(), rebar("-v eunit") end,

     fun teardown/1,

     [{"All cover reports are generated",

       assert_files_in("the temporary eunit directory",

                       expected_cover_generated_files())},

      {"Only production modules get coverage reports",

       assert_files_not_in("the temporary eunit directory",

                           [".eunit/myapp_mymod_tests.COVER.html"])}]}.

cover_with_suite_test_() ->

    {"Ensure Cover runs with Tests in a test dir and a test suite",

     setup,

     fun() ->

             setup_cover_project_with_suite(),

             rebar("-v eunit suite=mysuite")

     end,

     fun teardown/1,

     [{"All cover reports are generated",

       assert_files_in("the temporary eunit directory",

                       expected_cover_generated_files())},

      {"Only production modules get coverage reports",

       assert_files_not_in("the temporary eunit directory",

                           [".eunit/myapp_mymod_tests.COVER.html",

                            ".eunit/mysuite"])}]}.             

expected_cover_generated_files() ->

    [".eunit/index.html",

     ".eunit/myapp_app.COVER.html",

     ".eunit/myapp_mymod.COVER.html",

     ".eunit/myapp_sup.COVER.html"].

%% ====================================================================

%% Environment and Setup Tests

%% ====================================================================

environment_test_() ->

    {"Sanity check the testing environment",

     setup, fun make_tmp_dir/0, fun remove_tmp_dir/1,

     [{"Ensure a test project can be created", 

       ?_assert(filelib:is_dir(?TMP_DIR))},

      {"Ensure the rebar script can be found, copied, and run",

       [?_assert(filelib:is_file(?REBAR_SCRIPT)),

        fun assert_rebar_runs/0]}]}.

assert_rebar_runs() ->

    prepare_rebar_script(),

    ?assert(string:str(os:cmd("./" ++ ?TMP_DIR ++ "rebar"), "Usage: rebar") =/= 0).

basic_setup_test_() ->

    {"Create a simple project with a 'test' directory, a test, and a module",

     setup, fun setup_basic_project/0, fun teardown/1,

     %% Test the setup function

     assert_dirs_in("Basic Project",

                    ["src", "ebin", "test"]) ++

     assert_files_in("Basic Project",

                     ["test/myapp_mymod_tests.erl", "src/myapp_mymod.erl"])}.

%% ====================================================================

%% Setup and Teardown

%% ====================================================================

-define(myapp_mymod, 

        ["-module(myapp_mymod).\n",

         "-export([myfunc/0]).\n",

         "-include_lib(\"eunit/include/eunit.hrl\").\n",

         "myfunc() -> ok.\n",

         "myprivate_test() -> ?assert(true).\n"]).

-define(myapp_mymod_tests, 

        ["-module(myapp_mymod_tests).\n",

         "-compile([export_all]).\n",

         "-include_lib(\"eunit/include/eunit.hrl\").\n",

         "myfunc_test() -> ?assertMatch(ok, myapp_mymod:myfunc()).\n"]).

-define(mysuite,

        ["-module(mysuite).\n",

         "-export([all_test_/0]).\n",

         "-include_lib(\"eunit/include/eunit.hrl\").\n",

         "all_test_() -> [myapp_mymod_defined_in_mysuite_tests].\n"]).

-define(myapp_mymod_defined_in_mysuite_tests, 

        ["-module(myapp_mymod_defined_in_mysuite_tests).\n",

         "-compile([export_all]).\n",

         "-include_lib(\"eunit/include/eunit.hrl\").\n",

         "myfunc_test() -> ?assertMatch(ok, myapp_mymod:myfunc()).\n"]).

make_tmp_dir() ->

    file:make_dir(?TMP_DIR).

setup_environment() ->

    make_tmp_dir(),

    prepare_rebar_script(),

    file:set_cwd(?TMP_DIR).

setup_basic_project() ->

    setup_environment(),

    rebar("create-app appid=myapp"),

    file:make_dir("test"),

    file:write_file("test/myapp_mymod_tests.erl", ?myapp_mymod_tests),

    file:write_file("src/myapp_mymod.erl", ?myapp_mymod).

setup_cover_project() ->

    setup_basic_project(),

    file:write_file("rebar.config", "{cover_enabled, true}.\n").

setup_cover_project_with_suite() ->

    setup_cover_project(),

    file:write_file("test/mysuite.erl", ?mysuite),

    file:write_file("test/myapp_mymod_defined_in_mysuite_tests.erl",

                    ?myapp_mymod_defined_in_mysuite_tests).

teardown(_) ->

    file:set_cwd(".."),

    remove_tmp_dir(),

    ok.

remove_tmp_dir() ->

    remove_tmp_dir(arg_for_eunit).

remove_tmp_dir(_) ->    

    case os:type() of

        {unix, _} ->

            os:cmd("rm -rf " ++ ?TMP_DIR ++ " 2>/dev/null");

        {win32, _} ->

            %% os:cmd("rmdir /S /Q " ++ ?TMP_DIR ++ " 2>NUL")

            exit("Windows is not supported yet.")

    end.

%% ====================================================================

%% Helper Functions

%% ====================================================================

prepare_rebar_script() ->

    {ok, _} = file:copy(?REBAR_SCRIPT, ?TMP_DIR ++ "rebar"),

    [] = os:cmd("chmod u+x " ++ ?TMP_DIR ++ "rebar").

rebar() ->

    rebar([]).

rebar(Args) when is_list(Args) ->

    Out = os:cmd("./rebar " ++ Args), 

    %?debugMsg("**** Begin"), ?debugMsg(Out), ?debugMsg("**** End"),

    Out.

assert_dirs_in(Name, [Dir|T]) ->

    [{Name ++ " has directory: " ++ Dir, ?_assert(filelib:is_dir(Dir))} |

     assert_dirs_in(Name, T)];                                                                         

assert_dirs_in(_, []) -> [].

assert_files_in(Name, [File|T]) ->

    [{Name ++ " has file: " ++ File, ?_assert(filelib:is_file(File))} |

     assert_files_in(Name, T)];   

assert_files_in(_, []) -> [].

assert_files_not_in(Name, [File|T]) ->

    [{Name ++ " does not have file: " ++ File, ?_assertNot(filelib:is_file(File))} |

     assert_files_not_in(Name, T)];   

assert_files_not_in(_, []) -> [].