Fix faulty connection handling

OTP-10126

1 files changed, 240 insertions, 0 deletions

diff --git a/lib/common_test/test/ct_gen_conn_SUITE_data/conn_SUITE.erl b/lib/common_test/test/ct_gen_conn_SUITE_data/conn_SUITE.erl
new file mode 100644
index 0000000000..c0707160cf
--- /dev/null
+++ b/lib/common_test/test/ct_gen_conn_SUITE_data/conn_SUITE.erl
@@ -0,0 +1,240 @@
+%%
+%% %CopyrightBegin%
+%%
+%% Copyright Ericsson AB 2008-2010. All Rights Reserved.
+%%
+%% The contents of this file are subject to the Erlang Public License,
+%% Version 1.1, (the "License"); you may not use this file except in
+%% compliance with the License. You should have received a copy of the
+%% Erlang Public License along with this software. If not, it can be
+%% retrieved online at http://www.erlang.org/.
+%%
+%% Software distributed under the License is distributed on an "AS IS"
+%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
+%% the License for the specific language governing rights and limitations
+%% under the License.
+%%
+%% %CopyrightEnd%
+%%
+
+%%%-------------------------------------------------------------------
+%%% File    : 
+%%% Description : 
+%%%-------------------------------------------------------------------
+-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)},
+    
+    ok = proto:close(Handle1),
+    timer:sleep(100),
+    {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),
+    timer:sleep(100),
+    {true,false} = {is_process_alive(Handle2),is_process_alive(ConnPid2)},
+    ConnPid2x = ct_gen_conn:get_conn_pid(Handle2),
+    true = is_process_alive(ConnPid2x),
+
+    ok = proto:close(Handle2),
+    timer:sleep(100),
+    {false,false} = {is_process_alive(Handle2),is_process_alive(ConnPid2x)},
+
+    application:set_env(ct_test, reconnect, false),
+    ok = proto:kill_conn_proc(Handle3),
+    timer:sleep(100),
+    {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,Handle3,_,_},
+	     {undefined,Handle2,_,_},
+	     {undefined,Handle1,_,_}] = ct_util:get_connections(ConnPid),
+    ct:pal("CONNS = ~n~p", [Conns]),
+
+    ok = proto:close(Handle1),
+    timer:sleep(100),
+    {false,true} = {is_process_alive(Handle1),is_process_alive(ConnPid)},
+
+    ok = proto:kill_conn_proc(Handle2),
+    timer:sleep(100),
+    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),
+
+    ok = proto:close(Handle2),
+    timer:sleep(100),
+    {false,true} = {is_process_alive(Handle2),is_process_alive(NewConnPid)},
+
+    application:set_env(ct_test, reconnect, false),
+    ok = proto:kill_conn_proc(Handle3),
+    timer:sleep(100),
+    {false,false} = {is_process_alive(Handle3),is_process_alive(NewConnPid)},    
+
+    ok.
+
+names_to_multi_conn_pids() ->
+    [{require,conn1,multi_conn_pid},
+     {require,conn2,multi_conn_pid},
+     {require,conn3,multi_conn_pid}].
+
+names_to_multi_conn_pids(Config) ->
+    application:set_env(ct_test, reconnect, true),
+
+    Handle1 = proto:open(conn1),
+    ConnPid1 = ct_gen_conn:get_conn_pid(Handle1),
+    {true,true} = {is_process_alive(Handle1),is_process_alive(ConnPid1)},
+    Handle2 = proto:open(conn2),
+    ConnPid2 = ct_gen_conn:get_conn_pid(Handle2),
+    {true,true} = {is_process_alive(Handle2),is_process_alive(ConnPid2)},
+    Handle3 = proto:open(conn3),
+    ConnPid3 = ct_gen_conn:get_conn_pid(Handle3),
+    {true,true} = {is_process_alive(Handle3),is_process_alive(ConnPid3)},
+
+    Handle1 = proto:open(conn1),
+
+    ok = proto:close(conn1),
+    timer:sleep(100),
+    {false,false} = {is_process_alive(Handle1),is_process_alive(ConnPid1)},
+
+    ok = proto:kill_conn_proc(Handle2),
+    timer:sleep(100),
+    Handle2 = proto:open(conn2),  % 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),
+
+    ok = proto:close(conn2),
+    timer:sleep(100),
+    {false,false} = {is_process_alive(Handle2),is_process_alive(ConnPid2x)},
+    Handle2y = proto:open(conn2),
+    ConnPid2y = ct_gen_conn:get_conn_pid(Handle2y),
+    {true,true} = {is_process_alive(Handle2y),is_process_alive(ConnPid2y)},
+    ok = proto:close(conn2),
+    timer:sleep(100),
+    {false,false} = {is_process_alive(Handle2y),is_process_alive(ConnPid2y)},
+
+    application:set_env(ct_test, reconnect, false),
+    ok = proto:kill_conn_proc(Handle3),
+    timer:sleep(100),
+    {false,false} = {is_process_alive(Handle3),is_process_alive(ConnPid3)},
+
+    ok.
+
+names_to_single_conn_pids() ->
+    [{require,conn1,single_conn_pid},
+     {require,conn2,single_conn_pid},
+     {require,conn3,single_conn_pid}].
+
+names_to_single_conn_pids(Config) ->
+    application:set_env(ct_test, reconnect, true),
+
+    Handle1 = proto:open(conn1),
+    ConnPid = ct_gen_conn:get_conn_pid(Handle1),
+    {true,true} = {is_process_alive(Handle1),is_process_alive(ConnPid)},
+    Handle2 = proto:open(conn2),
+    ConnPid = ct_gen_conn:get_conn_pid(Handle2),
+    {true,true} = {is_process_alive(Handle2),is_process_alive(ConnPid)},
+    Handle3 = proto:open(conn3),
+    ConnPid = ct_gen_conn:get_conn_pid(Handle3),
+    {true,true} = {is_process_alive(Handle3),is_process_alive(ConnPid)},
+
+    Handle1 = proto:open(conn1),
+
+    Conns = [{conn1,Handle1,_,_},
+	     {conn2,Handle2,_,_},
+	     {conn3,Handle3,_,_}] = lists:sort(ct_util:get_connections(ConnPid)),
+    ct:pal("CONNS on ~p = ~n~p", [ConnPid,Conns]),
+
+    ok = proto:close(conn1),
+    timer:sleep(100),
+    {false,true} = {is_process_alive(Handle1),is_process_alive(ConnPid)},
+
+    ok = proto:kill_conn_proc(Handle2),
+    timer:sleep(100),
+    {true,false} = {is_process_alive(Handle2),is_process_alive(ConnPid)},
+    Handle2 = proto:open(conn2),  % should've been reconnected already
+    NewConnPid = ct_gen_conn:get_conn_pid(Handle2),
+    true = is_process_alive(NewConnPid),
+    
+    Conns1 = [{conn2,Handle2,_,_},
+	      {conn3,Handle3,_,_}] = 
+	lists:sort(ct_util:get_connections(NewConnPid)),
+    ct:pal("CONNS on ~p = ~n~p", [NewConnPid,Conns1]),
+
+    ok = proto:close(conn2),
+    timer:sleep(100),
+    {false,true} = {is_process_alive(Handle2),is_process_alive(NewConnPid)},
+
+    application:set_env(ct_test, reconnect, false),
+    ok = proto:kill_conn_proc(Handle3),
+    timer:sleep(100),
+    {false,false} = {is_process_alive(Handle3),is_process_alive(NewConnPid)},
+
+    ok.
+
+