]> Sergey Matveev's repositories - public-inbox.git/blobdiff - t/lei_store.t
lei q: import flags when clobbering/augmenting Maildirs
[public-inbox.git] / t / lei_store.t
index a189f8970737ea613c0a8a75ec0f772c1095e63b..1c3f78413bd8f84a6a2cb5321c367ad79208f486 100644 (file)
@@ -1,5 +1,5 @@
 #!perl -w
-# Copyright (C) 2020 all contributors <meta@public-inbox.org>
+# Copyright (C) 2020-2021 all contributors <meta@public-inbox.org>
 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
 use strict;
 use v5.10.1;
@@ -11,33 +11,35 @@ require_ok 'PublicInbox::LeiStore';
 require_ok 'PublicInbox::ExtSearch';
 my ($home, $for_destroy) = tmpdir();
 my $opt = { 1 => \(my $out = ''), 2 => \(my $err = '') };
-my $store_dir = "$home/lst";
-my $lst = PublicInbox::LeiStore->new($store_dir, { creat => 1 });
-ok($lst, '->new');
-my $oid = $lst->add_eml(eml_load('t/data/0001.patch'));
-like($oid, qr/\A[0-9a-f]+\z/, 'add returned OID');
+my $store_dir = "$home/sto";
+local $ENV{GIT_COMMITTER_EMAIL} = 'lei@example.com';
+local $ENV{GIT_COMMITTER_NAME} = 'lei user';
+my $sto = PublicInbox::LeiStore->new($store_dir, { creat => 1 });
+ok($sto, '->new');
+my $smsg = $sto->add_eml(eml_load('t/data/0001.patch'));
+like($smsg->{blob}, qr/\A[0-9a-f]+\z/, 'add returned OID');
 my $eml = eml_load('t/data/0001.patch');
-is($lst->add_eml($eml), undef, 'idempotent');
-$lst->done;
-is_deeply([$lst->mbox_keywords($eml)], [], 'no keywords');
+is($sto->add_eml($eml), undef, 'idempotent');
+$sto->done;
+is_deeply([$sto->mbox_keywords($eml)], [], 'no keywords');
 $eml->header_set('Status', 'RO');
-is_deeply([$lst->mbox_keywords($eml)], ['seen'], 'seen extracted');
+is_deeply([$sto->mbox_keywords($eml)], ['seen'], 'seen extracted');
 $eml->header_set('X-Status', 'A');
-is_deeply([$lst->mbox_keywords($eml)], [qw(answered seen)],
+is_deeply([$sto->mbox_keywords($eml)], [qw(answered seen)],
        'seen+answered extracted');
 $eml->header_set($_) for qw(Status X-Status);
 
-is_deeply([$lst->maildir_keywords('/foo:2,')], [], 'Maildir no keywords');
-is_deeply([$lst->maildir_keywords('/foo:2,S')], ['seen'], 'Maildir seen');
-is_deeply([$lst->maildir_keywords('/foo:2,RS')], ['answered', 'seen'],
+is_deeply([$sto->maildir_keywords('/foo:2,')], [], 'Maildir no keywords');
+is_deeply([$sto->maildir_keywords('/foo:2,S')], ['seen'], 'Maildir seen');
+is_deeply([$sto->maildir_keywords('/foo:2,RS')], ['answered', 'seen'],
        'Maildir answered + seen');
-is_deeply([$lst->maildir_keywords('/foo:2,RSZ')], ['answered', 'seen'],
+is_deeply([$sto->maildir_keywords('/foo:2,RSZ')], ['answered', 'seen'],
        'Maildir answered + seen w/o Z');
 {
-       my $es = $lst->search;
+       my $es = $sto->search;
        my $msgs = $es->over->query_xover(0, 1000);
        is(scalar(@$msgs), 1, 'one message');
-       is($msgs->[0]->{blob}, $oid, 'blob matches');
+       is($msgs->[0]->{blob}, $smsg->{blob}, 'blob matches');
        my $mset = $es->mset("mid:$msgs->[0]->{mid}");
        is($mset->size, 1, 'search works');
        is_deeply($es->mset_to_artnums($mset), [ $msgs->[0]->{num} ],
@@ -47,47 +49,91 @@ is_deeply([$lst->maildir_keywords('/foo:2,RSZ')], ['answered', 'seen'],
 }
 
 for my $parallel (0, 1) {
-       $lst->{priv_eidx}->{parallel} = $parallel;
-       my $docids = $lst->set_eml_keywords($eml, qw(seen draft));
+       $sto->{priv_eidx}->{parallel} = $parallel;
+       my $docids = $sto->set_eml_keywords($eml, qw(seen draft));
        is(scalar @$docids, 1, 'set keywords on one doc');
-       $lst->done;
-       my @kw = $lst->search->msg_keywords($docids->[0]);
+       $sto->done;
+       my @kw = $sto->search->msg_keywords($docids->[0]);
        is_deeply(\@kw, [qw(draft seen)], 'kw matches');
 
-       $docids = $lst->add_eml_keywords($eml, qw(seen draft));
-       $lst->done;
+       $docids = $sto->add_eml_keywords($eml, qw(seen draft));
+       $sto->done;
        is(scalar @$docids, 1, 'idempotently added keywords to doc');
-       @kw = $lst->search->msg_keywords($docids->[0]);
+       @kw = $sto->search->msg_keywords($docids->[0]);
        is_deeply(\@kw, [qw(draft seen)], 'kw matches after noop');
 
-       $docids = $lst->remove_eml_keywords($eml, qw(seen draft));
+       $docids = $sto->remove_eml_keywords($eml, qw(seen draft));
        is(scalar @$docids, 1, 'removed from one doc');
-       $lst->done;
-       @kw = $lst->search->msg_keywords($docids->[0]);
+       $sto->done;
+       @kw = $sto->search->msg_keywords($docids->[0]);
        is_deeply(\@kw, [], 'kw matches after remove');
 
-       $docids = $lst->remove_eml_keywords($eml, qw(answered));
+       $docids = $sto->remove_eml_keywords($eml, qw(answered));
        is(scalar @$docids, 1, 'removed from one doc (idempotently)');
-       $lst->done;
-       @kw = $lst->search->msg_keywords($docids->[0]);
+       $sto->done;
+       @kw = $sto->search->msg_keywords($docids->[0]);
        is_deeply(\@kw, [], 'kw matches after remove (idempotent)');
 
-       $docids = $lst->add_eml_keywords($eml, qw(answered));
+       $docids = $sto->add_eml_keywords($eml, qw(answered));
        is(scalar @$docids, 1, 'added to empty doc');
-       $lst->done;
-       @kw = $lst->search->msg_keywords($docids->[0]);
+       $sto->done;
+       @kw = $sto->search->msg_keywords($docids->[0]);
        is_deeply(\@kw, ['answered'], 'kw matches after add');
 
-       $docids = $lst->set_eml_keywords($eml);
+       $docids = $sto->set_eml_keywords($eml);
        is(scalar @$docids, 1, 'set to clobber');
-       $lst->done;
-       @kw = $lst->search->msg_keywords($docids->[0]);
+       $sto->done;
+       @kw = $sto->search->msg_keywords($docids->[0]);
        is_deeply(\@kw, [], 'set clobbers all');
+
+       my $set = eml_load('t/plack-qp.eml');
+       $set->header_set('Message-ID', "<set\@$parallel>");
+       my $ret = $sto->set_eml($set, 'seen');
+       is(ref $ret, 'PublicInbox::Smsg', 'initial returns smsg');
+       my $ids = $sto->set_eml($set, qw(seen));
+       is_deeply($ids, [ $ret->{num} ], 'set_eml idempotent');
+       $ids = $sto->set_eml($set, qw(seen answered));
+       is_deeply($ids, [ $ret->{num} ], 'set_eml to change kw');
+       $sto->done;
+       @kw = $sto->search->msg_keywords($ids->[0]);
+       is_deeply(\@kw, [qw(answered seen)], 'set changed kw');
 }
 
 SKIP: {
        require_mods(qw(Storable), 1);
-       ok($lst->can('ipc_do'), 'ipc_do works if we have Storable');
+       ok($sto->can('ipc_do'), 'ipc_do works if we have Storable');
+       $eml->header_set('Message-ID', '<ipc-test@example>');
+       my $pid = $sto->ipc_worker_spawn('lei-store');
+       ok($pid > 0, 'got a worker');
+       my $smsg = $sto->ipc_do('set_eml', $eml, qw(seen));
+       is(ref($smsg), 'PublicInbox::Smsg', 'set_eml works over ipc');
+       my $ids = $sto->ipc_do('set_eml', $eml, qw(seen));
+       is_deeply($ids, [ $smsg->{num} ], 'docid returned');
+
+       $eml->header_set('Message-ID');
+       my $no_mid = $sto->ipc_do('set_eml', $eml, qw(seen));
+       my $wait = $sto->ipc_do('done');
+       my @kw = $sto->search->msg_keywords($no_mid->{num});
+       is_deeply(\@kw, [qw(seen)], 'ipc set changed kw');
+
+       is(ref($smsg), 'PublicInbox::Smsg', 'no mid works ipc');
+       $ids = $sto->ipc_do('set_eml', $eml, qw(seen));
+       is_deeply($ids, [ $no_mid->{num} ], 'docid returned w/o mid w/ ipc');
+       $sto->ipc_do('done');
+       $sto->ipc_worker_stop;
+       $ids = $sto->ipc_do('set_eml', $eml, qw(seen answered));
+       is_deeply($ids, [ $no_mid->{num} ], 'docid returned w/o mid w/o ipc');
+       $wait = $sto->ipc_do('done');
+
+       my $lse = $sto->search;
+       @kw = $lse->msg_keywords($no_mid->{num});
+       is_deeply(\@kw, [qw(answered seen)], 'set changed kw w/o ipc');
+       is($lse->kw_changed($eml, [qw(answered seen)]), 0,
+               'kw_changed false when unchanged');
+       is($lse->kw_changed($eml, [qw(answered seen flagged)]), 1,
+               'kw_changed true when +flagged');
+       is($lse->kw_changed(eml_load('t/plack-qp.eml'), ['seen']), undef,
+               'kw_changed undef on unknown message');
 }
 
 done_testing;