diff options
| -rw-r--r-- | src/rebar_file_utils.erl | 32 | ||||
| -rw-r--r-- | test/rebar_file_utils_SUITE.erl | 86 |
2 files changed, 117 insertions, 1 deletions
diff --git a/src/rebar_file_utils.erl b/src/rebar_file_utils.erl index 0fc1403..8fe8965 100644 --- a/src/rebar_file_utils.erl +++ b/src/rebar_file_utils.erl @@ -30,7 +30,10 @@ cp_r/2, mv/2, delete_each/1, - write_file_if_contents_differ/2]). + write_file_if_contents_differ/2, + system_tmpdir/0, + system_tmpdir/1, + reset_dir/1]). -include("rebar.hrl"). @@ -123,6 +126,33 @@ write_file_if_contents_differ(Filename, Bytes) -> file:write_file(Filename, ToWrite) end. +%% returns an os appropriate tmpdir given a path +-spec system_tmpdir() -> file:filename(). +-spec system_tmpdir(PathComponents) -> file:filename() when + PathComponents :: [file:name()]. + +system_tmpdir() -> system_tmpdir([]). +system_tmpdir(PathComponents) -> + Tmp = case erlang:system_info(system_architecture) of + "win32" -> + "./tmp"; + _SysArch -> + "/tmp" + end, + filename:join([Tmp|PathComponents]). + +%% recursively removes a directory and then recreates the same +%% directory but empty +-spec reset_dir(Path) -> ok | {error, Reason} when + Path :: file:name(), + Reason :: file:posix(). + +reset_dir(Path) -> + %% delete the directory if it exists + _ = ec_file:remove(Path, [recursive]), + %% recreate the directory + filelib:ensure_dir(filename:join([Path, "dummy.beam"])). + %% =================================================================== %% Internal functions %% =================================================================== diff --git a/test/rebar_file_utils_SUITE.erl b/test/rebar_file_utils_SUITE.erl new file mode 100644 index 0000000..67926de --- /dev/null +++ b/test/rebar_file_utils_SUITE.erl @@ -0,0 +1,86 @@ +-module(rebar_file_utils_SUITE). + +-export([all/0, + groups/0, + init_per_group/2, + end_per_group/2, + raw_tmpdir/1, + empty_tmpdir/1, + simple_tmpdir/1, + multi_tmpdir/1, + reset_nonexistent_dir/1, + reset_empty_dir/1, + reset_dir/1]). + +-include_lib("common_test/include/ct.hrl"). +-include_lib("eunit/include/eunit.hrl"). +-include_lib("kernel/include/file.hrl"). + + +all() -> + [{group, tmpdir}, + {group, reset_dir}]. + +groups() -> + [{tmpdir, [], [raw_tmpdir, empty_tmpdir, simple_tmpdir, multi_tmpdir]}, + {reset_dir, [], [reset_nonexistent_dir, reset_empty_dir, reset_dir]}]. + +init_per_group(reset_dir, Config) -> + TmpDir = rebar_file_utils:system_tmpdir(["rebar_file_utils_SUITE", "resetable"]), + [{tmpdir, TmpDir}|Config]; +init_per_group(_, Config) -> Config. +end_per_group(_, Config) -> Config. + +raw_tmpdir(_Config) -> + case rebar_file_utils:system_tmpdir() of + "/tmp" -> ok; + "./tmp" -> ok + end. + +empty_tmpdir(_Config) -> + case rebar_file_utils:system_tmpdir([]) of + "/tmp" -> ok; + "./tmp" -> ok + end. + +simple_tmpdir(_Config) -> + case rebar_file_utils:system_tmpdir(["test"]) of + "/tmp/test" -> ok; + "./tmp/test" -> ok + end. + +multi_tmpdir(_Config) -> + case rebar_file_utils:system_tmpdir(["a", "b", "c"]) of + "/tmp/a/b/c" -> ok; + "./tmp/a/b/c" -> ok + end. + +reset_nonexistent_dir(Config) -> + TmpDir = ?config(tmpdir, Config), + _ = ec_file:remove(TmpDir, [recursive]), + ?assertNot(filelib:is_dir(TmpDir)), + ok = rebar_file_utils:reset_dir(TmpDir), + ?assert(filelib:is_dir(TmpDir)), + {ok, []} = file:list_dir_all(TmpDir). + +reset_empty_dir(Config) -> + TmpDir = ?config(tmpdir, Config), + _ = ec_file:remove(TmpDir, [recursive]), + _ = filelib:ensure_dir(filename:join([TmpDir, "dummy.beam"])), + ?assert(filelib:is_dir(TmpDir)), + ok = rebar_file_utils:reset_dir(TmpDir), + ?assert(filelib:is_dir(TmpDir)), + {ok, []} = file:list_dir_all(TmpDir). + +reset_dir(Config) -> + TmpDir = ?config(tmpdir, Config), + _ = ec_file:remove(TmpDir, [recursive]), + _ = filelib:ensure_dir(filename:join([TmpDir, "dummy.beam"])), + ?assert(filelib:is_dir(TmpDir)), + lists:foreach(fun(Name) -> file:write_file(filename:join([TmpDir, Name]), <<>>) end, + ["a", "b", "c"]), + lists:foreach(fun(File) -> ?assert(filelib:is_file(filename:join([TmpDir, File]))) end, + ["a", "b", "c"]), + ok = rebar_file_utils:reset_dir(TmpDir), + ?assert(filelib:is_dir(TmpDir)), + {ok, []} = file:list_dir_all(TmpDir). |