t/10scheduler.t

   1 #!/usr/bin/perl
   2 use strict;
   3 use warnings;
   4 use Test::Simple tests => 3;
   5
   6 use Fedora::Rebuild::Scheduler;
   7
   8 sub eqwithundef {
   9     my ($a, $b) = @_;
  10     if (defined $a and defined $b and $a eq $b) {
  11         return 1;
  12     }
  13     if (!defined $a and !defined $b) {
  14         return 1;
  15     }
  16     return 0;
  17 }
  18
  19 sub teststatus {
  20     my ($message, $expected_exception, $expected_retval, $status) = @_;
  21
  22     ok(
  23         (
  24             eqwithundef($expected_exception, $$status[1]) and
  25             eqwithundef($expected_retval, $$status[0])
  26         ),
  27         $message . ', got: exception=' . ($$status[1]//'<undef>') .
  28             ' retval=' . ($$status[0]//'<undef>')
  29     );
  30 }
  31
  32 my $scheduler = Fedora::Rebuild::Scheduler->new(1);
  33
  34 sub worker {
  35     my $arg = shift;
  36     if ($arg < 0) {
  37         die("Dying worker as requested\n");
  38     }
  39     return $arg;
  40 }
  41
  42 my $job_to_die = $scheduler->schedule(\&worker, -1);
  43 my $job_to_succeed = $scheduler->schedule(\&worker, 1);
  44 my $job_to_fail = $scheduler->schedule(\&worker, 0);
  45
  46 my %finished = $scheduler->finish();
  47
  48 teststatus('First job should die', "Dying worker as requested\n", undef,
  49     $finished{$job_to_die});
  50 teststatus('Second job should succeed', "", 1,
  51     $finished{$job_to_succeed});
  52 teststatus('Third job should fail', "", 0,
  53     $finished{$job_to_fail});