t/search-thr-index.t

   1 # Copyright (C) 2017-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 File::Temp qw/tempdir/;
   7 use Email::MIME;
   8 eval { require PublicInbox::SearchIdx; };
   9 plan skip_all => "Xapian missing for search" if $@;
  10 my $tmpdir = tempdir('pi-search-thr-index.XXXXXX', TMPDIR => 1, CLEANUP => 1);
  11 my $git_dir = "$tmpdir/a.git";
  12
  13 is(0, system(qw(git init -q --bare), $git_dir), "git init (main)");
  14 my $rw = PublicInbox::SearchIdx->new($git_dir, 1);
  15 ok($rw, "search indexer created");
  16 my $data = <<'EOF';
  17 Subject: [RFC 00/14]
  18 Message-Id: <1-bw@g>
  19
  20 Subject: [RFC 09/14]
  21 Message-Id: <10-bw@g>
  22 In-Reply-To: <1-bw@g>
  23 References: <1-bw@g>
  24
  25 Subject: [RFC 03/14]
  26 Message-Id: <4-bw@g>
  27 In-Reply-To: <1-bw@g>
  28 References: <1-bw@g>
  29
  30 EOF
  31
  32 my $num = 0;
  33 # nb. using internal API, fragile!
  34 my $xdb = $rw->_xdb_acquire;
  35 $xdb->begin_transaction;
  36 my @mids;
  37
  38 foreach (reverse split(/\n\n/, $data)) {
  39         $_ .= "\n";
  40         my $mime = Email::MIME->new(\$_);
  41         $mime->header_set('From' => 'bw@g');
  42         $mime->header_set('To' => 'git@vger.kernel.org');
  43         my $bytes = bytes::length($mime->as_string);
  44         my $mid = $mime->header('Message-Id');
  45         my $doc_id = $rw->add_message($mime, $bytes, ++$num, 'ignored', $mid);
  46         push @mids, $mid;
  47         ok($doc_id, 'message added: '. $mid);
  48 }
  49
  50 my $prev;
  51 foreach my $mid (@mids) {
  52         my $res = $rw->get_thread($mid);
  53         is(3, $res->{total}, "got all messages from $mid");
  54 }
  55
  56 done_testing();
  57
  58 1;