93 lines
3 KiB
Erlang
93 lines
3 KiB
Erlang
|
%% -------------------------------------------------------------------
|
||
|
|
%%
|
||
|
|
%% bdberl: Port Driver Stress tests
|
||
|
|
%% Copyright (c) 2008 The Hive. All rights reserved.
|
||
|
|
%%
|
||
|
|
%% -------------------------------------------------------------------
|
||
|
|
-module(stress_SUITE).
|
||
|
|
-compile(export_all).
|
||
|
|
-include_lib("ct.hrl").
|
||
|
|
|
||
|
|
%% NOTE: all of the tests are set for a low number of iterations to guarantee
|
||
|
|
%% that they all pass and run in a reasonable amount of time. That kinda defeats
|
||
|
|
%% the purpose of the test, tho. Work is ongoing to make this a useful test suite.
|
||
|
|
|
||
|
|
all() ->
|
||
|
|
[rewrite_array_test,
|
||
|
|
rewrite_bytes_test,
|
||
|
|
write_array_test,
|
||
|
|
write_bytes_test].
|
||
|
|
|
||
|
|
init_per_suite(Config) ->
|
||
|
|
{ok, Cwd} = file:get_cwd(),
|
||
|
|
{ok, _} = file:copy(lists:append([Cwd, "/../../../int_test/DB_CONFIG"]),
|
||
|
|
lists:append([Cwd, "/DB_CONFIG"])),
|
||
|
|
crypto:start(),
|
||
|
|
Config.
|
||
|
|
|
||
|
|
end_per_suite(_Config) ->
|
||
|
|
ok.
|
||
|
|
|
||
|
|
init_per_testcase(TestCase, Config) ->
|
||
|
|
Size = 1024 * 1024,
|
||
|
|
Chunk = crypto:rand_bytes(Size),
|
||
|
|
Name = io_lib:format("~p.db", [TestCase]),
|
||
|
|
{ok, Db} = bdberl:open(Name, hash),
|
||
|
|
[{size, Size}, {chunk, Chunk}, {db, Db}|Config].
|
||
|
|
|
||
|
|
end_per_testcase(_TestCase, Config) ->
|
||
|
|
bdberl:close(?config(db, Config)),
|
||
|
|
ok.
|
||
|
|
|
||
|
|
%%---------------------------------------------------------------------------
|
||
|
|
|
||
|
|
rewrite_array_test(Config) ->
|
||
|
|
%% If you try to run this one for more than 2K iterations than the Erlang
|
||
|
|
%% VM will die with a memory allocation error when creating the binary.
|
||
|
|
ct:print("Running rewrite_array test for 2000 iterations..."),
|
||
|
|
Chunk = ?config(chunk, Config),
|
||
|
|
rewrite_array(?config(db, Config), Chunk, [Chunk], 20).
|
||
|
|
|
||
|
|
rewrite_array(_Db, _Block, _Bytes, 0) ->
|
||
|
|
ok;
|
||
|
|
rewrite_array(Db, Block, Bytes, Iter) ->
|
||
|
|
bdberl:put(Db, 1, Bytes),
|
||
|
|
rewrite_array(Db, Block, [Block|Bytes], Iter - 1).
|
||
|
|
|
||
|
|
%%---------------------------------------------------------------------------
|
||
|
|
|
||
|
|
rewrite_bytes_test(Config) ->
|
||
|
|
ct:print("Running rewrite_bytes test for 2500 iterations..."),
|
||
|
|
rewrite_bytes(?config(db, Config), ?config(chunk, Config), 25).
|
||
|
|
|
||
|
|
rewrite_bytes(_Db, _Bytes, 0) ->
|
||
|
|
ok;
|
||
|
|
rewrite_bytes(Db, Bytes, Iter) ->
|
||
|
|
bdberl:put(Db, 1, Bytes),
|
||
|
|
rewrite_bytes(Db, Bytes, Iter - 1).
|
||
|
|
|
||
|
|
%%---------------------------------------------------------------------------
|
||
|
|
|
||
|
|
write_array_test(Config) ->
|
||
|
|
ct:print("Running write_array test for 150 iterations..."),
|
||
|
|
Chunk = ?config(chunk, Config),
|
||
|
|
write_array(?config(db, Config), Chunk, [Chunk], 15).
|
||
|
|
|
||
|
|
write_array(_Db, _Block, _Bytes, 0) ->
|
||
|
|
ok;
|
||
|
|
write_array(Db, Block, Bytes, Iter) ->
|
||
|
|
bdberl:put(Db, Iter, Bytes),
|
||
|
|
write_array(Db, Block, [Block|Bytes], Iter - 1).
|
||
|
|
|
||
|
|
%%---------------------------------------------------------------------------
|
||
|
|
|
||
|
|
write_bytes_test(Config) ->
|
||
|
|
ct:print("Running write_bytes test for 2500 iterations..."),
|
||
|
|
write_bytes(?config(db, Config), ?config(chunk, Config), 25).
|
||
|
|
|
||
|
|
write_bytes(_Db, _Bytes, 0) ->
|
||
|
|
ok;
|
||
|
|
write_bytes(Db, Bytes, Iter) ->
|
||
|
|
bdberl:put(Db, Iter, Bytes),
|
||
|
|
write_bytes(Db, Bytes, Iter - 1).
|