t/msgmap.t

   1 # Copyright (C) 2015-2020 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 PublicInbox::TestCommon;
   7 require_mods('DBD::SQLite');
   8 use_ok 'PublicInbox::Msgmap';
   9 my ($tmpdir, $for_destroy) = tmpdir();
  10 my $d = PublicInbox::Msgmap->new($tmpdir, 1);
  11
  12 my %mid2num;
  13 my %num2mid;
  14 my @mids = qw(a@b c@d e@f g@h aa@bb aa@cc);
  15 is_deeply([$d->minmax], [undef,undef], "empty min max on new DB");
  16
  17 foreach my $mid (@mids) {
  18         my $n = $d->mid_insert($mid);
  19         ok($n, "mid $mid inserted");
  20         $mid2num{$mid} = $n;
  21         $num2mid{$n} = $mid;
  22 }
  23
  24 $@ = undef;
  25 my $ret = $d->mid_insert('a@b');
  26 is($ret, undef, 'duplicate mid_insert in undef result');
  27 is($d->num_for('a@b'), $mid2num{'a@b'}, 'existing number not clobbered');
  28 my $next = (sort(keys %num2mid))[-1];
  29 is($d->mid_insert('ok@unique'), $next + 1,
  30         'got expected num after failing mid_insert');
  31
  32 foreach my $n (keys %num2mid) {
  33         is($d->mid_for($n), $num2mid{$n}, "num:$n maps correctly");
  34 }
  35 foreach my $mid (@mids) {
  36         is($d->num_for($mid), $mid2num{$mid}, "mid:$mid maps correctly");
  37 }
  38
  39 is(undef, $d->last_commit, "last commit not set");
  40 my $lc = 'deadbeef' x 5;
  41 is(undef, $d->last_commit($lc), 'previous last commit (undef) returned');
  42 is($lc, $d->last_commit, 'last commit was set correctly');
  43
  44 my $nc = 'deaddead' x 5;
  45 is($lc, $d->last_commit($nc), 'returned previously set commit');
  46 is($nc, $d->last_commit, 'new commit was set correctly');
  47
  48 is($d->mid_delete('a@b'), 1, 'deleted a@b');
  49 is($d->mid_delete('a@b') + 0, 0, 'delete again returns zero');
  50 is(undef, $d->num_for('a@b'), 'num_for fails on deleted msg');
  51 $d = undef;
  52
  53 ok($d = PublicInbox::Msgmap->new($tmpdir, 1), 'idempotent DB creation');
  54 my ($min, $max) = $d->minmax;
  55 ok($min > 0, "article min OK");
  56 ok($max > 0 && $max < 10, "article max OK");
  57 ok($min < $max, "article counts OK");
  58
  59 my $orig = $d->mid_insert('spam@1');
  60 $d->mid_delete('spam@1');
  61 is($d->mid_insert('spam@2'), 1 + $orig, "last number not recycled");
  62
  63 my $tmp = $d->tmp_clone($tmpdir);
  64 is_deeply([$d->minmax], [$tmp->minmax], 'Cloned temporary DB matches');
  65 ok($tmp->mid_delete('spam@2'), 'temporary DB is writable');
  66
  67 is(eval {
  68         $tmp->atfork_prepare;
  69         $tmp->atfork_parent;
  70         'ok'
  71 }, 'ok', 'atfork_* work on tmp_clone');
  72
  73 done_testing();