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

Further corrupt file unit test for CDB

Browse source
This commit is contained in:
martinsumner 2016-11-08 23:07:03 +00:00
parent cc3cbc983b
commit 0905ea51e2
1 changed files with 27 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

34
src/leveled_cdb.erl
View file

@ -1324,14 +1324,10 @@ dump(FileName) ->
case read_next_term(Handle, VL, crc) of case read_next_term(Handle, VL, crc) of
{_, Value} -> {_, Value} ->
{ok, CurrLoc} = file:position(Handle, cur), {ok, CurrLoc} = file:position(Handle, cur),
Return = {Key,Value} = get(Handle, Key)
case get(Handle, Key) of
{Key,Value} -> {Key ,Value};
X -> {wonky, X}
end
end, end,
{ok, _} = file:position(Handle, CurrLoc), {ok, _} = file:position(Handle, CurrLoc),
[Return | Acc] [{Key,Value} | Acc]
end, end,
lists:foldr(Fn1, [], lists:seq(0, NumberOfPairs-1)). lists:foldr(Fn1, [], lists:seq(0, NumberOfPairs-1)).
@ -1813,7 +1809,7 @@ corruptfile_test() ->
lists:foreach(fun(Offset) -> corrupt_testfile_at_offset(Offset) end, lists:foreach(fun(Offset) -> corrupt_testfile_at_offset(Offset) end,
lists:seq(1, 40)), lists:seq(1, 40)),
ok = file:delete("../test/corrupt_test.pnd"). ok = file:delete("../test/corrupt_test.pnd").
corrupt_testfile_at_offset(Offset) -> corrupt_testfile_at_offset(Offset) ->
{ok, F1} = file:open("../test/corrupt_test.pnd", ?WRITE_OPS), {ok, F1} = file:open("../test/corrupt_test.pnd", ?WRITE_OPS),
{ok, EofPos} = file:position(F1, eof), {ok, EofPos} = file:position(F1, eof),
@ -1829,6 +1825,30 @@ corrupt_testfile_at_offset(Offset) ->
?assertMatch({"Key100", "Value100"}, cdb_get(P2, "Key100")), ?assertMatch({"Key100", "Value100"}, cdb_get(P2, "Key100")),
ok = cdb_close(P2). ok = cdb_close(P2).
crc_corrupt_writer_test() ->
file:delete("../test/corruptwrt_test.pnd"),
{ok, P1} = cdb_open_writer("../test/corruptwrt_test.pnd",
#cdb_options{binary_mode=false}),
KVList = generate_sequentialkeys(100, []),
ok = cdb_mput(P1, KVList),
?assertMatch(probably, cdb_keycheck(P1, "Key1")),
?assertMatch({"Key1", "Value1"}, cdb_get(P1, "Key1")),
?assertMatch({"Key100", "Value100"}, cdb_get(P1, "Key100")),
ok = cdb_close(P1),
{ok, Handle} = file:open("../test/corruptwrt_test.pnd", ?WRITE_OPS),
{ok, EofPos} = file:position(Handle, eof),
% zero the last byte of the last value
ok = file:pwrite(Handle, EofPos - 5, <<0:8/integer>>),
ok = file:close(Handle),
{ok, P2} = cdb_open_writer("../test/corruptwrt_test.pnd",
#cdb_options{binary_mode=false}),
?assertMatch(probably, cdb_keycheck(P2, "Key1")),
?assertMatch({"Key1", "Value1"}, cdb_get(P2, "Key1")),
?assertMatch(missing, cdb_get(P2, "Key100")),
ok = cdb_put(P2, "Key100", "Value100"),
?assertMatch({"Key100", "Value100"}, cdb_get(P2, "Key100")),
ok = cdb_close(P2).
nonsense_coverage_test() -> nonsense_coverage_test() ->
{ok, Pid} = gen_fsm:start(?MODULE, [#cdb_options{}], []), {ok, Pid} = gen_fsm:start(?MODULE, [#cdb_options{}], []),
ok = gen_fsm:send_all_state_event(Pid, nonsense), ok = gen_fsm:send_all_state_event(Pid, nonsense),