tests: hold new pg socket open during basebackup

Author: kelvich

Cluster.pm

@@ -6,6 +6,7 @@ use TestLib;
 use Test::More;
 use Cwd;
 use IPC::Run;
+use Socket;
 
 sub new
 {
@@ -248,4 +249,31 @@ sub is_data_identic()
 	return 1;
 }
 
+sub hold_socket()
+{
+	my ($self, $node_off) = @_;
+	my $node = $self->{nodes}->[$node_off];
+	my $iaddr = inet_aton('127.0.0.1');
+	my $paddr = sockaddr_in($node->{_port}, $iaddr);
+	my $proto = getprotobyname("tcp");
+
+	socket(my $sock, PF_INET, SOCK_STREAM, $proto)
+		or die "socket failed: $!";
+
+	# As in postmaster, don't use SO_REUSEADDR on Windows
+	setsockopt($sock, SOL_SOCKET, SO_REUSEADDR, pack("l", 1))
+		unless $TestLib::windows_os;
+	(bind($sock, $paddr) && listen($sock, SOMAXCONN))
+		or die "socket bind and listen failed: $!";
+
+	return $sock;
+}
+
+sub release_socket()
+{
+	my ($self, $sock) = @_;
+	close($sock);
+}
+
+
 1;

t/005_add_stop_node.pl

@@ -56,6 +56,7 @@
 my $pgb2= $cluster->pgbench_async(1, ('-N', '-n', -T => '3600', -c => '2') );
 
 my $new_node_off = $cluster->add_node();
+my $sock = $cluster->hold_socket($new_node_off);
 my $connstr = $cluster->connstr($new_node_off);
 my $new_node_id = $cluster->safe_psql(0, "SELECT mtm.add_node('$connstr')");
 
@@ -65,6 +66,7 @@
 $cluster->pgbench(0, ('-N', '-n', -t => '100') );
 
 my $end_lsn = $cluster->backup_and_init(0, $new_node_off, $new_node_id);
+$cluster->release_socket($sock);
 $cluster->{nodes}->[$new_node_off]->start;
 $cluster->await_nodes( (3,0,1,2) );
 $cluster->safe_psql(0, "SELECT mtm.join_node('$new_node_id', '$end_lsn')");