delroth/hydra
0
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

notifications.t: create a helper to parse the actual messages from the evaluator using the Event code

Browse source
This commit is contained in:
Graham Christensen 2022-02-07 14:59:19 -05:00
parent c30f084f32
commit be531c6c57
1 changed files with 22 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
t/scripts/hydra-eval-jobset/notifications.t
View file

@ -5,6 +5,24 @@ use Setup;
use Test2::V0;
use File::Copy;
use Hydra::PostgresListener;
use Hydra::Event;
# expectEvent(Hydra::PostgresLister, name of the channel to expect, a sub which gets the parsed event)
sub expectEvent {
my ($listener, $expectedChannel, $then) = @_;
my $message = $listener->block_for_messages(0)->();
my $channel = $message->{"channel"};
if ($channel eq $expectedChannel) {
my $event = Hydra::Event->new_event($message->{"channel"}, $message->{"payload"});
local $_ = $event->{event};
$then->();
} else {
is($expectedChannel, $channel, "Expecting a message on channel $channel");
}
}
my $ctx = test_context(
hydra_config => q|
@ -38,15 +56,11 @@ my $builds = $ctx->makeAndEvaluateJobset(
);
my $jobset = $builds->{"stable-job-queued"}->jobset;
my $traceID;
subtest "on the initial evaluation" => sub {
my $startedMsg = $listener->block_for_messages(0)->();
is($startedMsg->{"channel"}, "eval_started", "every eval starts with a notification");
my ($traceID, $jobsetID) = split("\t", $startedMsg->{"payload"});
isnt($traceID, "", "we got a trace id");
is($jobsetID, $jobset->get_column('id'), "the jobset ID matches");
expectEvent($listener, "eval_started", sub {
isnt($_->{"trace_id"}, "", "We got a trace ID");
is($_->{"jobset_id"}, $jobset->get_column('id'), "the jobset ID matches");
});
is($listener->block_for_messages(0)->()->{"channel"}, "build_queued", "expect 1/4 builds being queued");
is($listener->block_for_messages(0)->()->{"channel"}, "build_queued", "expect 2/4 builds being queued");