t/shared_kv.t

   1 #!perl -w
   2 # Copyright (C) all contributors <meta@public-inbox.org>
   3 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
   4 use strict;
   5 use v5.10.1;
   6 use Test::More;
   7 use PublicInbox::TestCommon;
   8 require_mods(qw(DBD::SQLite));
   9 use_ok 'PublicInbox::SharedKV';
  10 my ($tmpdir, $for_destroy) = tmpdir();
  11 local $ENV{TMPDIR} = $tmpdir;
  12 my $skv = PublicInbox::SharedKV->new;
  13 my $skv_tmpdir = $skv->{"tmp$$.$skv"};
  14 ok(-d $skv_tmpdir, 'created a temporary dir');
  15 $skv->dbh;
  16 my $dead = "\xde\xad";
  17 my $beef = "\xbe\xef";
  18 my $cafe = "\xca\xfe";
  19 ok($skv->set($dead, $beef), 'set');
  20 is($skv->get($dead), $beef, 'get');
  21 ok($skv->set($dead, $beef), 'set idempotent');
  22 ok(!$skv->set_maybe($dead, $cafe), 'set_maybe ignores');
  23 ok($skv->set_maybe($cafe, $dead), 'set_maybe sets');
  24 is($skv->xchg($dead, $cafe), $beef, 'xchg');
  25 is($skv->get($dead), $cafe, 'get after xchg');
  26 is($skv->xchg($dead, undef), $cafe, 'xchg to undef');
  27 is($skv->get($dead), undef, 'get after xchg to undef');
  28 is($skv->get($cafe), $dead, 'get after set_maybe');
  29 is($skv->xchg($dead, $cafe), undef, 'xchg from undef');
  30 is($skv->count, 2, 'count works');
  31
  32 my %seen;
  33 my $sth = $skv->each_kv_iter;
  34 while (my ($k, $v) = $sth->fetchrow_array) {
  35         $seen{$k} = $v;
  36 }
  37 is($seen{$dead}, $cafe, '$dead has expected value');
  38 is($seen{$cafe}, $dead, '$cafe has expected value');
  39 is(scalar keys %seen, 2, 'iterated through all');
  40
  41 undef $skv;
  42 ok(!-d $skv_tmpdir, 'temporary dir gone');
  43 $skv = PublicInbox::SharedKV->new("$tmpdir/dir", 'base');
  44 ok(-e "$tmpdir/dir/base.sqlite3", 'file created');
  45 $skv->dbh;
  46 ok($skv->set_maybe('02', '2'), "`02' set");
  47 ok($skv->set_maybe('2', '2'), "`2' set (no match on `02')");
  48
  49 done_testing;