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
88rtq_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
4949status_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
277277start_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
285283kill_rtq (QPid ) ->
286- ? clean_env ,
284+ ? CLEAN_ENV ,
287285 riak_repl_test_util :kill_and_wait (QPid ).
288286
289287object_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-
309299pull (N ) ->
310300 lists :foldl (fun (_Nth , _LastSeq ) ->
311301 pull ()
0 commit comments