108 lines
3.6 KiB
Perl
108 lines
3.6 KiB
Perl
use strict;
|
|
use warnings;
|
|
use Setup;
|
|
|
|
my %ctx = test_init();
|
|
|
|
use Hydra::Event;
|
|
use Hydra::Task;
|
|
require Hydra::Schema;
|
|
require Hydra::Model::DB;
|
|
|
|
use Test2::V0;
|
|
|
|
my $db = Hydra::Model::DB->new;
|
|
hydra_setup($db);
|
|
|
|
my $taskretries = $db->resultset('TaskRetries');
|
|
|
|
subtest "get_seconds_to_next_retry" => sub {
|
|
subtest "Without any records in the database" => sub {
|
|
is($taskretries->get_seconds_to_next_retry(), undef, "Without any records our next retry moment is forever away.");
|
|
};
|
|
|
|
subtest "With only tasks whose retry timestamps are in the future" => sub {
|
|
$taskretries->create({
|
|
channel => "bogus",
|
|
pluginname => "bogus",
|
|
payload => "bogus",
|
|
attempts => 1,
|
|
retry_at => time() + 100,
|
|
});
|
|
is($taskretries->get_seconds_to_next_retry(), within(100, 2), "We should retry in roughly 100 seconds");
|
|
};
|
|
|
|
subtest "With tasks whose retry timestamp are in the past" => sub {
|
|
$taskretries->create({
|
|
channel => "bogus",
|
|
pluginname => "bogus",
|
|
payload => "bogus",
|
|
attempts => 1,
|
|
retry_at => time() - 100,
|
|
});
|
|
is($taskretries->get_seconds_to_next_retry(), 0, "We should retry immediately");
|
|
};
|
|
|
|
$taskretries->delete_all();
|
|
};
|
|
|
|
subtest "get_retryable_taskretries_row" => sub {
|
|
subtest "Without any records in the database" => sub {
|
|
is($taskretries->get_retryable_taskretries_row(), undef, "Without any records we have no tasks to retry.");
|
|
is($taskretries->get_retryable_task(), undef, "Without any records we have no tasks to retry.");
|
|
};
|
|
|
|
subtest "With only tasks whose retry timestamps are in the future" => sub {
|
|
$taskretries->create({
|
|
channel => "bogus",
|
|
pluginname => "bogus",
|
|
payload => "bogus",
|
|
attempts => 1,
|
|
retry_at => time() + 100,
|
|
});
|
|
is($taskretries->get_retryable_taskretries_row(), undef, "We still have nothing to do");
|
|
is($taskretries->get_retryable_task(), undef, "We still have nothing to do");
|
|
};
|
|
|
|
subtest "With tasks whose retry timestamp are in the past" => sub {
|
|
$taskretries->create({
|
|
channel => "build_started",
|
|
pluginname => "bogus plugin",
|
|
payload => "123",
|
|
attempts => 1,
|
|
retry_at => time() - 100,
|
|
});
|
|
|
|
my $row = $taskretries->get_retryable_taskretries_row();
|
|
isnt($row, undef, "We should retry immediately");
|
|
is($row->channel, "build_started", "Channel name should match");
|
|
is($row->pluginname, "bogus plugin", "Plugin name should match");
|
|
is($row->payload, "123", "Payload should match");
|
|
is($row->attempts, 1, "We've had one attempt");
|
|
|
|
my $task = $taskretries->get_retryable_task();
|
|
is($task->{"event"}->{"channel_name"}, "build_started");
|
|
is($task->{"plugin_name"}, "bogus plugin");
|
|
is($task->{"event"}->{"payload"}, "123");
|
|
is($task->{"record"}->get_column("id"), $row->get_column("id"));
|
|
};
|
|
};
|
|
|
|
subtest "save_task" => sub {
|
|
my $event = Hydra::Event->new_event("build_started", "1");
|
|
my $task = Hydra::Task->new(
|
|
$event,
|
|
"FooPluginName",
|
|
);
|
|
|
|
my $retry = $taskretries->save_task($task);
|
|
|
|
is($retry->channel, "build_started", "Channel name should match");
|
|
is($retry->pluginname, "FooPluginName", "Plugin name should match");
|
|
is($retry->payload, "1", "Payload should match");
|
|
is($retry->attempts, 1, "We've had one attempt");
|
|
is($retry->retry_at, within(time() + 1, 5), "The retry at should be approximately one second away");
|
|
};
|
|
|
|
done_testing;
|