test: add DNS cache coverage
This commit is contained in:
@@ -2,6 +2,7 @@ use Test::More;
|
|||||||
use Test::MockObject;
|
use Test::MockObject;
|
||||||
use Mojo::Promise;
|
use Mojo::Promise;
|
||||||
use Urupam::Validation;
|
use Urupam::Validation;
|
||||||
|
use Socket qw(AF_INET);
|
||||||
|
|
||||||
use_ok('Urupam::Validation');
|
use_ok('Urupam::Validation');
|
||||||
|
|
||||||
@@ -80,6 +81,18 @@ sub with_ssrf_ua {
|
|||||||
return $code->();
|
return $code->();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
sub with_subprocess_stub {
|
||||||
|
my ( $result, $code, $calls_ref ) = @_;
|
||||||
|
no warnings 'redefine';
|
||||||
|
local *Mojo::IOLoop::subprocess = sub {
|
||||||
|
my ( $class, $work, $finish, $host ) = @_;
|
||||||
|
$$calls_ref++ if defined $calls_ref;
|
||||||
|
$finish->( undef, undef, $result );
|
||||||
|
return;
|
||||||
|
};
|
||||||
|
return $code->();
|
||||||
|
}
|
||||||
|
|
||||||
subtest 'is_valid_url_length' => sub {
|
subtest 'is_valid_url_length' => sub {
|
||||||
ok( $validator->is_valid_url_length('http://example.com'),
|
ok( $validator->is_valid_url_length('http://example.com'),
|
||||||
'valid URL length passes' );
|
'valid URL length passes' );
|
||||||
@@ -271,6 +284,38 @@ subtest 'is_blocked_url' => sub {
|
|||||||
);
|
);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
subtest '_resolve_host - caches results' => sub {
|
||||||
|
my $calls = 0;
|
||||||
|
my $result = {
|
||||||
|
error => 0,
|
||||||
|
results => [ { addr => '127.0.0.1', family => AF_INET } ],
|
||||||
|
};
|
||||||
|
|
||||||
|
with_subprocess_stub(
|
||||||
|
$result,
|
||||||
|
sub {
|
||||||
|
my ( $value, $error ) =
|
||||||
|
wait_promise( $validator->_resolve_host('example.com') );
|
||||||
|
is( $error, undef, 'first resolve has no error' );
|
||||||
|
is( scalar @$value, 1, 'first resolve returns one address' );
|
||||||
|
},
|
||||||
|
\$calls
|
||||||
|
);
|
||||||
|
|
||||||
|
with_subprocess_stub(
|
||||||
|
$result,
|
||||||
|
sub {
|
||||||
|
my ( $value, $error ) =
|
||||||
|
wait_promise( $validator->_resolve_host('example.com') );
|
||||||
|
is( $error, undef, 'cached resolve has no error' );
|
||||||
|
is( scalar @$value, 1, 'cached resolve returns one address' );
|
||||||
|
},
|
||||||
|
\$calls
|
||||||
|
);
|
||||||
|
|
||||||
|
is( $calls, 1, 'subprocess called once due to cache' );
|
||||||
|
};
|
||||||
|
|
||||||
subtest 'validate_short_code' => sub {
|
subtest 'validate_short_code' => sub {
|
||||||
my @valid = (
|
my @valid = (
|
||||||
[ 'abc123456789', 'alphanumeric code passes' ],
|
[ 'abc123456789', 'alphanumeric code passes' ],
|
||||||
|
|||||||
Reference in New Issue
Block a user