124 lines
3.2 KiB
Perl
124 lines
3.2 KiB
Perl
#!/usr/bin/env perl
|
||
|
||
use strict; use warnings;
|
||
|
||
BEGIN { $ENV{DANCER_ENVDIR} = '/dev/null'; }
|
||
|
||
use Test::More 0.88;
|
||
use Test::File::ShareDir::Dist { 'App-Netdisco' => 'share/' };
|
||
|
||
use lib 'xt/lib';
|
||
|
||
use App::Netdisco;
|
||
use App::Netdisco::DB; # fake device row
|
||
use App::Netdisco::Backend::Job;
|
||
|
||
use Try::Tiny;
|
||
use Dancer qw/:moose :script !pass/;
|
||
|
||
# configure logging to force console output
|
||
my $CONFIG = config();
|
||
$CONFIG->{logger} = 'console';
|
||
$CONFIG->{log} = 'error';
|
||
Dancer::Logger->init('console', $CONFIG);
|
||
|
||
{
|
||
package MyWorker;
|
||
use Moo;
|
||
with 'App::Netdisco::Worker::Runner';
|
||
}
|
||
|
||
# clear user device_auth and set our own
|
||
config->{'device_auth'} = [{driver => 'snmp'}, {driver => 'cli'}];
|
||
|
||
# TESTS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
||
my $j1 = do_job('TestOne');
|
||
is($j1->status, 'done', 'status is done');
|
||
is($j1->log, 'OK: SNMP driver is successful.',
|
||
'workers are run in decreasing priority until done');
|
||
|
||
my $j2 = do_job('TestTwo');
|
||
is($j2->status, 'done', 'status is done');
|
||
is($j2->log, 'OK: CLI driver is successful.',
|
||
'lower priority driver not run if higher is successful');
|
||
|
||
config->{'device_auth'} = [];
|
||
|
||
my $j3 = do_job('TestOne');
|
||
is($j3->status, 'defer', 'status is defer');
|
||
is($j3->log, 'deferred job with no device creds',
|
||
'no matching config for workers');
|
||
|
||
config->{'device_auth'} = [{driver => 'snmp'}];
|
||
|
||
my $j4 = do_job('TestThree');
|
||
is($j4->status, 'done', 'status is done');
|
||
is($j4->log, 'OK: SNMP driver is successful.',
|
||
'respect user config filtering the driver');
|
||
|
||
config->{'device_auth'} = [
|
||
{driver => 'snmp', action => 'testthree'},
|
||
{driver => 'cli', action => 'foo'},
|
||
];
|
||
|
||
my $j5 = do_job('TestThree');
|
||
is($j5->status, 'done', 'status is done');
|
||
is($j5->log, 'OK: SNMP driver is successful.',
|
||
'respect user config filtering the action');
|
||
|
||
config->{'device_auth'} = [
|
||
{driver => 'snmp', action => 'testthree::_base_'},
|
||
{driver => 'cli', action => 'testthree::foo'},
|
||
];
|
||
|
||
my $j6 = do_job('TestThree');
|
||
is($j6->status, 'done', 'status is done');
|
||
is($j6->log, 'OK: SNMP driver is successful.',
|
||
'respect user config filtering the namespace');
|
||
|
||
config->{'device_auth'} = [{driver => 'snmp'}];
|
||
|
||
my $j7 = do_job('TestFour');
|
||
is($j7->status, 'done', 'status is done');
|
||
is($j7->log, 'OK: custom driver is successful.',
|
||
'override an action');
|
||
|
||
config->{'device_auth'} = [{driver => 'snmp'}];
|
||
|
||
my $j8 = do_job('TestFive');
|
||
is($j8->status, 'done', 'status is done');
|
||
is((scalar @{$j8->_statuslist}), 2, 'two workers ran');
|
||
is($j8->_last_priority, 100, 'priority is for snmp');
|
||
is($j8->log, 'OK: SNMP driver is successful.',
|
||
'add to an action');
|
||
|
||
done_testing;
|
||
|
||
# TESTS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
||
sub do_job {
|
||
my $pkg = shift;
|
||
|
||
# include local plugins
|
||
config->{'extra_worker_plugins'} = ["X::${pkg}"];
|
||
|
||
my $job = App::Netdisco::Backend::Job->new({
|
||
job => 0,
|
||
device => App::Netdisco::DB->resultset('Device')->new_result({ip => '192.0.2.1'}),
|
||
action => lc($pkg),
|
||
});
|
||
|
||
try {
|
||
#info sprintf 'test: started at %s', scalar localtime;
|
||
MyWorker->new()->run($job);
|
||
#info sprintf 'test: %s: %s', $job->status, $job->log;
|
||
}
|
||
catch {
|
||
$job->status('error');
|
||
$job->log("error running job: $_");
|
||
};
|
||
|
||
return $job;
|
||
}
|