]> Sergey Matveev's repositories - public-inbox.git/blobdiff - t/common.perl
learn: hoist out remove_or_add subroutine
[public-inbox.git] / t / common.perl
index bec57699de226ffa7405c1cdedcbc4c567219bec..ccc7be46099aaa7708cb65126d3400b98102a87a 100644 (file)
@@ -1,18 +1,92 @@
-# Copyright (C) 2015 all contributors <meta@public-inbox.org>
-# License: AGPLv3 or later (https://www.gnu.org/licenses/agpl-3.0.txt)
-require IO::File;
-use POSIX qw/dup/;
+# 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 ($cb) = @_;
-       my $headers;
-       my $io = IO::File->new_tmpfile;
-       my $dup = dup($io->fileno);
-       my $response = sub { $headers = \@_, $io };
-       $cb->($response);
-       $io = IO::File->new;
-       $io->fdopen($dup, 'r+');
-       $io->seek(0, 0);
-       $io->read(my $str, ($io->stat)[7]);
+       my ($res) = @_;
+       my $body = $res->[2];
+       my $str = '';
+       while (defined(my $chunk = $body->getline)) {
+               $str .= $chunk;
+       }
+       $body->close;
        $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;
+       defined $pid or die "fork failed: $!\n";
+       if ($pid == 0) {
+               # pretend to be systemd (cf. sd_listen_fds(3))
+               my $fd = 3; # 3 == SD_LISTEN_FDS_START
+               foreach my $s (@$socks) {
+                       my $fl = fcntl($s, F_GETFD, 0);
+                       if (($fl & FD_CLOEXEC) != FD_CLOEXEC) {
+                               warn "got FD:".fileno($s)." w/o CLOEXEC\n";
+                       }
+                       fcntl($s, F_SETFD, $fl &= ~FD_CLOEXEC);
+                       dup2(fileno($s), $fd++) or die "dup2 failed: $!\n";
+               }
+               $ENV{LISTEN_PID} = $$;
+               $ENV{LISTEN_FDS} = scalar @$socks;
+               %ENV = (%ENV, %$env) if $env;
+               exec @$cmd;
+               die "FAIL: ",join(' ', @$cmd), ": $!\n";
+       }
+       $pid;
+}
+
+sub require_git ($;$) {
+       my ($req, $maybe) = @_;
+       my ($req_maj, $req_min) = split(/\./, $req);
+       my ($cur_maj, $cur_min) = (`git --version` =~ /version (\d+)\.(\d+)/);
+
+       my $req_int = ($req_maj << 24) | ($req_min << 16);
+       my $cur_int = ($cur_maj << 24) | ($cur_min << 16);
+       if ($cur_int < $req_int) {
+               return 0 if $maybe;
+               plan skip_all => "git $req+ required, have $cur_maj.$cur_min";
+       }
+       1;
+}
+
+1;