Skip to content

Commit d6acf2d

Browse files
committed
Updates based on feedback from PR review
1 parent e55eb67 commit d6acf2d

File tree

1 file changed

+8
-18
lines changed

1 file changed

+8
-18
lines changed

test/riak_repl2_rtq_tests.erl

Lines changed: 8 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,12 @@
22
-compile(export_all).
33
-include_lib("eunit/include/eunit.hrl").
44

5-
-define(setup_env, application:set_env(riak_repl, rtq_max_bytes, 10*1024*1024)).
6-
-define(clean_env, application:unset_env(riak_repl, rtq_max_bytes)).
5+
-define(SETUP_ENV, application:set_env(riak_repl, rtq_max_bytes, 10*1024*1024)).
6+
-define(CLEAN_ENV, application:unset_env(riak_repl, rtq_max_bytes)).
77

88
rtq_trim_test() ->
99
%% make sure the queue is 10mb
10-
?setup_env,
10+
?SETUP_ENV,
1111
{ok, Pid} = riak_repl2_rtq:start_test(),
1212
try
1313
gen_server:call(Pid, {register, rtq_test}),
@@ -22,7 +22,7 @@ rtq_trim_test() ->
2222
%% the queue is now empty
2323
?assert(gen_server:call(Pid, {is_empty, rtq_test}))
2424
after
25-
?clean_env,
25+
?CLEAN_ENV,
2626
exit(Pid, kill)
2727
end.
2828

@@ -48,12 +48,12 @@ accumulate(Pid, Acc, C) ->
4848

4949
status_test_() ->
5050
{setup, fun() ->
51-
?setup_env,
51+
?SETUP_ENV,
5252
{ok, QPid} = riak_repl2_rtq:start_link(),
5353
QPid
5454
end,
5555
fun(QPid) ->
56-
?clean_env,
56+
?CLEAN_ENV,
5757
riak_repl_test_util:kill_and_wait(QPid)
5858
end,
5959
fun(_QPid) -> [
@@ -275,15 +275,13 @@ overload_test_() ->
275275
]}.
276276

277277
start_rtq() ->
278-
?setup_env,
279-
%% application:start(lager),
280-
%% lager:set_loglevel(lager_console_backend, debug),
278+
?SETUP_ENV,
281279
{ok, Pid} = riak_repl2_rtq:start_link(),
282280
gen_server:call(Pid, {register, rtq_test}),
283281
Pid.
284282

285283
kill_rtq(QPid) ->
286-
?clean_env,
284+
?CLEAN_ENV,
287285
riak_repl_test_util:kill_and_wait(QPid).
288286

289287
object_format() -> riak_core_capability:get({riak_kv, object_format}, v0).
@@ -298,14 +296,6 @@ push_object(Bucket, Key) ->
298296
riak_repl2_rtq:push(1, Obj),
299297
Obj.
300298

301-
push_random() ->
302-
push_random(5).
303-
304-
push_random(Num) ->
305-
RandomData = crypto:rand_bytes(1024 * 1024),
306-
[riak_repl2_rtq:push(1, RandomData) || _ <- lists:seq(1, Num)],
307-
RandomData.
308-
309299
pull(N) ->
310300
lists:foldl(fun(_Nth, _LastSeq) ->
311301
pull()

0 commit comments

Comments
 (0)