]> Sergey Matveev's repositories - public-inbox.git/blob - t/spamcheck_spamc.t
view: single id="t" for multi-Subject messages
[public-inbox.git] / t / spamcheck_spamc.t
1 # Copyright (C) 2016-2020 all contributors <meta@public-inbox.org>
2 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
3 use strict;
4 use warnings;
5 use Test::More;
6 use Email::Simple;
7 use IO::File;
8 use Fcntl qw(:DEFAULT SEEK_SET);
9 use PublicInbox::TestCommon;
10 my ($tmpdir, $for_destroy) = tmpdir();
11
12 use_ok 'PublicInbox::Spamcheck::Spamc';
13 my $spamc = PublicInbox::Spamcheck::Spamc->new;
14 $spamc->{checkcmd} = [qw(cat)];
15
16 {
17         open my $fh, '+>', "$tmpdir/file" or die "open failed: $!";
18         ok(!$spamc->spamcheck($fh), 'empty '.ref($fh));
19 }
20 ok(!$spamc->spamcheck(IO::File->new_tmpfile), 'IO::File->new_tmpfile');
21
22 my $dst = '';
23 my $src = <<'EOF';
24 Date: Thu, 01 Jan 1970 00:00:00 +0000
25 To: <e@example.com>
26 From: <e@example.com>
27 Subject: test
28 Message-ID: <testmessage@example.com>
29
30 EOF
31 ok($spamc->spamcheck(Email::Simple->new($src), \$dst), 'Email::Simple works');
32 is($dst, $src, 'input == output');
33
34 $dst = '';
35 $spamc->{checkcmd} = ['sh', '-c', 'cat; false'];
36 ok(!$spamc->spamcheck(Email::Simple->new($src), \$dst), 'Failed check works');
37 is($dst, $src, 'input == output for spammy example');
38
39 for my $l (qw(ham spam)) {
40         my $file = "$tmpdir/$l.out";
41         $spamc->{$l.'cmd'} = ['tee', $file ];
42         my $method = $l.'learn';
43         ok($spamc->$method(Email::Simple->new($src)), "$method OK");
44         open my $fh, '<', $file or die "failed to open $file: $!";
45         is(eval { local $/, <$fh> }, $src, "$l command ran alright");
46 }
47
48 done_testing();