Skip to content

Commit

Permalink
Add verify_snmp_repl riak_test
Browse files Browse the repository at this point in the history
The verify_snmp_repl test reproduces the error condition (one cluster
configured for realtime replication with two or more other clusters) and
passes if and only if the riak_snmp_stat_poller module correctly handles
this configuration.

https://bashoeng.atlassian.net/browse/RIAK-1884
basho/riak_snmp#27
  • Loading branch information
jvoegele committed Jul 20, 2015
1 parent 62335cf commit 0ffcc61
Showing 1 changed file with 100 additions and 0 deletions.
100 changes: 100 additions & 0 deletions tests/verify_snmp_repl.erl
Original file line number Diff line number Diff line change
@@ -0,0 +1,100 @@
%% -------------------------------------------------------------------
%%
%% Copyright (c) 2010-2015 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(verify_snmp_repl).
-behavior(riak_test).
-export([confirm/0]).
-include_lib("eunit/include/eunit.hrl").
-compile({parse_transform, rt_intercept_pt}).

confirm() ->
Clusters = make_clusters(["cluster-1", "cluster-2", "cluster-3"], 1),
[{_, Leader, _}|_] = Clusters,
intercept_riak_snmp_stat_poller(Leader),
wait_for_snmp_stat_poller().

make_clusters(Names, NodeCount) ->
ClusterCount = length(Names),
Config = [{riak_snmp, [{polling_interval, 100}]}],
AllNodes = make_nodes(NodeCount, ClusterCount, Config),
Clusters = lists:zip(Names, AllNodes),
lists:foreach(fun make_named_cluster/1, Clusters),
lists:foreach(fun wait_until_ring_converged/1, Clusters),
lists:foreach(fun wait_until_leader_converge/1, Clusters),

ClustersWithLeaders = [{Name, repl_util:get_leader(hd(Nodes)), Nodes} || {Name, Nodes} <- Clusters],
enable_realtime(ClustersWithLeaders),
ClustersWithLeaders.

intercept_riak_snmp_stat_poller(Node) ->
RiakTestProcess = self(),
rt_intercept:add(
Node,
{riak_snmp_stat_poller,
[{{set_rows, 4},
{[RiakTestProcess],
fun(Table, Indexes, Cols, IndexCol)
when Table =:= replRealtimeStatusTable; Table =:= replFullsyncStatusTable ->
lager:log(info, self(), "JVOEGELE> set_rows(~p, ~p, ~p, ~p)", [Table, Indexes, Cols, IndexCol]),

This comment has been minimized.

Copy link
@JeetKunDoug

JeetKunDoug Jul 20, 2015

Contributor

Can probably drop this entirely, or at least the JVOEGELE part.

try
riak_snmp_stat_poller_orig:set_rows_orig(Table, Indexes, Cols, IndexCol),
RiakTestProcess ! pass
catch
Exception:Reason ->
RiakTestProcess ! {fail, {Exception, Reason}},
error({Exception, Reason})
end
end}}]}).

wait_for_snmp_stat_poller() ->
receive
pass -> pass;
{fail, Reason} -> {fail, Reason};
X -> {fail, {unknown, X}}
after
1000 -> {fail, timeout}
end.

make_nodes(NodeCount, ClusterCount, Config) ->
Nodes = rt:deploy_nodes(NodeCount * ClusterCount, Config),
sublists(Nodes, NodeCount).

sublists(List, Len) ->
lists:map(
fun(I) -> lists:sublist(List, I, Len) end,
lists:seq(1, length(List), Len)).

make_named_cluster({Name, Nodes}) ->
repl_util:make_cluster(Nodes),
repl_util:name_cluster(hd(Nodes), Name).

wait_until_ring_converged({_Name, Nodes}) ->
rt:wait_until_ring_converged(Nodes).

wait_until_leader_converge({_Name, Nodes}) ->
repl_util:wait_until_leader_converge(Nodes).

enable_realtime([{_, Node, _}|OtherClusters]) ->
lists:foreach(
fun({Cluster, _, _}) ->
repl_util:enable_realtime(Node, Cluster)
end,
OtherClusters).

0 comments on commit 0ffcc61

Please sign in to comment.