mirror of
https://github.com/valitydev/riak_test.git
synced 2024-11-06 08:35:22 +00:00
d32d007f4d
Changing to fetching the list of peers first, then check if the riak_kv service is up. If the service is up, then check the peers. Otherwise it is possible to see the service down, then peers up because it went up in the interim. Also, making KV vnode delay configurable.
77 lines
3.2 KiB
Erlang
77 lines
3.2 KiB
Erlang
%% -------------------------------------------------------------------
|
|
%%
|
|
%% Copyright (c) 2013-2014 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(ensemble_basic2).
|
|
-export([confirm/0]).
|
|
-compile({parse_transform, rt_intercept_pt}).
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
|
|
confirm() ->
|
|
NumNodes = 5,
|
|
NVal = 5,
|
|
Config = ensemble_util:fast_config(NVal),
|
|
lager:info("Building cluster and waiting for ensemble to stablize"),
|
|
Nodes = ensemble_util:build_cluster(NumNodes, Config, NVal),
|
|
Node = hd(Nodes),
|
|
Ensembles = ensemble_util:ensembles(Node),
|
|
lager:info("Killing all ensemble leaders"),
|
|
ok = ensemble_util:kill_leaders(Node, Ensembles),
|
|
ensemble_util:wait_until_stable(Node, NVal),
|
|
Peers = [PeerId || {PeerId, _PeerPid} <- ensemble_util:peers(Node)],
|
|
lager:info("Verifying peers wait for riak_kv_service"),
|
|
Delay = rt_config:get(kv_vnode_delay, 5000),
|
|
rt_intercept:add_and_save(Node, {riak_kv_vnode, [{{init, 1}, {[Delay],
|
|
fun(Args) ->
|
|
timer:sleep(Delay),
|
|
riak_kv_vnode_orig:init_orig(Args)
|
|
end}}]}),
|
|
rt:stop_and_wait(Node),
|
|
rt:start(Node),
|
|
lager:info("Polling peers while riak_kv starts. We should see none"),
|
|
UpNoPeersFun =
|
|
fun() ->
|
|
PL = ensemble_util:peers(Node),
|
|
NodePeers = [P || {P, _} <- PL],
|
|
NonRootPeers = [P || P <- NodePeers, element(1, P) /= root],
|
|
S = rpc:call(Node, riak_core_node_watcher, services, [Node]),
|
|
case S of
|
|
L when is_list(L) ->
|
|
case lists:member(riak_kv, L) of
|
|
true ->
|
|
true;
|
|
false ->
|
|
?assertEqual([], NonRootPeers)
|
|
end;
|
|
Err ->
|
|
?assertEqual(ok, {peer_get_error, Err})
|
|
end
|
|
end,
|
|
rt:wait_until(UpNoPeersFun),
|
|
lager:info("Perfect. riak_kv is now up and no peers started before that. "
|
|
"Now check they come back up"),
|
|
SPeers = lists:sort(Peers),
|
|
?assertEqual(ok, rt:wait_until(fun() ->
|
|
L = ensemble_util:peers(Node),
|
|
L2 = lists:sort([P || {P, _} <- L]),
|
|
SPeers == L2
|
|
end)),
|
|
lager:info("All expected peers are back. Life is good"),
|
|
pass.
|