sub register_tests {
    my $self = shift;
    $self->register_test("test_local", 1);
    $self->register_test("test_returnval", 1);
}

sub test_local {
    my $self = shift;
    
    my $connection = $self->qp->connection;
    $connection->remote_ip('127.0.0.2'); # standard dnsbl test value
    
    $self->hook_connect($self->qp->transaction);
    
    ok($self->qp->connection->notes('dnsbl_sockets'));
}

sub test_returnval {
    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");
    print("# dnsbl result: $note\n");
}