riak_test/tests/gh_riak_core_155.erl
engelsanchez c498fae357 Merge pull request #131 from basho/rz-intercept
Intercepts - Powerful Feature to Augment Code/State
2013-01-03 06:38:11 -08:00

70 lines
2.5 KiB
Erlang

%% -------------------------------------------------------------------
%%
%% Copyright (c) 2012 Basho Technologies, Inc.
%%
%% This file is provided to you under the Apache License,
%% Version 2.0 (the "License"); you may not use this file
%% except in compliance with the License. You may obtain
%% a copy of the License at
%%
%% http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing,
%% software distributed under the License is distributed on an
%% "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
%% KIND, either express or implied. See the License for the
%% specific language governing permissions and limitations
%% under the License.
%%
%% -------------------------------------------------------------------
-module(gh_riak_core_155).
-behavior(riak_test).
-export([confirm/0]).
-compile(export_all).
-include_lib("eunit/include/eunit.hrl").
confirm() ->
[Node] = rt:build_cluster(1),
%% Generate a valid preflist for our get requests
rpc:call(Node, riak_core, wait_for_service, [riak_kv]),
BKey = {<<"bucket">>, <<"value">>},
DocIdx = riak_core_util:chash_std_keyfun(BKey),
PL = rpc:call(Node, riak_core_apl, get_apl, [DocIdx, 3, riak_kv]),
lager:info("Adding delayed start to app.config"),
NewConfig = [{riak_core, [{delayed_start, 1000}]}],
rt:update_app_config(Node, NewConfig),
%% Restart node, add intercept that delay proxy startup, and issue gets.
%% Gets will come in before proxies started, and should trigger crash.
rt:stop_and_wait(Node),
rt:async_start(Node),
rt:wait_until_pingable(Node),
rt_intercept:load_intercepts([Node]),
rt_intercept:add(Node, {riak_core_vnode_proxy_sup,
[{{start_proxies,1}, sleep_start_proxies}]}),
lager:info("Installed intercept to delay riak_kv proxy startup"),
lager:info("Issuing 10000 gets against ~p", [Node]),
perform_gets(10000, Node, PL, BKey),
lager:info("Verifying ~p has not crashed", [Node]),
[begin
?assertEqual(pong, net_adm:ping(Node)),
timer:sleep(1000)
end || _ <- lists:seq(1,10)],
lager:info("Test passed"),
pass.
perform_gets(Count, Node, PL, BKey) ->
rpc:call(Node, riak_kv_vnode, get, [PL, BKey, make_ref()]),
perform_gets2(Count, Node, PL, BKey).
perform_gets2(0, _, _, _) ->
ok;
perform_gets2(Count, Node, PL, BKey) ->
rpc:call(Node, riak_kv_vnode, get, [PL, BKey, make_ref()], 1000),
perform_gets(Count - 1, Node, PL, BKey).