t/qspawn.t

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