%%
%% %CopyrightBegin%
%%
%% Copyright Ericsson AB 2008-2016. All Rights Reserved.
%%
%% Licensed 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.
%%
%% %CopyrightEnd%
%%
%%%-------------------------------------------------------------------
%%% File : conn_SUITE
%%% Description : Check that the generic connection handling in CT
%%% works as expected.
%%%-------------------------------------------------------------------
-module(conn_SUITE).
%% Note: This directive should only be used in test suites.
-compile(export_all).
-include_lib("common_test/include/ct.hrl").
%%--------------------------------------------------------------------
%% COMMON TEST CALLBACK FUNCTIONS
%%--------------------------------------------------------------------
suite() ->
[{timetrap,{seconds,5}}].
init_per_suite(Config) ->
Config.
end_per_suite(_Config) ->
ok.
init_per_testcase(_TestCase, Config) ->
Config.
end_per_testcase(_TestCase, _Config) ->
ok.
all() ->
[handles_to_multi_conn_pids, handles_to_single_conn_pids,
names_to_multi_conn_pids, names_to_single_conn_pids].
%%--------------------------------------------------------------------
%% TEST CASES
%%--------------------------------------------------------------------
handles_to_multi_conn_pids() ->
[{require,multi_conn_pid}].
handles_to_multi_conn_pids(_Config) ->
application:set_env(ct_test, reconnect, true),
Handle1 = proto:open(multi_conn_pid),
ConnPid1 = ct_gen_conn:get_conn_pid(Handle1),
{true,true} = {is_process_alive(Handle1),is_process_alive(ConnPid1)},
Handle2 = proto:open(multi_conn_pid),
ConnPid2 = ct_gen_conn:get_conn_pid(Handle2),
{true,true} = {is_process_alive(Handle2),is_process_alive(ConnPid2)},
Handle3 = proto:open(multi_conn_pid),
ConnPid3 = ct_gen_conn:get_conn_pid(Handle3),
{true,true} = {is_process_alive(Handle3),is_process_alive(ConnPid3)},
monitor_procs([Handle1,ConnPid1,Handle2,ConnPid2,Handle3,ConnPid3]),
ok = proto:close(Handle1),
ok = wait_procs_down([Handle1,ConnPid1]),
{false,false} = {is_process_alive(Handle1),is_process_alive(ConnPid1)},
{true,true} = {is_process_alive(Handle2),is_process_alive(ConnPid2)},
ok = proto:kill_conn_proc(Handle2),
ok = wait_procs_down([ConnPid2]),
{true,false} = {is_process_alive(Handle2),is_process_alive(ConnPid2)},
ConnPid2x = ct_gen_conn:get_conn_pid(Handle2),
true = is_process_alive(ConnPid2x),
monitor_procs([ConnPid2x]),
ok = proto:close(Handle2),
ok = wait_procs_down([Handle2,ConnPid2x]),
{false,false} = {is_process_alive(Handle2),is_process_alive(ConnPid2x)},
application:set_env(ct_test, reconnect, false),
ok = proto:kill_conn_proc(Handle3),
ok = wait_procs_down([Handle3,ConnPid3]),
{false,false} = {is_process_alive(Handle3),is_process_alive(ConnPid3)},
ok.
handles_to_single_conn_pids() ->
[{require,single_conn_pid}].
handles_to_single_conn_pids(_Config) ->
application:set_env(ct_test, reconnect, true),
Handle1 = proto:open(single_conn_pid),
ConnPid = ct_gen_conn:get_conn_pid(Handle1),
{true,true} = {is_process_alive(Handle1),is_process_alive(ConnPid)},
Handle2 = proto:open(single_conn_pid),
ConnPid = ct_gen_conn:get_conn_pid(Handle2),
{true,true} = {is_process_alive(Handle2),is_process_alive(ConnPid)},
Handle3 = proto:open(single_conn_pid),
ConnPid = ct_gen_conn:get_conn_pid(Handle3),
{true,true} = {is_process_alive(Handle3),is_process_alive(ConnPid)},
Conns = [{undefined,Handle1,_,_},
{undefined,Handle2,_,_},
{undefined,Handle3,_,_}] = lists:sort(ct_util:get_connections(ConnPid)),
ct:pal("CONNS = ~n~p", [Conns]),
monitor_procs([Handle1,Handle2,Handle3,ConnPid]),
ok = proto:close(Handle1),
ok = wait_procs_down([Handle1]),
{false,true} = {is_process_alive(Handle1),is_process_alive(ConnPid)},
ok = proto:kill_conn_proc(Handle2),
ok = wait_procs_down([ConnPid]),
NewConnPid = ct_gen_conn:get_conn_pid(Handle2),
NewConnPid = ct_gen_conn:get_conn_pid(Handle3),
true = is_process_alive(Handle2),
true = is_process_alive(Handle3),
false = is_process_alive(ConnPid),
monitor_procs([NewConnPid]),
ok = proto:close(Handle2),
ok = wait_procs_down([Handle2]),
{false,true} = {is_process_alive(Handle2),is_process_alive(NewConnPid)},
application:set_env(ct_test, reconnect, false),
ok = proto:kill_conn_proc(Handle3),
ok = wait_procs_down([Handle3,NewConnPid]),
{false,false} = {is_process_alive(Handle3),is_process_alive(NewConnPid)},
ok.
names_to_multi_conn_pids() ->
[{require,mconn1,multi_conn_pid},
{require,mconn2,multi_conn_pid},
{require,mconn3,multi_conn_pid}].
names_to_multi_conn_pids(_Config) ->
application:set_env(ct_test, reconnect, true),
Handle1 = proto:open(mconn1),
ConnPid1 = ct_gen_conn:get_conn_pid(Handle1),
{true,true} = {is_process_alive(Handle1),is_process_alive(ConnPid1)},
Handle2 = proto:open(mconn2),
ConnPid2 = ct_gen_conn:get_conn_pid(Handle2),
{true,true} = {is_process_alive(Handle2),is_process_alive(ConnPid2)},
Handle3 = proto:open(mconn3),
ConnPid3 = ct_gen_conn:get_conn_pid(Handle3),
{true,true} = {is_process_alive(Handle3),is_process_alive(ConnPid3)},
Handle1 = proto:open(mconn1),
monitor_procs([Handle1,ConnPid1,Handle2,ConnPid2,Handle3,ConnPid3]),
ok = proto:close(mconn1),
ok = wait_procs_down([Handle1,ConnPid1]),
{false,false} = {is_process_alive(Handle1),is_process_alive(ConnPid1)},
ok = proto:kill_conn_proc(Handle2),
ok = wait_procs_down([ConnPid2]),
Handle2 = proto:open(mconn2), % should've been reconnected already
{true,false} = {is_process_alive(Handle2),is_process_alive(ConnPid2)},
ConnPid2x = ct_gen_conn:get_conn_pid(Handle2),
true = is_process_alive(ConnPid2x),
monitor_procs([ConnPid2x]),
ok = proto:close(mconn2),
ok = wait_procs_down([Handle2,ConnPid2x]),
{false,false} = {is_process_alive(Handle2),is_process_alive(ConnPid2x)},
Handle2y = proto:open(mconn2),
ConnPid2y = ct_gen_conn:get_conn_pid(Handle2y),
{true,true} = {is_process_alive(Handle2y),is_process_alive(ConnPid2y)},
monitor_procs([Handle2y,ConnPid2y]),
ok = proto:close(mconn2),
ok = wait_procs_down([Handle2y,ConnPid2y]),
{false,false} = {is_process_alive(Handle2y),is_process_alive(ConnPid2y)},
application:set_env(ct_test, reconnect, false),
ok = proto:kill_conn_proc(Handle3),
ok = wait_procs_down([Handle3,ConnPid3]),
{false,false} = {is_process_alive(Handle3),is_process_alive(ConnPid3)},
ok.
names_to_single_conn_pids() ->
[{require,sconn1,single_conn_pid},
{require,sconn2,single_conn_pid},
{require,sconn3,single_conn_pid}].
names_to_single_conn_pids(_Config) ->
application:set_env(ct_test, reconnect, true),
Handle1 = proto:open(sconn1),
ConnPid = ct_gen_conn:get_conn_pid(Handle1),
{true,true} = {is_process_alive(Handle1),is_process_alive(ConnPid)},
Handle2 = proto:open(sconn2),
ConnPid = ct_gen_conn:get_conn_pid(Handle2),
{true,true} = {is_process_alive(Handle2),is_process_alive(ConnPid)},
Handle3 = proto:open(sconn3),
ConnPid = ct_gen_conn:get_conn_pid(Handle3),
{true,true} = {is_process_alive(Handle3),is_process_alive(ConnPid)},
Handle1 = proto:open(sconn1),
Conns = [{sconn1,Handle1,_,_},
{sconn2,Handle2,_,_},
{sconn3,Handle3,_,_}] = lists:sort(ct_util:get_connections(ConnPid)),
ct:pal("CONNS on ~p = ~n~p", [ConnPid,Conns]),
monitor_procs([Handle1,Handle2,Handle3,ConnPid]),
ok = proto:close(sconn1),
ok = wait_procs_down([Handle1]),
{false,true} = {is_process_alive(Handle1),is_process_alive(ConnPid)},
ok = proto:kill_conn_proc(Handle2),
ok = wait_procs_down([ConnPid]),
{true,false} = {is_process_alive(Handle2),is_process_alive(ConnPid)},
Handle2 = proto:open(sconn2), % should've been reconnected already
NewConnPid = ct_gen_conn:get_conn_pid(Handle2),
true = is_process_alive(NewConnPid),
monitor_procs([NewConnPid]),
Conns1 = [{sconn2,Handle2,_,_},
{sconn3,Handle3,_,_}] =
lists:sort(ct_util:get_connections(NewConnPid)),
ct:pal("CONNS on ~p = ~n~p", [NewConnPid,Conns1]),
ok = proto:close(sconn2),
ok = wait_procs_down([Handle2]),
{false,true} = {is_process_alive(Handle2),is_process_alive(NewConnPid)},
application:set_env(ct_test, reconnect, false),
ok = proto:kill_conn_proc(Handle3),
ok = wait_procs_down([Handle3,NewConnPid]),
{false,false} = {is_process_alive(Handle3),is_process_alive(NewConnPid)},
ok.
%%%-----------------------------------------------------------------
monitor_procs(Pids) ->
[erlang:monitor(process,Pid) || Pid <- Pids],
ok.
wait_procs_down([]) ->
ok;
wait_procs_down(Pids) ->
receive
{'DOWN',_,process,Pid,_} ->
wait_procs_down(lists:delete(Pid,Pids))
after 2000 ->
timeout
end.
