mirror of
https://github.com/valitydev/riak_test.git
synced 2024-11-06 16:45:29 +00:00
6bb72e2747
Establish a new PB connection to the legacy node after it is upgraded
in order to avoid a failure. The PB connection may close if the node
upgrade takes too long and its reuse in such a case can lead to test
failure because use of the pid returns {error, disconnected} errors.
85 lines
3.3 KiB
Erlang
85 lines
3.3 KiB
Erlang
%% -------------------------------------------------------------------
|
|
%%
|
|
%% Copyright (c) 2013 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.
|
|
%%
|
|
%% -------------------------------------------------------------------
|
|
%%% @author Russell Brown <russelldb@basho.com>
|
|
%%% @copyright (C) 2013, Basho Technologies
|
|
%%% @doc
|
|
%%% riak_test for counter cabability
|
|
%%% @end
|
|
|
|
-module(verify_counter_capability).
|
|
-behavior(riak_test).
|
|
-export([confirm/0]).
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
|
|
-define(BUCKET, <<"test-counters">>).
|
|
-define(KEY, <<"foo">>).
|
|
|
|
|
|
confirm() ->
|
|
%% Create a mixed cluster of legacy and previous
|
|
%% Create a PB client
|
|
%% GET / PUT on older and newer cluster
|
|
%% Upgrade nodes to previous
|
|
%% Get put on all nodes
|
|
Config = [],
|
|
[Legacy, Previous]=Nodes = rt:build_cluster([{legacy, Config}, {previous, Config}]),
|
|
?assertEqual(ok, rt:wait_until_capability(Previous, {riak_kv, crdt}, [])),
|
|
verify_counter_converge:set_allow_mult_true(Nodes),
|
|
|
|
{PrevPB, PrevHttp} = get_clients(Legacy),
|
|
{PB, Http} = get_clients(Previous),
|
|
|
|
?assertMatch({error, {ok, "404", _, _}}, rhc:counter_incr(PrevHttp, ?BUCKET, ?KEY, 1)),
|
|
?assertMatch({error, {ok, "404", _, _}}, rhc:counter_val(PrevHttp, ?BUCKET, ?KEY)),
|
|
|
|
?assertMatch({error, {ok, "503", _, _}}, rhc:counter_incr(Http, ?BUCKET, ?KEY, 1)),
|
|
?assertMatch({error, {ok, "503", _, _}}, rhc:counter_val(Http, ?BUCKET, ?KEY)),
|
|
|
|
?assertEqual({error,<<"Unknown message code.">>}, riakc_pb_socket:counter_incr(PrevPB, ?BUCKET, ?KEY, 1)),
|
|
?assertEqual({error,<<"Unknown message code.">>}, riakc_pb_socket:counter_val(PrevPB, ?BUCKET, ?KEY)),
|
|
?assertEqual({error,<<"\"Counters are not supported\"">>}, riakc_pb_socket:counter_incr(PB, ?BUCKET, ?KEY, 1)),
|
|
?assertEqual({error,<<"\"Counters are not supported\"">>}, riakc_pb_socket:counter_val(PB, ?BUCKET, ?KEY)),
|
|
|
|
riakc_pb_socket:stop(PrevPB),
|
|
|
|
rt:upgrade(Legacy, previous),
|
|
|
|
PrevPB2 = rt:pbc(Legacy),
|
|
|
|
?assertEqual(ok, rt:wait_until_capability(Previous, {riak_kv, crdt}, [pncounter])),
|
|
|
|
?assertMatch(ok, rhc:counter_incr(PrevHttp, ?BUCKET, ?KEY, 1)),
|
|
?assertMatch({ok, 1}, rhc:counter_val(PrevHttp, ?BUCKET, ?KEY)),
|
|
|
|
?assertMatch(ok, rhc:counter_incr(Http, ?BUCKET, ?KEY, 1)),
|
|
?assertMatch({ok, 2}, rhc:counter_val(Http, ?BUCKET, ?KEY)),
|
|
|
|
?assertEqual(ok, riakc_pb_socket:counter_incr(PrevPB2, ?BUCKET, ?KEY, 1)),
|
|
?assertEqual({ok, 3}, riakc_pb_socket:counter_val(PrevPB2, ?BUCKET, ?KEY)),
|
|
?assertEqual(ok, riakc_pb_socket:counter_incr(PB, ?BUCKET, ?KEY, 1)),
|
|
?assertEqual({ok, 4}, riakc_pb_socket:counter_val(PB, ?BUCKET, ?KEY)),
|
|
|
|
[riakc_pb_socket:stop(C) || C <- [PB, PrevPB2]],
|
|
|
|
pass.
|
|
|
|
get_clients(Node) ->
|
|
{rt:pbc(Node), rt:httpc(Node)}.
|