2 # Copyright (C) 2020-2021 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/sto";
15 local $ENV{GIT_COMMITTER_EMAIL} = 'lei@example.com';
16 local $ENV{GIT_COMMITTER_NAME} = 'lei user';
17 my $sto = PublicInbox::LeiStore->new($store_dir, { creat => 1 });
19 my $smsg = $sto->add_eml(eml_load('t/data/0001.patch'));
20 like($smsg->{blob}, qr/\A[0-9a-f]+\z/, 'add returned OID');
21 my $eml = eml_load('t/data/0001.patch');
22 is($sto->add_eml($eml), undef, 'idempotent');
25 my $es = $sto->search;
26 ok($es->can('isrch'), ref($es). ' can ->isrch (for SolverGit)');
27 my $msgs = $es->over->query_xover(0, 1000);
28 is(scalar(@$msgs), 1, 'one message');
29 is($msgs->[0]->{blob}, $smsg->{blob}, 'blob matches');
30 my $mset = $es->mset("mid:$msgs->[0]->{mid}");
31 is($mset->size, 1, 'search works');
32 is_deeply($es->mset_to_artnums($mset), [ $msgs->[0]->{num} ],
34 my @kw = $es->msg_keywords(($mset->items)[0]);
35 is_deeply(\@kw, [], 'no flags');
38 for my $parallel (0, 1) {
39 $sto->{priv_eidx}->{parallel} = $parallel;
40 my $docids = $sto->set_eml_vmd($eml, { kw => [ qw(seen draft) ] });
41 is(scalar @$docids, 1, 'set keywords on one doc');
43 my @kw = $sto->search->msg_keywords($docids->[0]);
44 is_deeply(\@kw, [qw(draft seen)], 'kw matches');
46 $docids = $sto->add_eml_vmd($eml, {kw => [qw(seen draft)]});
48 is(scalar @$docids, 1, 'idempotently added keywords to doc');
49 @kw = $sto->search->msg_keywords($docids->[0]);
50 is_deeply(\@kw, [qw(draft seen)], 'kw matches after noop');
52 $docids = $sto->remove_eml_vmd($eml, {kw => [qw(seen draft)]});
53 is(scalar @$docids, 1, 'removed from one doc');
55 @kw = $sto->search->msg_keywords($docids->[0]);
56 is_deeply(\@kw, [], 'kw matches after remove');
58 $docids = $sto->remove_eml_vmd($eml, {kw=> [qw(answered)]});
59 is(scalar @$docids, 1, 'removed from one doc (idempotently)');
61 @kw = $sto->search->msg_keywords($docids->[0]);
62 is_deeply(\@kw, [], 'kw matches after remove (idempotent)');
64 $docids = $sto->add_eml_vmd($eml, {kw => [qw(answered)]});
65 is(scalar @$docids, 1, 'added to empty doc');
67 @kw = $sto->search->msg_keywords($docids->[0]);
68 is_deeply(\@kw, ['answered'], 'kw matches after add');
70 $docids = $sto->set_eml_vmd($eml, { kw => [] });
71 is(scalar @$docids, 1, 'set to clobber');
73 @kw = $sto->search->msg_keywords($docids->[0]);
74 is_deeply(\@kw, [], 'set clobbers all');
76 my $set = eml_load('t/plack-qp.eml');
77 $set->header_set('Message-ID', "<set\@$parallel>");
78 my $ret = $sto->set_eml($set, { kw => [ 'seen' ] });
79 is(ref $ret, 'PublicInbox::Smsg', 'initial returns smsg');
80 my $ids = $sto->set_eml($set, { kw => [ 'seen' ] });
81 is_deeply($ids, [ $ret->{num} ], 'set_eml idempotent');
82 $ids = $sto->set_eml($set, { kw => [ qw(seen answered) ] });
83 is_deeply($ids, [ $ret->{num} ], 'set_eml to change kw');
85 @kw = $sto->search->msg_keywords($ids->[0]);
86 is_deeply(\@kw, [qw(answered seen)], 'set changed kw');
90 require_mods(qw(Storable), 1);
91 ok($sto->can('ipc_do'), 'ipc_do works if we have Storable');
92 $eml->header_set('Message-ID', '<ipc-test@example>');
93 my $pid = $sto->ipc_worker_spawn('lei-store');
94 ok($pid > 0, 'got a worker');
95 my $smsg = $sto->ipc_do('set_eml', $eml, { kw => [ qw(seen) ] });
96 is(ref($smsg), 'PublicInbox::Smsg', 'set_eml works over ipc');
97 my $ids = $sto->ipc_do('set_eml', $eml, { kw => [ qw(seen) ] });
98 is_deeply($ids, [ $smsg->{num} ], 'docid returned');
100 $eml->header_set('Message-ID');
101 my $no_mid = $sto->ipc_do('set_eml', $eml, { kw => [ qw(seen) ] });
102 my $wait = $sto->ipc_do('done');
103 my @kw = $sto->search->msg_keywords($no_mid->{num});
104 is_deeply(\@kw, [qw(seen)], 'ipc set changed kw');
106 is(ref($smsg), 'PublicInbox::Smsg', 'no mid works ipc');
107 $ids = $sto->ipc_do('set_eml', $eml, { kw => [ qw(seen) ] });
108 is_deeply($ids, [ $no_mid->{num} ], 'docid returned w/o mid w/ ipc');
109 $sto->ipc_do('done');
110 $sto->ipc_worker_stop;
111 $ids = $sto->ipc_do('set_eml', $eml, { kw => [ qw(seen answered) ] });
112 is_deeply($ids, [ $no_mid->{num} ], 'docid returned w/o mid w/o ipc');
113 $wait = $sto->ipc_do('done');
115 my $lse = $sto->search;
116 @kw = $lse->msg_keywords($no_mid->{num});
117 is_deeply(\@kw, [qw(answered seen)], 'set changed kw w/o ipc');
118 is($lse->kw_changed($eml, [qw(answered seen)]), 0,
119 'kw_changed false when unchanged');
120 is($lse->kw_changed($eml, [qw(answered seen flagged)]), 1,
121 'kw_changed true when +flagged');
122 is($lse->kw_changed(eml_load('t/plack-qp.eml'), ['seen']), undef,
123 'kw_changed undef on unknown message');