mirror of
https://github.com/valitydev/riak_test.git
synced 2024-11-06 16:45:29 +00:00
b103c52a87
Reduces the probability of a race condition between the calculation of spiral/histogram metrics and SNMP stat cache refresh by reducing the SNMP poll interval to 1 second during test execution
98 lines
3.7 KiB
Erlang
98 lines
3.7 KiB
Erlang
%% -------------------------------------------------------------------
|
|
%%
|
|
%% Copyright (c) 2010-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(verify_snmp).
|
|
-behavior(riak_test).
|
|
-export([confirm/0]).
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
|
|
-prereq("curl").
|
|
|
|
confirm() ->
|
|
%% Bring up a small cluster
|
|
Config = [{riak_snmp, [{polling_interval, 1000}]}],
|
|
[Node1] = rt:deploy_nodes(1, Config),
|
|
?assertEqual(ok, rt:wait_until_nodes_ready([Node1])),
|
|
|
|
Keys = [{vnodeGets,<<"vnode_gets">>},
|
|
{vnodePuts,<<"vnode_puts">>},
|
|
{nodeGets,<<"node_gets">>},
|
|
{nodePuts,<<"node_puts">>},
|
|
{nodeGetTimeMean,<<"node_get_fsm_time_mean">>},
|
|
{nodeGetTimeMedian,<<"node_get_fsm_time_median">>},
|
|
{nodeGetTime95,<<"node_get_fsm_time_95">>},
|
|
{nodeGetTime99,<<"node_get_fsm_time_99">>},
|
|
{nodeGetTime100,<<"node_get_fsm_time_100">>},
|
|
{nodePutTimeMean,<<"node_put_fsm_time_mean">>},
|
|
{nodePutTimeMedian,<<"node_put_fsm_time_median">>},
|
|
{nodePutTime95,<<"node_put_fsm_time_95">>},
|
|
{nodePutTime99,<<"node_put_fsm_time_99">>},
|
|
{nodePutTime100,<<"node_put_fsm_time_100">>}],
|
|
|
|
lager:info("Waiting for SNMP to start."),
|
|
|
|
rpc:call(Node1, riak_core, wait_for_application, [snmp]),
|
|
rpc:call(Node1, riak_core, wait_for_application, [riak_snmp]),
|
|
|
|
lager:info("Mapping SNMP names to OIDs"),
|
|
|
|
OIDPairs = [ begin
|
|
{value, OID} = rpc:call(Node1, snmpa, name_to_oid, [SKey]),
|
|
{OID ++ [0], HKey}
|
|
end || {SKey, HKey} <- Keys ],
|
|
|
|
lager:info("Doing some reads and writes to record some stats."),
|
|
|
|
rt:systest_write(Node1, 10),
|
|
rt:systest_read(Node1, 10),
|
|
|
|
lager:info("Waiting for HTTP Stats to be non-zero"),
|
|
?assertEqual(ok,
|
|
rt:wait_until(Node1, fun(N) ->
|
|
Stats = get_stats(N),
|
|
proplists:get_value(<<"vnode_gets">>, Stats) =/= 0
|
|
end)),
|
|
|
|
|
|
verify_eq(OIDPairs, Node1),
|
|
pass.
|
|
|
|
verify_eq(Keys, Node) ->
|
|
{OIDs, HKeys} = lists:unzip(Keys),
|
|
?assertEqual(ok,
|
|
rt:wait_until(Node,
|
|
fun(N) ->
|
|
Stats = get_stats(Node),
|
|
SStats = rpc:call(N, snmpa, get, [snmp_master_agent, OIDs]),
|
|
SPairs = lists:zip(SStats, HKeys),
|
|
lists:all(
|
|
fun({A,B}) ->
|
|
Stat = proplists:get_value(B, Stats),
|
|
lager:info("Comparing ~p | Stats ~p ~~ SNMP ~p", [B, Stat, A]),
|
|
A == Stat
|
|
end,
|
|
SPairs)
|
|
end)).
|
|
|
|
get_stats(Node) ->
|
|
StatString = os:cmd(io_lib:format("curl -s -S ~s/stats", [rt:http_url(Node)])),
|
|
{struct, Stats} = mochijson2:decode(StatString),
|
|
Stats.
|