2 # Copyright (C) 2020 all contributors <meta@public-inbox.org>
3 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
7 use PublicInbox::TestCommon;
8 require_mods(qw(DBD::SQLite Search::Xapian));
10 require_ok 'PublicInbox::LeiStore';
11 require_ok 'PublicInbox::ExtSearch';
12 my ($home, $for_destroy) = tmpdir();
13 my $opt = { 1 => \(my $out = ''), 2 => \(my $err = '') };
14 my $store_dir = "$home/lst";
15 my $lst = PublicInbox::LeiStore->new($store_dir, { creat => 1 });
17 my $oid = $lst->add_eml(eml_load('t/data/0001.patch'));
18 like($oid, qr/\A[0-9a-f]+\z/, 'add returned OID');
19 my $eml = eml_load('t/data/0001.patch');
20 is($lst->add_eml($eml), undef, 'idempotent');
22 is_deeply([$lst->mbox_keywords($eml)], [], 'no keywords');
23 $eml->header_set('Status', 'RO');
24 is_deeply([$lst->mbox_keywords($eml)], ['seen'], 'seen extracted');
25 $eml->header_set('X-Status', 'A');
26 is_deeply([$lst->mbox_keywords($eml)], [qw(answered seen)],
27 'seen+answered extracted');
28 $eml->header_set($_) for qw(Status X-Status);
30 is_deeply([$lst->maildir_keywords('/foo:2,')], [], 'Maildir no keywords');
31 is_deeply([$lst->maildir_keywords('/foo:2,S')], ['seen'], 'Maildir seen');
32 is_deeply([$lst->maildir_keywords('/foo:2,RS')], ['answered', 'seen'],
33 'Maildir answered + seen');
34 is_deeply([$lst->maildir_keywords('/foo:2,RSZ')], ['answered', 'seen'],
35 'Maildir answered + seen w/o Z');
37 my $es = $lst->search;
38 my $msgs = $es->over->query_xover(0, 1000);
39 is(scalar(@$msgs), 1, 'one message');
40 is($msgs->[0]->{blob}, $oid, 'blob matches');
41 my $mset = $es->mset("mid:$msgs->[0]->{mid}");
42 is($mset->size, 1, 'search works');
43 is_deeply($es->mset_to_artnums($mset), [ $msgs->[0]->{num} ],
45 my @kw = $es->msg_keywords(($mset->items)[0]);
46 is_deeply(\@kw, [], 'no flags');
49 for my $parallel (0, 1) {
50 $lst->{priv_eidx}->{parallel} = $parallel;
51 my $docids = $lst->set_eml_keywords($eml, qw(seen draft));
52 is(scalar @$docids, 1, 'set keywords on one doc');
54 my @kw = $lst->search->msg_keywords($docids->[0]);
55 is_deeply(\@kw, [qw(draft seen)], 'kw matches');
57 $docids = $lst->add_eml_keywords($eml, qw(seen draft));
59 is(scalar @$docids, 1, 'idempotently added keywords to doc');
60 @kw = $lst->search->msg_keywords($docids->[0]);
61 is_deeply(\@kw, [qw(draft seen)], 'kw matches after noop');
63 $docids = $lst->remove_eml_keywords($eml, qw(seen draft));
64 is(scalar @$docids, 1, 'removed from one doc');
66 @kw = $lst->search->msg_keywords($docids->[0]);
67 is_deeply(\@kw, [], 'kw matches after remove');
69 $docids = $lst->remove_eml_keywords($eml, qw(answered));
70 is(scalar @$docids, 1, 'removed from one doc (idempotently)');
72 @kw = $lst->search->msg_keywords($docids->[0]);
73 is_deeply(\@kw, [], 'kw matches after remove (idempotent)');
75 $docids = $lst->add_eml_keywords($eml, qw(answered));
76 is(scalar @$docids, 1, 'added to empty doc');
78 @kw = $lst->search->msg_keywords($docids->[0]);
79 is_deeply(\@kw, ['answered'], 'kw matches after add');
81 $docids = $lst->set_eml_keywords($eml);
82 is(scalar @$docids, 1, 'set to clobber');
84 @kw = $lst->search->msg_keywords($docids->[0]);
85 is_deeply(\@kw, [], 'set clobbers all');