learn: hoist out remove_or_add subroutine

[public-inbox.git] / t / common.perl

diff --git a/t/common.perl b/t/common.perl
index 5a898e32fceea4c1d1d0f1f55a240257ea3d9f31..ccc7be46099aaa7708cb65126d3400b98102a87a 100644 (file)
--- a/t/common.perl
+++ b/t/common.perl
@@ -1,10 +1,11 @@
-# Copyright (C) 2015-2018 all contributors <meta@public-inbox.org>
+# Copyright (C) 2015-2019 all contributors <meta@public-inbox.org>
 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
 
 use Fcntl qw(FD_CLOEXEC F_SETFD F_GETFD);
 use POSIX qw(dup2);
 use strict;
 use warnings;
+use IO::Socket::INET;
 
 sub stream_to_string {
        my ($res) = @_;
@@ -17,6 +18,39 @@ sub stream_to_string {
        $str;
 }
 
+sub tcp_server () {
+       IO::Socket::INET->new(
+               LocalAddr => '127.0.0.1',
+               ReuseAddr => 1,
+               Proto => 'tcp',
+               Type => Socket::SOCK_STREAM(),
+               Listen => 1024,
+               Blocking => 0,
+       )
+}
+
+sub unix_server ($) {
+       my $s = IO::Socket::UNIX->new(
+               Listen => 1024,
+               Type => Socket::SOCK_STREAM(),
+               Local => $_[0],
+       );
+       $s->blocking(0);
+       $s;
+}
+
+sub tcp_connect {
+       my ($dest, %opt) = @_;
+       my $s = IO::Socket::INET->new(
+               Proto => 'tcp',
+               Type => Socket::SOCK_STREAM(),
+               PeerAddr => $dest->sockhost . ':' . $dest->sockport,
+               %opt,
+       );
+       $s->autoflush(1);
+       $s;
+}
+
 sub spawn_listener {
        my ($env, $cmd, $socks) = @_;
        my $pid = fork;