t/import.t

   1 # Copyright (C) 2016-2018 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 PublicInbox::MIME;
   7 use PublicInbox::Git;
   8 use PublicInbox::Import;
   9 use File::Temp qw/tempdir/;
  10 my $dir = tempdir('pi-import-XXXXXX', TMPDIR => 1, CLEANUP => 1);
  11
  12 is(system(qw(git init -q --bare), $dir), 0, 'git init successful');
  13 my $git = PublicInbox::Git->new($dir);
  14
  15 my $im = PublicInbox::Import->new($git, 'testbox', 'test@example');
  16 my $mime = PublicInbox::MIME->create(
  17         header => [
  18                 From => 'a@example.com',
  19                 To => 'b@example.com',
  20                 'Content-Type' => 'text/plain',
  21                 Subject => 'this is a subject',
  22                 'Message-ID' => '<a@example.com>',
  23         ],
  24         body => "hello world\n",
  25 );
  26 like($im->add($mime), qr/\A:\d+\z/, 'added one message');
  27 $im->done;
  28 my @revs = $git->qx(qw(rev-list HEAD));
  29 is(scalar @revs, 1, 'one revision created');
  30
  31 $mime->header_set('Message-ID', '<b@example.com>');
  32 $mime->header_set('Subject', 'msg2');
  33 like($im->add($mime, sub { $mime }), qr/\A:\d+\z/, 'added 2nd message');
  34 $im->done;
  35 @revs = $git->qx(qw(rev-list HEAD));
  36 is(scalar @revs, 2, '2 revisions exist');
  37
  38 is($im->add($mime), undef, 'message only inserted once');
  39 $im->done;
  40 @revs = $git->qx(qw(rev-list HEAD));
  41 is(scalar @revs, 2, '2 revisions exist');
  42
  43 foreach my $c ('c'..'z') {
  44         $mime->header_set('Message-ID', "<$c\@example.com>");
  45         $mime->header_set('Subject', "msg - $c");
  46         like($im->add($mime), qr/\A:\d+\z/, "added $c message");
  47 }
  48 $im->done;
  49 @revs = $git->qx(qw(rev-list HEAD));
  50 is(scalar @revs, 26, '26 revisions exist after mass import');
  51 my ($mark, $msg) = $im->remove($mime);
  52 like($mark, qr/\A:\d+\z/, 'got mark');
  53 is(ref($msg), 'PublicInbox::MIME', 'got old message deleted');
  54
  55 is(undef, $im->remove($mime), 'remove is idempotent');
  56
  57 # mismatch on identical Message-ID
  58 $mime->header_set('Message-ID', '<a@example.com>');
  59 ($mark, $msg) = $im->remove($mime);
  60 is($mark, 'MISMATCH', 'mark == MISMATCH on mismatch');
  61 is($msg->header('Message-ID'), '<a@example.com>', 'Message-ID matches');
  62 isnt($msg->header('Subject'), $mime->header('Subject'), 'subject mismatch');
  63
  64 $mime->header_set('Message-Id', '<failcheck@example.com>');
  65 is($im->add($mime, sub { undef }), undef, 'check callback fails');
  66 is($im->remove($mime), undef, 'message not added, so not removed');
  67
  68 $im->done;
  69 done_testing();