X-Git-Url: http://www.git.stargrave.org/?a=blobdiff_plain;f=t%2Fcommon.perl;h=ccc7be46099aaa7708cb65126d3400b98102a87a;hb=29beb70160d15ffe46cb07e9d9468acfcda34db7;hp=bec57699de226ffa7405c1cdedcbc4c567219bec;hpb=f76f265a851944b5dedcc3be5f3b5224b6ebda89;p=public-inbox.git

diff --git a/t/common.perl b/t/common.perl
index bec57699..ccc7be46 100644
--- a/t/common.perl
+++ b/t/common.perl
@@ -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;