t/qspawn.t

   1 # Copyright (C) all contributors <meta@public-inbox.org>
   2 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
   3 use v5.12;
   4 use Test::More;
   5 use_ok 'PublicInbox::Qspawn';
   6 use_ok 'PublicInbox::Limiter';
   7
   8 {
   9         my $cmd = [qw(sh -c), 'echo >&2 err; echo out'];
  10         my $qsp = PublicInbox::Qspawn->new($cmd, {}, { 2 => 1 });
  11         my $res;
  12         $qsp->psgi_qx({}, undef, sub { $res = ${$_[0]} });
  13         is($res, "err\nout\n", 'captured stderr and stdout');
  14
  15         $res = undef;
  16         $qsp = PublicInbox::Qspawn->new($cmd, {}, { 2 => \*STDOUT });
  17         $qsp->psgi_qx({}, undef, sub { $res = ${$_[0]} });
  18         is($res, "err\nout\n", 'captured stderr and stdout');
  19 }
  20
  21 sub finish_err ($) {
  22         my ($qsp) = @_;
  23         $qsp->finish;
  24         $qsp->{qsp_err} && ${$qsp->{qsp_err}};
  25 }
  26
  27 my $limiter = PublicInbox::Limiter->new(1);
  28 {
  29         my $x = PublicInbox::Qspawn->new([qw(true)]);
  30         $x->{qsp_err} = \(my $err = '');
  31         my $run = 0;
  32         $x->start($limiter, sub {
  33                 my ($self) = @_;
  34                 is(0, sysread($self->{rpipe}, my $buf, 1), 'read zero bytes');
  35                 ok(!finish_err($self), 'no error on finish');
  36                 $run = 1;
  37         });
  38         is($run, 1, 'callback ran alright');
  39 }
  40
  41 {
  42         my @err; local $SIG{__WARN__} = sub { push @err, @_ };
  43         my $x = PublicInbox::Qspawn->new([qw(false)]);
  44         $x->{qsp_err} = \(my $err = '');
  45         my $run = 0;
  46         $x->start($limiter, sub {
  47                 my ($self) = @_;
  48                 is(0, sysread($self->{rpipe}, my $buf, 1),
  49                                 'read zero bytes from false');
  50                 ok(finish_err($self), 'error on finish');
  51                 $run = 1;
  52         });
  53         is($run, 1, 'callback ran alright');
  54         ok(scalar @err, 'got warning');
  55 }
  56
  57 foreach my $cmd ([qw(sleep 1)], [qw(sh -c), 'sleep 1; false']) {
  58         my @err; local $SIG{__WARN__} = sub { push @err, @_ };
  59         my $s = PublicInbox::Qspawn->new($cmd);
  60         $s->{qsp_err} = \(my $err = '');
  61         my @run;
  62         $s->start($limiter, sub {
  63                 my ($self) = @_;
  64                 push @run, 'sleep';
  65                 is(0, sysread($self->{rpipe}, my $buf, 1), 'read zero bytes');
  66         });
  67         my $n = 0;
  68         my @t = map {
  69                 my $i = $n++;
  70                 my $x = PublicInbox::Qspawn->new([qw(true)]);
  71                 $x->start($limiter, sub {
  72                         my ($self) = @_;
  73                         push @run, $i;
  74                 });
  75                 [$x, $i]
  76         } (0..2);
  77
  78         if ($cmd->[-1] =~ /false\z/) {
  79                 ok(finish_err($s), 'got error on false after sleep');
  80                 ok(scalar @err, 'got warning');
  81         } else {
  82                 ok(!finish_err($s), 'no error on sleep');
  83                 is_deeply([], \@err, 'no warnings');
  84         }
  85         ok(!finish_err($_->[0]), "true $_->[1] succeeded") foreach @t;
  86         is_deeply([qw(sleep 0 1 2)], \@run, 'ran in order');
  87 }
  88
  89 done_testing();
  90
  91 1;