91 lines
2.6 KiB
Plaintext
91 lines
2.6 KiB
Plaintext
|
#!perl -w
|
||
|
|
|
||
|
|
use strict;
|
||
|
|
use warnings;
|
||
|
|
|
||
|
|
use Qpsmtpd::Constants;
|
||
|
|
|
||
|
|
sub register_tests {
|
||
|
|
my $self = shift;
|
||
|
|
|
||
|
|
$self->register_test('test_hook_connect', 2);
|
||
|
|
$self->register_test('test_hook_rcpt', 2);
|
||
|
|
$self->register_test('test_ip_whitelisted', 3);
|
||
|
|
$self->register_test('test_is_set_rblsmtpd', 4);
|
||
|
|
$self->register_test('test_hook_disconnect', 1);
|
||
|
|
$self->register_test('test_reject_type', 3);
|
||
|
|
}
|
||
|
|
|
||
|
|
sub test_ip_whitelisted {
|
||
|
|
my $self = shift;
|
||
|
|
|
||
|
|
$self->qp->connection->remote_ip('192.168.99.5');
|
||
|
|
ok( $self->ip_whitelisted(), "+");
|
||
|
|
|
||
|
|
$self->qp->connection->remote_ip('192.168.99.6');
|
||
|
|
ok( ! $self->ip_whitelisted(), "-");
|
||
|
|
|
||
|
|
$self->qp->connection->remote_ip('192.168.99.5');
|
||
|
|
$self->qp->connection->notes('whitelisthost', 'hello honey!');
|
||
|
|
ok( $self->ip_whitelisted(), "+");
|
||
|
|
$self->qp->connection->notes('whitelisthost', undef);
|
||
|
|
};
|
||
|
|
|
||
|
|
sub test_is_set_rblsmtpd {
|
||
|
|
my $self = shift;
|
||
|
|
|
||
|
|
$self->qp->connection->remote_ip('10.1.1.1');
|
||
|
|
ok( ! defined $self->is_set_rblsmtpd('10.1.1.1'), "undef");
|
||
|
|
|
||
|
|
$ENV{RBLSMTPD} = "Yes we can!";
|
||
|
|
cmp_ok( 'Yes we can!','eq',$self->is_set_rblsmtpd('10.1.1.1'), "value");
|
||
|
|
|
||
|
|
$ENV{RBLSMTPD} = "Oh yeah?";
|
||
|
|
cmp_ok( 'Oh yeah?','eq',$self->is_set_rblsmtpd('10.1.1.1'), "value");
|
||
|
|
|
||
|
|
$ENV{RBLSMTPD} = '';
|
||
|
|
cmp_ok( 1,'==',$self->is_set_rblsmtpd('10.1.1.1'), "empty");
|
||
|
|
};
|
||
|
|
|
||
|
|
sub test_hook_connect {
|
||
|
|
my $self = shift;
|
||
|
|
|
||
|
|
my $conn = $self->qp->connection;
|
||
|
|
$conn->relay_client(0); # other tests may leave it enabled
|
||
|
|
$conn->remote_ip('127.0.0.2'); # standard dnsbl test value
|
||
|
|
|
||
|
|
cmp_ok( DECLINED, '==', $self->hook_connect($self->qp->transaction),
|
||
|
|
"connect +");
|
||
|
|
|
||
|
|
ok($self->connection->notes('dnsbl_sockets'), "sockets +");
|
||
|
|
ok($self->connection->notes('dnsbl_domains'), "domains +");
|
||
|
|
}
|
||
|
|
|
||
|
|
sub test_hook_rcpt {
|
||
|
|
my $self = shift;
|
||
|
|
|
||
|
|
my $address = Qpsmtpd::Address->parse('<rcpt@example.com>');
|
||
|
|
my ($ret, $note) = $self->hook_rcpt($self->qp->transaction, $address);
|
||
|
|
is($ret, DENY, "Check we got a DENY ($note)");
|
||
|
|
#print("# dnsbl result: $note\n");
|
||
|
|
}
|
||
|
|
sub test_hook_disconnect {
|
||
|
|
my $self = shift;
|
||
|
|
|
||
|
|
cmp_ok( DECLINED, '==', $self->hook_connect($self->qp->transaction),
|
||
|
|
"hook_disconnect +");
|
||
|
|
}
|
||
|
|
|
||
|
|
sub test_reject_type {
|
||
|
|
my $self = shift;
|
||
|
|
|
||
|
|
$self->{_args}{reject_type} = undef;
|
||
|
|
cmp_ok( $self->get_reject_type(), '==', DENY, "default");
|
||
|
|
|
||
|
|
$self->{_args}{reject_type} = 'temp';
|
||
|
|
cmp_ok( $self->get_reject_type(), '==', DENYSOFT, "defer");
|
||
|
|
|
||
|
|
$self->{_args}{reject_type} = 'disconnect';
|
||
|
|
cmp_ok( $self->get_reject_type(), '==', DENY_DISCONNECT, "disconnect");
|
||
|
|
};
|