umgeher/leveled
0
Fork 0
Code Issues Pull requests Releases Activity Actions

Add unit test of archive files

Browse source
This commit is contained in:
Martin Sumner 2017-09-28 10:50:54 +01:00
parent 3950942da3
commit 0f5911ab70
1 changed files with 19 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/leveled_penciller.erl
View file

@ -900,14 +900,14 @@ start_from_file(PCLopts) ->
ok = archive_files(RootPath, FileList0), ok = archive_files(RootPath, FileList0),
{ok, State0}. {ok, State0}.
archive_files(RootPath, FileList) -> archive_files(RootPath, UsedFileList) ->
{ok, AllFiles} = file:list_dir(sst_rootpath(RootPath)), {ok, AllFiles} = file:list_dir(sst_rootpath(RootPath)),
FileCheckFun = FileCheckFun =
fun(FN, UnusedFiles) -> fun(FN, UnusedFiles) ->
FN0 = "./" ++ FN, FN0 = "./" ++ FN,
case filename:extension(FN0) of case filename:extension(FN0) of
?SST_FILEX -> ?SST_FILEX ->
case lists:member(FN0, FileList) of case lists:member(FN0, UsedFileList) of
true -> true ->
UnusedFiles; UnusedFiles;
false -> false ->
@ -1377,6 +1377,23 @@ clean_dir_test() ->
ok = clean_subdir(RootPath ++ "/test.bob"), ok = clean_subdir(RootPath ++ "/test.bob"),
ok = file:delete(RootPath ++ "/test.bob"). ok = file:delete(RootPath ++ "/test.bob").
archive_files_test() ->
RootPath = "../test/ledger",
SSTPath = sst_rootpath(RootPath),
ok = filelib:ensure_dir(SSTPath),
ok = file:write_file(SSTPath ++ "/test1.sst", "hello_world"),
ok = file:write_file(SSTPath ++ "/test2.sst", "hello_world"),
ok = file:write_file(SSTPath ++ "/test3.bob", "hello_world"),
UsedFiles = ["./test1.sst"],
ok = archive_files(RootPath, UsedFiles),
{ok, AllFiles} = file:list_dir(SSTPath),
?assertMatch(true, lists:member("test1.sst", AllFiles)),
?assertMatch(false, lists:member("test2.sst", AllFiles)),
?assertMatch(true, lists:member("test3.bob", AllFiles)),
?assertMatch(true, lists:member("test2.bak", AllFiles)),
ok = clean_subdir(SSTPath).
simple_server_test() -> simple_server_test() ->
RootPath = "../test/ledger", RootPath = "../test/ledger",
clean_testdir(RootPath), clean_testdir(RootPath),