[PATCH 2/4] tests for the spamassassin plugin

---
t/plugin_tests/spamassassin |  147 +++++++++++++++++++++++++++++++++++++++++++
1 file changed, 147 insertions(+)
create mode 100644 t/plugin_tests/spamassassin

diff --git a/t/plugin_tests/spamassassin b/t/plugin_tests/spamassassin
new file mode 100644
index 0000000..5056a16
--- /dev/null
+++ b/t/plugin_tests/spamassassin
@@ -0,0 +1,147 @@
+#!perl -w
+
+use strict;
+use warnings;
+
+use Mail::Header;
+use Qpsmtpd::Constants;
+
+my @sample_headers = (
+    'No, score=-5.4 required=4.0 autolearn=ham',
+    'No, score=-8.2 required=4.0 autolearn=ham',
+    'No, score=-102.3 required=4.0 autolearn=disabled',
+    'No, score=-0.1 required=5.0 tests=AWL,BAYES_00,FREEMAIL_FROM,HTML_MESSAGE,RCVD_IN_DNSWL_NONE,RDNS_NONE autolearn=no version=3.3.2',
+    'No, score=4.4 required=5.0 autolearn=no',
+    'Yes, score=14.3 required=5.0 autolearn=no',
+    'Yes, score=18.3 required=5.0 autolearn=spam',
+    'Yes, score=26.6 required=4.0 autolearn=unavailable',
+    'No, score=-1.7 required=4.0 autolearn=unavailable version=3.3.2',
+    'No, hits=-1.0 required=4.0 autolearn=unavailable version=3.3.2',
+);
+
+sub register_tests {
+    my $self = shift;
+
+    my %args = ( );
+    $self->register( $self->qp, %args );
+
+    $self->register_test('test_parse_spam_header', 10);
+    $self->register_test('test_get_spam_results', 19);
+    $self->register_test('test_check_spam_munge_subject', 4);
+    $self->register_test('test_check_spam_reject', 2);
+}
+
+sub test_check_spam_reject {
+    my $self = shift;
+
+    my $transaction = $self->qp->transaction;
+    $self->setup_headers();
+
+    # message scored a 10, should pass
+    $self->{_args}->{reject} = 12;
+    $transaction->notes('spamassassin', { score => 10 } );
+    my $r = $self->check_spam_reject($transaction);
+    cmp_ok( DECLINED, '==', $r, "check_spam_reject, $r");
+    
+    # message scored a 15, should fail
+    $self->{_args}->{reject} = 12;
+    $transaction->notes('spamassassin', { score => 15 } );
+    ($r) = $self->check_spam_reject($transaction);
+    cmp_ok( DENY, '==', $r, "check_spam_reject, $r");
+};
+
+sub test_check_spam_munge_subject {
+    my $self = shift;
+
+    my $transaction = $self->qp->transaction;
+    $self->setup_headers();
+    my $subject = 'DSPAM smells better than SpamAssassin';
+
+    $self->{_args}{munge_subject_threshold} = 5;
+    $transaction->notes('spamassassin', { score => 6 } );
+    $transaction->header->add('Subject', $subject);
+    $self->check_spam_munge_subject($transaction);
+    my $r = $transaction->header->get('Subject'); chomp $r;
+    cmp_ok($r, 'eq', "*** SPAM *** $subject", "check_spam_munge_subject +");
+
+    $transaction->header->delete('Subject');  # cleanup
+    $self->{_args}{munge_subject_threshold} = 5;
+    $transaction->notes('spamassassin', { score => 3 } );
+    $transaction->header->add('Subject', $subject);
+    $self->check_spam_munge_subject($transaction);
+    $r = $transaction->header->get('Subject'); chomp $r;
+    cmp_ok($r, 'eq', $subject, "check_spam_munge_subject -");
+
+    $transaction->header->delete('Subject');  # cleanup
+    $transaction->notes('spamassassin', { score => 3, required => 4 } );
+    $transaction->header->add('Subject', $subject);
+    $self->check_spam_munge_subject($transaction);
+    $r = $transaction->header->get('Subject'); chomp $r;
+    cmp_ok($r, 'eq', $subject, "check_spam_munge_subject -");
+
+    $transaction->header->delete('Subject');  # cleanup
+    $transaction->notes('spamassassin', { score => 5, required => 4 } );
+    $transaction->header->add('Subject', $subject);
+    $self->check_spam_munge_subject($transaction);
+    $r = $transaction->header->get('Subject'); chomp $r;
+    cmp_ok($r, 'eq', "*** SPAM *** $subject", "check_spam_munge_subject +");
+};
+
+sub test_get_spam_results {
+    my $self = shift;
+
+    my $transaction = $self->qp->transaction;
+    $self->setup_headers();
+
+    foreach my $h ( @sample_headers ) {
+        $transaction->notes('spamassassin', undef);  # empty cache
+        $transaction->header->delete('X-Spam-Status'); # delete previous header
+        $transaction->header->add('X-Spam-Status', $h);
+        my $r_ref = $self->get_spam_results($transaction);
+        if ( $h =~ /hits=/ ) {
+            $r_ref->{hits} = delete $r_ref->{score}; # SA v2 compat
+        };
+        my $r2 = _reassemble_header($r_ref);
+        cmp_ok( $h, 'eq', $r2, "get_spam_results ($h)" );
+
+        # this time it should be cached
+        $r_ref = $self->get_spam_results($transaction);
+        next if $h =~ /hits=/;   # caching is broken for SA v2 headers 
+        $r2 = _reassemble_header($r_ref);
+        cmp_ok( $h, 'eq', $r2, "get_spam_results ($h)" );
+    };
+
+};
+
+sub test_parse_spam_header {
+    my $self = shift;
+
+    foreach my $h ( @sample_headers ) {
+        my $r_ref = $self->parse_spam_header($h);
+        if ( $h =~ /hits=/ ) { 
+            $r_ref->{hits} = delete $r_ref->{score}; # SA v2 compat
+        };
+        my $r2 = _reassemble_header($r_ref);
+        cmp_ok( $h, 'eq', $r2, "parse_spam_header ($h)" );
+    };
+};
+
+sub setup_headers {
+    my $self = shift;
+
+    my $transaction = $self->qp->transaction;
+    my $header = Mail::Header->new(Modify => 0, MailFrom => "COERCE");
+    $transaction->header( $header );
+};
+
+sub _reassemble_header {
+    my $info_ref = shift;
+    my $string = $info_ref->{'is_spam'};
+    $string .= ",";
+    foreach ( qw/ hits score required tests autolearn version / ) {
+        next if ! defined $info_ref->{$_};
+        $string .= " $_=$info_ref->{$_}";
+    };
+    return $string;
+};
+
-- 
1.7.9.6