mirror of
https://github.com/valitydev/riak_test.git
synced 2024-11-06 16:45:29 +00:00
61 lines
2.1 KiB
Erlang
61 lines
2.1 KiB
Erlang
%% -------------------------------------------------------------------
|
|
%%
|
|
%% 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, State,
|
|
?DONT_INCREMENT_PROMPT),
|
|
lager:info("Result is ~p~n", [Result]),
|
|
lager:info("~n~n------------------------------------------------------", []),
|
|
Pid ! Result.
|
|
|