root/trunk/t/retry-delay.t @ 164

# $Id$
# -*-perl-*-

use strict;
use warnings;

require 't/lib/db-common.pl';

use TheSchwartz;
use Test::More tests => 24;

run_tests(8, sub {
    my $client = test_client(dbs => ['ts1']);

    # insert a job which will fail, fail, then succeed.
    {
        my $handle = $client->insert("Worker::CompleteEventually");
        isa_ok $handle, 'TheSchwartz::JobHandle', "inserted job";

        $client->can_do("Worker::CompleteEventually");
        $client->work_until_done;

        is($handle->failures, 1, "job has failed once");

        my $job = Worker::CompleteEventually->grab_job($client);
        ok(!$job, "a job isn't ready yet"); # hasn't been two seconds
        sleep 3;   # 2 seconds plus 1 buffer second

        $job = Worker::CompleteEventually->grab_job($client);
        ok($job, "got a job, since time has gone by");

        Worker::CompleteEventually->work_safely($job);
        is($handle->failures, 2, "job has failed twice");

        $job = Worker::CompleteEventually->grab_job($client);
        ok($job, "got the job back");

        Worker::CompleteEventually->work_safely($job);
        ok(! $handle->is_pending, "job has exited");
        is($handle->exit_status, 0, "job succeeded");
    }

    teardown_dbs('ts1');
});

############################################################################
package Worker::CompleteEventually;
use base 'TheSchwartz::Worker';

sub work {
    my ($class, $job) = @_;
    my $failures = $job->failures;
    if ($failures < 2) {
        $job->failed;
    } else {
        $job->completed;
    }
    return;
}

sub keep_exit_status_for { 20 }  # keep exit status for 20 seconds after on_complete

sub max_retries { 2 }

sub retry_delay {
    my $class = shift;
    my $fails = shift;
    return [undef,2,0]->[$fails];  # fails 2 seconds first time, then immediately
}