byterazor/qpsmtpd
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add tests

Browse Source
This commit is contained in:
Jared Johnson 2014-10-28 14:38:15 -05:00
parent e9400cc085
commit b5d6c2e4c8
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
t/plugin_tests/greylisting
View File

@ -19,6 +19,7 @@ sub register_tests {
$self->register_test('test_hook_data'); $self->register_test('test_hook_data');
$self->register_test('test_get_db_key'); $self->register_test('test_get_db_key');
$self->register_test('test_get_db_location'); $self->register_test('test_get_db_location');
$self->register_test('test_exclude');
$self->register_test("test_greylist_geoip"); $self->register_test("test_greylist_geoip");
$self->register_test("test_greylist_p0f_genre"); $self->register_test("test_greylist_p0f_genre");
$self->register_test("test_greylist_p0f_distance"); $self->register_test("test_greylist_p0f_distance");
@ -87,6 +88,15 @@ sub test_get_db_location {
ok( $db, "db location: $db"); ok( $db, "db location: $db");
}; };
sub test_exclude {
my ( $self ) = @_;
$self->connection->relay_client(1);
ok( $self->exclude(), "Relay client results in exclude() hit" );
$self->connection->relay_client(0);
ok( ! $self->exclude(), "Non-relay client results in exclude() miss" );
};
sub test_greylist_geoip { sub test_greylist_geoip {
my $self = shift; my $self = shift;
@ -98,11 +108,13 @@ sub test_greylist_geoip {
foreach my $cc ( @valid ) { foreach my $cc ( @valid ) {
$self->connection->notes('geoip_country', $cc ); $self->connection->notes('geoip_country', $cc );
ok( $self->geoip_match(), "match + ($cc)"); ok( $self->geoip_match(), "match + ($cc)");
ok( $self->exclude(), "match + ($cc) results in exclude() hit");
}; };
foreach my $cc ( @invalid ) { foreach my $cc ( @invalid ) {
$self->connection->notes('geoip_country', $cc ); $self->connection->notes('geoip_country', $cc );
ok( ! $self->geoip_match(), "bad - ($cc)"); ok( ! $self->geoip_match(), "bad - ($cc)");
ok( ! $self->exclude(), "miss - ($cc) results in exclude() miss");
}; };
}; };
@ -136,10 +148,12 @@ sub test_greylist_p0f_link {
$self->{_args}{'p0f'} = 'link,dsl'; $self->{_args}{'p0f'} = 'link,dsl';
$self->connection->notes('p0f'=> { link=>'DSL' } ); $self->connection->notes('p0f'=> { link=>'DSL' } );
ok( $self->p0f_match(), 'p0f link hit'); ok( $self->p0f_match(), 'p0f link hit');
ok( ! $self->exclude(), 'p0f link hit results in exclude() miss' );
$self->{_args}{'p0f'} = 'link,dsl'; $self->{_args}{'p0f'} = 'link,dsl';
$self->connection->notes('p0f'=> { link=>'Ethernet' } ); $self->connection->notes('p0f'=> { link=>'Ethernet' } );
ok( ! $self->p0f_match(), 'p0f link miss'); ok( ! $self->p0f_match(), 'p0f link miss');
ok( $self->exclude(), 'p0f link miss results in exclude() hit' );
} }
sub test_greylist_p0f_uptime { sub test_greylist_p0f_uptime {