mirror of
https://github.com/valitydev/riak_test.git
synced 2024-11-06 08:35:22 +00:00
Refactored so reusable functions are available for other tests.
verify_listkeys and verify_job_enable_ac should be refactored to use the shared functions. YZ flags not currently tested.
This commit is contained in:
Ted Burghart
Torben Hoffmann
parent
7bfa8c0e8c
commit
94d38b1af2
362
tests/job_enable_common.erl
Normal file
362
tests/job_enable_common.erl
Normal file
@ -0,0 +1,362 @@
|
||||
%% -------------------------------------------------------------------
|
||||
%%
|
||||
%% Copyright (c) 2016 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(job_enable_common).
|
||||
|
||||
% Public API for use by other tests
|
||||
-export([
|
||||
bin_bucket/1,
|
||||
bin_key/1,
|
||||
bin_val/1,
|
||||
close_client/1,
|
||||
enabled_string/1,
|
||||
index_2i/0,
|
||||
index_yz/0,
|
||||
load_data/1,
|
||||
num_buckets/0, num_buckets/1,
|
||||
num_keys/0, num_keys/1,
|
||||
open_client/2,
|
||||
populated_bucket/0,
|
||||
setup_cluster/1,
|
||||
test_buckets/0,
|
||||
test_keys/0,
|
||||
test_nums/0,
|
||||
test_operation/4,
|
||||
test_vals/0,
|
||||
undefined_bucket/0
|
||||
]).
|
||||
|
||||
-include_lib("eunit/include/eunit.hrl").
|
||||
-include_lib("riakc/include/riakc.hrl").
|
||||
-include("job_enable_common.hrl").
|
||||
|
||||
-define(DEFAULT_NUM_BUCKETS, 7).
|
||||
-define(DEFAULT_NUM_KEYS, 9).
|
||||
|
||||
%% ===================================================================
|
||||
%% Test API
|
||||
%% ===================================================================
|
||||
|
||||
enabled_string(true) ->
|
||||
"enabled";
|
||||
enabled_string(false) ->
|
||||
"disabled".
|
||||
|
||||
bin_bucket(Num) ->
|
||||
erlang:list_to_binary(["Bucket_", erlang:integer_to_list(Num)]).
|
||||
|
||||
bin_key(Num) ->
|
||||
erlang:list_to_binary(["Key_", erlang:integer_to_list(Num)]).
|
||||
|
||||
bin_val(Num) ->
|
||||
erlang:list_to_binary(["Val_", erlang:integer_to_list(Num)]).
|
||||
|
||||
index_2i() ->
|
||||
{integer_index, "valnum_index_2i"}.
|
||||
|
||||
index_yz() ->
|
||||
<<"valnum_index_yz">>.
|
||||
|
||||
num_buckets() ->
|
||||
Key = {?MODULE, num_buckets},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
Num = ?DEFAULT_NUM_BUCKETS,
|
||||
erlang:put(Key, Num),
|
||||
Num;
|
||||
Val ->
|
||||
Val
|
||||
end.
|
||||
|
||||
num_buckets(Num) when erlang:is_integer(Num) andalso Num > 0 ->
|
||||
Key = {?MODULE, num_buckets},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
erlang:put(Key, Num),
|
||||
Num;
|
||||
Num ->
|
||||
Num;
|
||||
_ ->
|
||||
erlang:erase({?MODULE, test_buckets}),
|
||||
erlang:erase({?MODULE, populated_bucket}),
|
||||
erlang:put(Key, Num),
|
||||
Num
|
||||
end.
|
||||
|
||||
num_keys() ->
|
||||
Key = {?MODULE, num_keys},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
Num = ?DEFAULT_NUM_KEYS,
|
||||
erlang:put(Key, Num),
|
||||
Num;
|
||||
Val ->
|
||||
Val
|
||||
end.
|
||||
|
||||
num_keys(Num) when erlang:is_integer(Num) andalso Num > 0 ->
|
||||
Key = {?MODULE, num_keys},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
erlang:put(Key, Num),
|
||||
Num;
|
||||
Num ->
|
||||
Num;
|
||||
_ ->
|
||||
erlang:erase({?MODULE, test_keys}),
|
||||
erlang:erase({?MODULE, test_nums}),
|
||||
erlang:erase({?MODULE, test_vals}),
|
||||
erlang:put(Key, Num),
|
||||
Num
|
||||
end.
|
||||
|
||||
populated_bucket() ->
|
||||
Key = {?MODULE, populated_bucket},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
Buckets = test_buckets(),
|
||||
Bucket = lists:nth(erlang:length(Buckets) div 2, Buckets),
|
||||
erlang:put(Key, Bucket),
|
||||
Bucket;
|
||||
Val ->
|
||||
Val
|
||||
end.
|
||||
|
||||
undefined_bucket() ->
|
||||
<<"Undefined_Bucket">>.
|
||||
|
||||
test_buckets() ->
|
||||
Key = {?MODULE, test_buckets},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
New = bin_buckets(num_buckets(), []),
|
||||
erlang:put(Key, New),
|
||||
New;
|
||||
Val ->
|
||||
Val
|
||||
end.
|
||||
|
||||
test_keys() ->
|
||||
Key = {?MODULE, test_keys},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
New = bin_keys(num_keys(), []),
|
||||
erlang:put(Key, New),
|
||||
New;
|
||||
Val ->
|
||||
Val
|
||||
end.
|
||||
|
||||
test_nums() ->
|
||||
Key = {?MODULE, test_nums},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
New = lists:seq(1, num_keys()),
|
||||
erlang:put(Key, New),
|
||||
New;
|
||||
Val ->
|
||||
Val
|
||||
end.
|
||||
|
||||
test_vals() ->
|
||||
Key = {?MODULE, test_vals},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
New = bin_vals(num_keys(), []),
|
||||
erlang:put(Key, New),
|
||||
New;
|
||||
Val ->
|
||||
Val
|
||||
end.
|
||||
|
||||
open_client(http = Type, Node) ->
|
||||
% HTTP connections are constant records, so re-use them
|
||||
Key = {?MODULE, httpc, Node},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
New = {Type, rhc, rt:httpc(Node)},
|
||||
erlang:put(Key, New),
|
||||
New;
|
||||
Conn ->
|
||||
Conn
|
||||
end;
|
||||
open_client(pbc = Type, Node) ->
|
||||
{Type, riakc_pb_socket, rt:pbc(Node)}.
|
||||
|
||||
close_client({http, _Mod, _RHC}) ->
|
||||
ok;
|
||||
close_client({pbc, Mod, PBC}) ->
|
||||
Mod:stop(PBC).
|
||||
|
||||
setup_cluster([Node | _] = Nodes) ->
|
||||
lager:info("Creating a cluster of ~b nodes ...", [erlang:length(Nodes)]),
|
||||
?assertEqual(ok, rt:join_cluster(Nodes)),
|
||||
load_data(Node),
|
||||
?assertEqual(ok, rt:wait_until_transfers_complete(Nodes)).
|
||||
|
||||
load_data(Node) ->
|
||||
lager:info("Writing known data to node ~p ...", [Node]),
|
||||
PBConn = rt:pbc(Node),
|
||||
load_data(PBConn, populated_bucket(), test_buckets()),
|
||||
riakc_pb_socket:stop(PBConn).
|
||||
|
||||
test_operation(Node, ?TOKEN_LIST_BUCKETS, Enabled, ClientType) ->
|
||||
{_, Mod, Conn} = Client = open_client(ClientType, Node),
|
||||
Result = Mod:list_buckets(Conn),
|
||||
close_client(Client),
|
||||
case Enabled of
|
||||
true ->
|
||||
?assertMatch({ok, _}, Result),
|
||||
{ok, Buckets} = Result,
|
||||
?assertEqual(test_buckets(), lists:sort(Buckets));
|
||||
false ->
|
||||
case ClientType of
|
||||
pbc ->
|
||||
?assertEqual({error, ?ERRMSG_LIST_BUCKETS_DISABLED}, Result);
|
||||
http ->
|
||||
?assertMatch({error, {"403", _}}, Result)
|
||||
end
|
||||
end;
|
||||
|
||||
test_operation(Node, ?TOKEN_LIST_KEYS, Enabled, ClientType) ->
|
||||
{_, Mod, Conn} = Client = open_client(ClientType, Node),
|
||||
Result = Mod:list_keys(Conn, populated_bucket()),
|
||||
close_client(Client),
|
||||
case Enabled of
|
||||
true ->
|
||||
?assertMatch({ok, _}, Result),
|
||||
{ok, Keys} = Result,
|
||||
?assertEqual(test_keys(), lists:sort(Keys));
|
||||
false ->
|
||||
case ClientType of
|
||||
pbc ->
|
||||
?assertEqual({error, ?ERRMSG_LIST_KEYS_DISABLED}, Result);
|
||||
http ->
|
||||
?assertMatch({error, {"403", _}}, Result)
|
||||
end
|
||||
end;
|
||||
|
||||
test_operation(Node, ?TOKEN_MAP_REDUCE, Enabled, ClientType) ->
|
||||
Bucket = populated_bucket(),
|
||||
{_, Mod, Conn} = Client = open_client(ClientType, Node),
|
||||
Result = Mod:mapred(Conn, Bucket, []),
|
||||
close_client(Client),
|
||||
case Enabled of
|
||||
true ->
|
||||
?assertMatch({ok, [{_, _}]}, Result),
|
||||
{ok, [{_, Pairs}]} = Result,
|
||||
Expect = case ClientType of
|
||||
pbc ->
|
||||
[{Bucket, Key} || Key <- test_keys()];
|
||||
http ->
|
||||
[[Bucket, Key] || Key <- test_keys()]
|
||||
end,
|
||||
?assertEqual(Expect, lists:sort(Pairs));
|
||||
false ->
|
||||
case ClientType of
|
||||
pbc ->
|
||||
?assertEqual({error, ?ERRMSG_MAP_REDUCE_DISABLED}, Result);
|
||||
http ->
|
||||
?assertMatch({error, {"403", _}}, Result)
|
||||
end
|
||||
end;
|
||||
|
||||
test_operation(Node, ?TOKEN_SEC_INDEX, Enabled, ClientType) ->
|
||||
Bucket = populated_bucket(),
|
||||
Index = index_2i(),
|
||||
Num = random:uniform(num_keys()),
|
||||
{_, Mod, Conn} = Client = open_client(ClientType, Node),
|
||||
Result = Mod:get_index(Conn, Bucket, Index, Num),
|
||||
close_client(Client),
|
||||
case Enabled of
|
||||
true ->
|
||||
Key = bin_key(Num),
|
||||
?assertMatch({ok, {index_results_v1, [Key], _, _}}, Result);
|
||||
false ->
|
||||
case ClientType of
|
||||
pbc ->
|
||||
?assertEqual({error, ?ERRMSG_SEC_INDEX_DISABLED}, Result);
|
||||
http ->
|
||||
?assertMatch({error, {"403", _}}, Result)
|
||||
end
|
||||
end;
|
||||
|
||||
%% This requires that YZ be running and that
|
||||
%% riakc_pb_socket:create_search_index(Connection, index_yz())
|
||||
%% or equivalent has been successfully called before invoking this test.
|
||||
%% This module's load/data/1 function DOES NOT do this for you.
|
||||
test_operation(Node, ?TOKEN_YZ_SEARCH, Enabled, ClientType) ->
|
||||
Index = index_yz(),
|
||||
Bucket = populated_bucket(),
|
||||
Num = random:uniform(num_keys()),
|
||||
Key = bin_key(Num),
|
||||
Query = <<"_yz_rb:", Bucket/binary, " AND _yz_rk:", Key/binary>>,
|
||||
{_, Mod, Conn} = Client = open_client(ClientType, Node),
|
||||
Result = Mod:search(Conn, Index, Query),
|
||||
close_client(Client),
|
||||
case Enabled of
|
||||
true ->
|
||||
?assertMatch({ok, #search_results{}}, Result);
|
||||
false ->
|
||||
case ClientType of
|
||||
pbc ->
|
||||
?assertEqual({error, ?ERRMSG_YZ_SEARCH_DISABLED}, Result);
|
||||
http ->
|
||||
?assertMatch({error, {"403", _}}, Result)
|
||||
end
|
||||
end.
|
||||
|
||||
%% ===================================================================
|
||||
%% Internal
|
||||
%% ===================================================================
|
||||
|
||||
bin_buckets(0, Result) ->
|
||||
lists:sort(Result);
|
||||
bin_buckets(Count, Result) ->
|
||||
bin_buckets((Count - 1), [bin_bucket(Count) | Result]).
|
||||
|
||||
bin_keys(0, Result) ->
|
||||
lists:sort(Result);
|
||||
bin_keys(Count, Result) ->
|
||||
bin_keys((Count - 1), [bin_key(Count) | Result]).
|
||||
|
||||
bin_vals(0, Result) ->
|
||||
lists:sort(Result);
|
||||
bin_vals(Count, Result) ->
|
||||
bin_vals((Count - 1), [bin_val(Count) | Result]).
|
||||
|
||||
load_data(PBConn, Bucket, [Bucket | Buckets]) ->
|
||||
Index = index_2i(),
|
||||
Load = fun({Num, Key, Val}) ->
|
||||
Obj1 = riakc_obj:new(Bucket, Key, Val),
|
||||
Meta1 = riakc_obj:get_update_metadata(Obj1),
|
||||
Meta2 = riakc_obj:set_secondary_index(Meta1, [{Index, [Num]}]),
|
||||
Obj2 = riakc_obj:update_metadata(Obj1, Meta2),
|
||||
?assertEqual(ok, riakc_pb_socket:put(PBConn, Obj2))
|
||||
end,
|
||||
lists:foreach(Load, lists:zip3(test_nums(), test_keys(), test_vals())),
|
||||
load_data(PBConn, Bucket, Buckets);
|
||||
load_data(PBConn, PopBucket, [Bucket | Buckets]) ->
|
||||
?assertEqual(ok, riakc_pb_socket:put(PBConn,
|
||||
riakc_obj:new(Bucket, <<"test_key">>, <<"test_value">>))),
|
||||
load_data(PBConn, PopBucket, Buckets);
|
||||
load_data(_, _, []) ->
|
||||
ok.
|
||||
|
||||
44
tests/job_enable_common.hrl
Normal file
44
tests/job_enable_common.hrl
Normal file
@ -0,0 +1,44 @@
|
||||
%% -------------------------------------------------------------------
|
||||
%%
|
||||
%% Copyright (c) 2016 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.
|
||||
%%
|
||||
%% -------------------------------------------------------------------
|
||||
|
||||
-ifndef(job_enable_common_included).
|
||||
-define(job_enable_common_included, true).
|
||||
|
||||
-define(ADVANCED_CONFIG_KEY, 'job_accept_class').
|
||||
-define(CUTTLEFISH_PREFIX, "async.enable").
|
||||
|
||||
-define(TOKEN_LIST_BUCKETS, 'list_buckets').
|
||||
-define(TOKEN_LIST_KEYS, 'list_keys').
|
||||
-define(TOKEN_MAP_REDUCE, 'map_reduce').
|
||||
-define(TOKEN_SEC_INDEX, 'secondary_index').
|
||||
-define(TOKEN_YZ_SEARCH, 'riak_search').
|
||||
|
||||
-define(ERRMSG_LIST_BUCKETS_DISABLED,
|
||||
<<"Operation 'list_buckets' is not enabled">>).
|
||||
-define(ERRMSG_LIST_KEYS_DISABLED,
|
||||
<<"Operation 'list_keys' is not enabled">>).
|
||||
-define(ERRMSG_MAP_REDUCE_DISABLED,
|
||||
<<"Operation 'map_reduce' is not enabled">>).
|
||||
-define(ERRMSG_SEC_INDEX_DISABLED,
|
||||
<<"Secondary index queries have been disabled in the configuration">>).
|
||||
-define(ERRMSG_YZ_SEARCH_DISABLED,
|
||||
<<"Operation 'riak_search' is not enabled">>).
|
||||
|
||||
-endif. % job_enable_common_included
|
||||
@ -1,6 +1,6 @@
|
||||
%% -------------------------------------------------------------------
|
||||
%%
|
||||
%% Copyright (c) 2012-2016 Basho Technologies, Inc.
|
||||
%% Copyright (c) 2016 Basho Technologies, Inc.
|
||||
%%
|
||||
%% This file is provided to you under the Apache License,
|
||||
%% Version 2.0 (the "License"); you may not use this file
|
||||
@ -19,190 +19,69 @@
|
||||
%% -------------------------------------------------------------------
|
||||
|
||||
%% Verify functionality of async job enable/disable flags in riak.conf.
|
||||
%% Toggling flags at runtime is tested in verify_job_enable_ac.
|
||||
-module(verify_job_enable_rc).
|
||||
|
||||
-behavior(riak_test).
|
||||
-export([confirm/0]).
|
||||
|
||||
% exports for use by other tests
|
||||
-export([
|
||||
close_client/1,
|
||||
listkeys_bucket/0,
|
||||
open_client/2,
|
||||
setup_cluster/1,
|
||||
test_buckets/0,
|
||||
test_keys/0,
|
||||
test_operation/4,
|
||||
undefined_bucket/0
|
||||
]).
|
||||
|
||||
-include_lib("eunit/include/eunit.hrl").
|
||||
-include("job_enable_common.hrl").
|
||||
|
||||
-define(NUM_BUCKETS, 1200).
|
||||
-define(NUM_KEYS, 1000).
|
||||
-define(LISTKEYS_BUCKET, <<"listkeys_bucket">>).
|
||||
-define(UNDEFINED_BUCKET, <<"880bf69d-5dab-44ee-8762-d24c6f759ce1">>).
|
||||
-define(TEST_ORDER, [true, false]).
|
||||
-define(TEST_OPS, [
|
||||
?TOKEN_LIST_BUCKETS,
|
||||
?TOKEN_LIST_KEYS,
|
||||
?TOKEN_MAP_REDUCE,
|
||||
?TOKEN_SEC_INDEX
|
||||
% ?TOKEN_YZ_SEARCH
|
||||
]).
|
||||
|
||||
%% ===================================================================
|
||||
%% Test API
|
||||
%% ===================================================================
|
||||
|
||||
confirm() ->
|
||||
Config1 = {cuttlefish, [
|
||||
{"async.enable.list_buckets", "on"},
|
||||
{"async.enable.list_keys", "on"}
|
||||
]},
|
||||
Config2 = {cuttlefish, [
|
||||
{"async.enable.list_buckets", "off"},
|
||||
{"async.enable.list_keys", "off"}
|
||||
]},
|
||||
Config3 = {cuttlefish, [
|
||||
{"async.enable.list_buckets", "on"},
|
||||
{"async.enable.list_keys", "off"}
|
||||
]},
|
||||
Config4 = {cuttlefish, [
|
||||
{"async.enable.list_buckets", "off"},
|
||||
{"async.enable.list_keys", "on"}
|
||||
]},
|
||||
Configs = [Config1, Config2, Config3, Config4],
|
||||
Configs = [
|
||||
{current, {cuttlefish,
|
||||
[{"storage_backend", "leveldb"}] ++ config(?TEST_OPS, Bool, [])}}
|
||||
% make it a 4 node cluster so things get scattered around
|
||||
% everything's enabled on the trailing nodes
|
||||
|| Bool <- ?TEST_ORDER ++ [true, true]],
|
||||
|
||||
lager:info("Deploying ~b nodes ...", [erlang:length(Configs)]),
|
||||
Nodes = rt:deploy_nodes(Configs),
|
||||
|
||||
setup_cluster(Nodes),
|
||||
job_enable_common:setup_cluster(Nodes),
|
||||
|
||||
Tests = [
|
||||
{list_buckets, [true, false, true, false]},
|
||||
{list_keys, [true, false, false, true]}
|
||||
],
|
||||
[run_test(Op, Enabled, Nodes) || {Op, Enabled} <- Tests].
|
||||
[run_test(Operation, ?TEST_ORDER, Nodes) || Operation <- ?TEST_OPS],
|
||||
|
||||
run_test(Operation, [Switch | Switches], [Node | Nodes]) ->
|
||||
Enabled = enabled_string(Switch),
|
||||
[begin
|
||||
lager:info("Tesing ~s ~s ~s on ~p", [Operation, Type, Enabled, Node]),
|
||||
test_operation(Node, Operation, Switch, Type)
|
||||
end || Type <- [pbc, http] ],
|
||||
run_test(Operation, Switches, Nodes);
|
||||
run_test(_, [], []) ->
|
||||
ok.
|
||||
|
||||
%% ===================================================================
|
||||
%% Private API
|
||||
%% ===================================================================
|
||||
|
||||
listkeys_bucket() ->
|
||||
?LISTKEYS_BUCKET.
|
||||
|
||||
undefined_bucket() ->
|
||||
?UNDEFINED_BUCKET.
|
||||
|
||||
test_buckets() ->
|
||||
Key = {?MODULE, test_buckets},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
New = binval_list(?NUM_BUCKETS, "Bucket", []),
|
||||
erlang:put(Key, New),
|
||||
New;
|
||||
Val ->
|
||||
Val
|
||||
end.
|
||||
|
||||
test_keys() ->
|
||||
Key = {?MODULE, test_keys},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
New = binval_list(?NUM_KEYS, "Key", []),
|
||||
erlang:put(Key, New),
|
||||
New;
|
||||
Val ->
|
||||
Val
|
||||
end.
|
||||
|
||||
open_client(http = Type, Node) ->
|
||||
% HTTP connections are constant records, so re-use them
|
||||
Key = {?MODULE, httpc, Node},
|
||||
case erlang:get(Key) of
|
||||
undefined ->
|
||||
New = {Type, rhc, rt:httpc(Node)},
|
||||
erlang:put(Key, New),
|
||||
New;
|
||||
Conn ->
|
||||
Conn
|
||||
end;
|
||||
open_client(pbc = Type, Node) ->
|
||||
{Type, riakc_pb_socket, rt:pbc(Node)}.
|
||||
|
||||
close_client({http, _Mod, _RHC}) ->
|
||||
ok;
|
||||
close_client({pbc, Mod, PBC}) ->
|
||||
Mod:stop(PBC).
|
||||
|
||||
setup_cluster([Node | _] = Nodes) ->
|
||||
lager:info("Creating a cluster of ~b nodes ...", [erlang:length(Nodes)]),
|
||||
?assertEqual(ok, rt:join_cluster(Nodes)),
|
||||
|
||||
lager:info("Writing some known data to the clusetr ..."),
|
||||
PB = rt:pbc(Node),
|
||||
lists:foreach(fun(Key) ->
|
||||
riakc_pb_socket:put(PB,
|
||||
riakc_obj:new(?LISTKEYS_BUCKET, Key, Key))
|
||||
end, test_keys()),
|
||||
lists:foreach(fun(Bucket) ->
|
||||
riakc_pb_socket:put(PB,
|
||||
riakc_obj:new(Bucket, <<"test_key">>, <<"test_value">>))
|
||||
end, test_buckets()),
|
||||
riakc_pb_socket:stop(PB),
|
||||
ok.
|
||||
|
||||
test_operation(Node, 'list_buckets', Enabled, ClientType) ->
|
||||
{_, Mod, Conn} = Client = open_client(ClientType, Node),
|
||||
{Status, Result} = Mod:list_buckets(Conn),
|
||||
close_client(Client),
|
||||
case Enabled of
|
||||
true ->
|
||||
?assertEqual(ok, Status),
|
||||
?assertEqual(test_buckets(), lists:sort(Result));
|
||||
false ->
|
||||
Expect = <<"Operation 'list_buckets' is not enabled">>,
|
||||
?assertEqual(error, Status),
|
||||
?assertEqual(Expect, Result)
|
||||
%% case ClientType of
|
||||
%% pbc ->
|
||||
%% ?assertEqual(Expect, Result);
|
||||
%% http ->
|
||||
%% ?assertEqual({"403", Expect}, Result)
|
||||
%% end
|
||||
end;
|
||||
|
||||
test_operation(Node, 'list_keys', Enabled, ClientType) ->
|
||||
{_, Mod, Conn} = Client = open_client(ClientType, Node),
|
||||
{Status, Result} = Mod:list_keys(Conn, ?LISTKEYS_BUCKET),
|
||||
close_client(Client),
|
||||
case Enabled of
|
||||
true ->
|
||||
?assertEqual(ok, Status),
|
||||
?assertEqual(test_keys(), lists:sort(Result));
|
||||
false ->
|
||||
Expect = <<"Operation 'list_keys' is not enabled">>,
|
||||
?assertEqual(error, Status),
|
||||
?assertEqual(Expect, Result)
|
||||
end.
|
||||
pass.
|
||||
|
||||
%% ===================================================================
|
||||
%% Internal
|
||||
%% ===================================================================
|
||||
|
||||
binval_list(0, _, Result) ->
|
||||
Result;
|
||||
binval_list(Count, Prefix, Result) ->
|
||||
binval_list((Count - 1), Prefix, [
|
||||
erlang:list_to_binary([Prefix, erlang:integer_to_list(Count)])
|
||||
| Result]).
|
||||
config([Operation | Operations], Bool, Result) ->
|
||||
Key = lists:flatten(io_lib:format(?CUTTLEFISH_PREFIX ".~s", [Operation])),
|
||||
Val = case Bool of
|
||||
true ->
|
||||
"on";
|
||||
false ->
|
||||
"off"
|
||||
end,
|
||||
config(Operations, Bool, [{Key, Val} | Result]);
|
||||
config([], _, Result) ->
|
||||
Result.
|
||||
|
||||
enabled_string(true) ->
|
||||
"enabled";
|
||||
enabled_string(false) ->
|
||||
"disabled".
|
||||
run_test(Operation, [Switch | Switches], [Node | Nodes]) ->
|
||||
Enabled = job_enable_common:enabled_string(Switch),
|
||||
[begin
|
||||
lager:info("Tesing ~s ~s ~s on ~p", [Operation, Type, Enabled, Node]),
|
||||
job_enable_common:test_operation(Node, Operation, Switch, Type)
|
||||
end || Type <- [pbc, http] ],
|
||||
run_test(Operation, Switches, Nodes);
|
||||
run_test(_, [], _) ->
|
||||
ok.
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user