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