mirror of
https://github.com/valitydev/riak_test.git
synced 2024-11-06 08:35:22 +00:00
tests for riak_shell
This commit is contained in:
Gordon Guthrie
parent
e5e8109ce6
commit
03a0bb95c9
59
tests/riak_shell_test_connecting.erl
Normal file
59
tests/riak_shell_test_connecting.erl
Normal file
@ -0,0 +1,59 @@
|
||||
%% -------------------------------------------------------------------
|
||||
%%
|
||||
%% 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(riak_shell_test_connecting).
|
||||
|
||||
-behavior(riak_test).
|
||||
|
||||
-include_lib("eunit/include/eunit.hrl").
|
||||
|
||||
-export([confirm/0]).
|
||||
|
||||
-define(DONT_INCREMENT_PROMPT, false).
|
||||
|
||||
%% we cant run the test in this process as it receives various messages
|
||||
%% and the running test interprets then as being messages to the shell
|
||||
confirm() ->
|
||||
Nodes = riak_shell_test_util:build_cluster(),
|
||||
lager:info("Built a cluster of ~p~n", [Nodes]),
|
||||
Self = self(),
|
||||
_Pid = spawn_link(fun() -> run_test(Self) end),
|
||||
riak_shell_test_util:loop().
|
||||
|
||||
run_test(Pid) ->
|
||||
State = riak_shell_test_util:shell_init(),
|
||||
lager:info("~n~nStart running the command set-------------------------", []),
|
||||
Cmds = [
|
||||
%% 'connection prompt on' means you need to do unicode printing and stuff
|
||||
{run,
|
||||
"connection_prompt off;"},
|
||||
{run,
|
||||
"show_cookie;"},
|
||||
{run,
|
||||
"show_connection;"},
|
||||
{run,
|
||||
"connect 'dev1@127.0.0.1';"},
|
||||
{{match, "riak_shell is connected to: 'dev1@127.0.0.1' on port 10017"},
|
||||
"show_connection;"}
|
||||
],
|
||||
Result = riak_shell_test_util:run_commands(Cmds, "Start", State,
|
||||
?DONT_INCREMENT_PROMPT),
|
||||
lager:info("Result is ~p~n", [Result]),
|
||||
lager:info("~n~n------------------------------------------------------", []),
|
||||
Pid ! Result.
|
||||
57
tests/riak_shell_test_connecting_error.erl
Normal file
57
tests/riak_shell_test_connecting_error.erl
Normal file
@ -0,0 +1,57 @@
|
||||
%% -------------------------------------------------------------------
|
||||
%%
|
||||
%% 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(riak_shell_test_connecting_error).
|
||||
|
||||
-behavior(riak_test).
|
||||
|
||||
-include_lib("eunit/include/eunit.hrl").
|
||||
|
||||
-export([confirm/0]).
|
||||
|
||||
-define(DONT_INCREMENT_PROMPT, false).
|
||||
|
||||
%% we cant run the test in this process as it receives various messages
|
||||
%% and the running test interprets then as being messages to the shell
|
||||
confirm() ->
|
||||
Nodes = riak_shell_test_util:build_cluster(),
|
||||
lager:info("Built a cluster of ~p~n", [Nodes]),
|
||||
Self = self(),
|
||||
_Pid = spawn_link(fun() -> run_test(Self) end),
|
||||
riak_shell_test_util:loop().
|
||||
|
||||
run_test(Pid) ->
|
||||
State = riak_shell_test_util:shell_init(),
|
||||
lager:info("~n~nStart running the command set-------------------------", []),
|
||||
Cmds = [
|
||||
%% 'connection prompt on' means you need to do unicode printing and stuff
|
||||
{run,
|
||||
"connection_prompt off;"},
|
||||
{run,
|
||||
"show_cookie;"},
|
||||
{run,
|
||||
"show_connection;"},
|
||||
{{match, "Connection to 'made up guff' failed"},
|
||||
"connect 'made up guff';"}
|
||||
],
|
||||
Result = riak_shell_test_util:run_commands(Cmds, "Start", State,
|
||||
?DONT_INCREMENT_PROMPT),
|
||||
lager:info("Result is ~p~n", [Result]),
|
||||
lager:info("~n~n------------------------------------------------------", []),
|
||||
Pid ! Result.
|
||||
72
tests/riak_shell_test_disconnecting.erl
Normal file
72
tests/riak_shell_test_disconnecting.erl
Normal file
@ -0,0 +1,72 @@
|
||||
%% -------------------------------------------------------------------
|
||||
%%
|
||||
%% 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(riak_shell_test_disconnecting).
|
||||
|
||||
-behavior(riak_test).
|
||||
|
||||
-include_lib("eunit/include/eunit.hrl").
|
||||
|
||||
-export([confirm/0]).
|
||||
|
||||
-define(DONT_INCREMENT_PROMPT, false).
|
||||
|
||||
%% we cant run the test in this process as it receives various messages
|
||||
%% and the running test interprets then as being messages to the shell
|
||||
confirm() ->
|
||||
Nodes = riak_shell_test_util:build_cluster(),
|
||||
lager:info("Built a cluster of ~p~n", [Nodes]),
|
||||
Self = self(),
|
||||
_Pid = spawn_link(fun() -> run_test(Self) end),
|
||||
riak_shell_test_util:loop().
|
||||
|
||||
run_test(Pid) ->
|
||||
State = riak_shell_test_util:shell_init(),
|
||||
lager:info("~n~nStart running the command set-------------------------", []),
|
||||
Cmds = [
|
||||
%% 'connection prompt on' means you need to do unicode printing and stuff
|
||||
{run,
|
||||
"connection_prompt off;"},
|
||||
{run,
|
||||
"show_cookie;"},
|
||||
{run,
|
||||
"show_connection;"},
|
||||
{stop_node,
|
||||
'dev1@127.0.0.1'},
|
||||
{drain,
|
||||
"Connected..."},
|
||||
{{match, "riak_shell is connected to: 'dev2@127.0.0.1' on port 10027"},
|
||||
"show_connection;"},
|
||||
{start_node,
|
||||
'dev1@127.0.0.1'},
|
||||
{drain,
|
||||
discard},
|
||||
{run,
|
||||
"reconnect;"},
|
||||
{drain,
|
||||
"Reconnected to 'dev1@127.0.0.1' on port 10017"},
|
||||
{{match, "riak_shell is connected to: 'dev1@127.0.0.1' on port 10017"},
|
||||
"show_connection;"}
|
||||
],
|
||||
Result = riak_shell_test_util:run_commands(Cmds, "Start", State,
|
||||
?DONT_INCREMENT_PROMPT),
|
||||
lager:info("Result is ~p~n", [Result]),
|
||||
lager:info("~n~n------------------------------------------------------", []),
|
||||
Pid ! Result.
|
||||
|
||||
60
tests/riak_shell_test_reconnecting.erl
Normal file
60
tests/riak_shell_test_reconnecting.erl
Normal file
@ -0,0 +1,60 @@
|
||||
%% -------------------------------------------------------------------
|
||||
%%
|
||||
%% 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(riak_shell_test_reconnecting).
|
||||
|
||||
-behavior(riak_test).
|
||||
|
||||
-include_lib("eunit/include/eunit.hrl").
|
||||
|
||||
-export([confirm/0]).
|
||||
|
||||
-define(DONT_INCREMENT_PROMPT, false).
|
||||
|
||||
%% we cant run the test in this process as it receives various messages
|
||||
%% and the running test interprets then as being messages to the shell
|
||||
confirm() ->
|
||||
Nodes = riak_shell_test_util:build_cluster(),
|
||||
lager:info("Built a cluster of ~p~n", [Nodes]),
|
||||
Self = self(),
|
||||
_Pid = spawn_link(fun() -> run_test(Self) end),
|
||||
riak_shell_test_util:loop().
|
||||
|
||||
run_test(Pid) ->
|
||||
State = riak_shell_test_util:shell_init(),
|
||||
lager:info("~n~nStart running the command set-------------------------", []),
|
||||
Cmds = [
|
||||
%% 'connection prompt on' means you need to do unicode printing and stuff
|
||||
{run,
|
||||
"connection_prompt off;"},
|
||||
{run,
|
||||
"show_cookie;"},
|
||||
{run,
|
||||
"show_connection;"},
|
||||
{run,
|
||||
"reconnect;"},
|
||||
{{match, "riak_shell is connected to: 'dev1@127.0.0.1' on port 10017"},
|
||||
"show_connection;"}
|
||||
],
|
||||
Result = riak_shell_test_util:run_commands(Cmds, "Start", State,
|
||||
?DONT_INCREMENT_PROMPT),
|
||||
lager:info("Result is ~p~n", [Result]),
|
||||
lager:info("~n~n------------------------------------------------------", []),
|
||||
Pid ! Result.
|
||||
|
||||
129
tests/riak_shell_test_util.erl
Normal file
129
tests/riak_shell_test_util.erl
Normal file
@ -0,0 +1,129 @@
|
||||
%% -------------------------------------------------------------------
|
||||
%%
|
||||
%% 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(riak_shell_test_util).
|
||||
|
||||
-define(CLUSTERSIZE, 2).
|
||||
-define(EMPTYCONFIG, []).
|
||||
|
||||
-export([
|
||||
shell_init/0,
|
||||
run_commands/4,
|
||||
build_cluster/0,
|
||||
loop/0
|
||||
]).
|
||||
|
||||
-define(PREFIX, "Riak-shell tests: ").
|
||||
|
||||
shell_init() ->
|
||||
Config = [
|
||||
{logging, off},
|
||||
{cookie, riak},
|
||||
{nodes, [
|
||||
'dev1@127.0.0.1',
|
||||
'dev2@127.0.0.1',
|
||||
'dev3@127.0.0.1',
|
||||
'dev4@127.0.0.1',
|
||||
'dev5@127.0.0.1',
|
||||
'dev6@127.0.0.1',
|
||||
'dev7@127.0.0.1',
|
||||
'dev8@127.0.0.1'
|
||||
]}
|
||||
],
|
||||
State = riak_shell_app:boot_TEST(Config),
|
||||
receive
|
||||
%% after initialising the riak_shell gets a succeed/fail connection
|
||||
%% message - we need to drain that msg here if we want the runner
|
||||
%% to work, or it will be off-by-1 in the test runner
|
||||
ConnectionMsg ->
|
||||
lager:info("Got a connection message ~p on shell_init",
|
||||
[ConnectionMsg])
|
||||
end,
|
||||
State.
|
||||
|
||||
build_cluster() ->
|
||||
rt:set_backend(eleveldb),
|
||||
_Nodes = rt:build_cluster(?CLUSTERSIZE, ?EMPTYCONFIG).
|
||||
|
||||
run_commands([], _Msg, _State, _ShouldIncrement) ->
|
||||
pass;
|
||||
run_commands([{drain, discard} | T], Msg, State, ShouldIncrement) ->
|
||||
{NewMsg, NewState, NewShdIncr} = riak_shell:loop_TEST(Msg, State, ShouldIncrement),
|
||||
lager:info("Message drained and discared unchecked ~p", [lists:flatten(NewMsg)]),
|
||||
run_commands(T, Msg, NewState, NewShdIncr);
|
||||
run_commands([{drain, Response} | T], Msg, State, ShouldIncrement) ->
|
||||
{NewMsg, NewState, NewShdIncr} = riak_shell:loop_TEST(Msg, State, ShouldIncrement),
|
||||
case lists:flatten(NewMsg) of
|
||||
Response -> lager:info("Message drained successfully ~p", [Response]),
|
||||
run_commands(T, Msg, NewState, NewShdIncr);
|
||||
Got -> print_error("Message Expected", "", Response, Got),
|
||||
fail
|
||||
end;
|
||||
run_commands([{start_node, Node} | T], Msg, State, ShouldIncrement) ->
|
||||
rt:start(Node),
|
||||
rt:wait_until_pingable(Node),
|
||||
run_commands(T, Msg, State, ShouldIncrement);
|
||||
run_commands([{stop_node, Node} | T], Msg, State, ShouldIncrement) ->
|
||||
rt:stop(Node),
|
||||
rt:wait_until_unpingable(Node),
|
||||
run_commands(T, Msg, State, ShouldIncrement);
|
||||
run_commands([sleep | T], Msg, State, ShouldIncrement) ->
|
||||
timer:sleep(1000),
|
||||
run_commands(T, Msg, State, ShouldIncrement);
|
||||
run_commands([{{match, Expected}, Cmd} | T], Msg, State, ShouldIncrement) ->
|
||||
{NewMsg, NewState, NewShdIncr} = run_cmd(Cmd, Msg, State, ShouldIncrement),
|
||||
%% when you start getting off-by-1 wierdness you will WANT to uncomment this
|
||||
case lists:flatten(NewMsg) of
|
||||
Expected -> lager:info("Successful match of ~p from ~p", [Expected, Cmd]),
|
||||
run_commands(T, NewMsg, NewState, NewShdIncr);
|
||||
Got -> print_error("Ran ~p:", Cmd, Expected, Got),
|
||||
fail
|
||||
end;
|
||||
run_commands([{run, Cmd} | T], Msg, State, ShouldIncrement) ->
|
||||
lager:info("Run command: ~p", [Cmd]),
|
||||
{NewMsg, NewState, NewShdIncr} = run_cmd(Cmd, Msg, State, ShouldIncrement),
|
||||
run_commands(T, NewMsg, NewState, NewShdIncr).
|
||||
|
||||
run_cmd(Cmd, Msg, State, ShouldIncrement) ->
|
||||
%% the riak-shell works by spawning a process that has captured
|
||||
%% standard input and then dropping into a receive that the spawned
|
||||
%% process sends a message to
|
||||
%% we have to emulate that here as we are the shell
|
||||
%% we are going to send a message at some time in the future
|
||||
%% and then go into a loop waiting for it
|
||||
timer:apply_after(500, riak_shell, send_to_shell, [self(), {command, Cmd}]),
|
||||
riak_shell:loop_TEST(Msg, State, ShouldIncrement).
|
||||
|
||||
print_error(Format, Cmd, Expected, Got) ->
|
||||
lager:info(?PREFIX ++ "Match Failure"),
|
||||
lager:info("**************************************************************"),
|
||||
lager:info(Format, [Cmd]),
|
||||
lager:info("Exp: ~s", [Expected]),
|
||||
lager:info("Got: " ++ Got, []),
|
||||
lager:info("**************************************************************").
|
||||
|
||||
loop() ->
|
||||
Return = receive
|
||||
pass -> pass;
|
||||
fail -> exit(fail);
|
||||
Other -> io:format("Got message ~p~n", [Other]),
|
||||
loop()
|
||||
end,
|
||||
application:stop(riak_shell),
|
||||
Return.
|
||||
Loading…
Reference in New Issue
Block a user