X-Git-Url: http://www.git.stargrave.org/?a=blobdiff_plain;f=t%2Fxcpdb-reshard.t;h=5fce5ead7612b1b81aa5a80bfd93db9cfdc03ee7;hb=2710d3105e70ff467eff9e977325628b4e9dd1c5;hp=88e6c3dc3494db9c8d86421d235b154ac1ddf49f;hpb=fd730c65a3242f609f4ec205ad605408850b2bec;p=public-inbox.git diff --git a/t/xcpdb-reshard.t b/t/xcpdb-reshard.t index 88e6c3dc..5fce5ead 100644 --- a/t/xcpdb-reshard.t +++ b/t/xcpdb-reshard.t @@ -3,16 +3,12 @@ use strict; use warnings; use Test::More; -my @mods = qw(DBI DBD::SQLite Search::Xapian); -foreach my $mod (@mods) { - eval "require $mod"; - plan skip_all => "missing $mod for $0" if $@; -}; -require './t/common.perl'; +use PublicInbox::TestCommon; +require_mods(qw(DBD::SQLite Search::Xapian)); require_git('2.6'); -use File::Temp qw/tempdir/; use PublicInbox::MIME; use PublicInbox::InboxWritable; +require PublicInbox::Search; my $mime = PublicInbox::MIME->create( header => [ @@ -25,7 +21,8 @@ my $mime = PublicInbox::MIME->create( ); my ($this) = (split('/', $0))[-1]; -my $tmpdir = tempdir($this.'-XXXXXX', TMPDIR => 1, CLEANUP => 1); +my ($tmpdir, $for_destroy) = tmpdir(); +local $ENV{PI_CONFIG} = "$tmpdir/config"; my $ibx = PublicInbox::Inbox->new({ inboxdir => "$tmpdir/testbox", name => $this, @@ -43,14 +40,16 @@ for my $i (1..$ndoc) { } $im->done; my @shards = grep(m!/\d+\z!, glob("$ibx->{inboxdir}/xap*/*")); -is(scalar(@shards), $nproc, 'got expected shards'); +is(scalar(@shards), $nproc - 1, 'got expected shards'); my $orig = $ibx->over->query_xover(1, $ndoc); my %nums = map {; "$_->{num}" => 1 } @$orig; # ensure we can go up or down in shards, or stay the same: for my $R (qw(2 4 1 3 3)) { delete $ibx->{search}; # release old handles - ok(run_script([@xcpdb, "-R$R", $ibx->{inboxdir}]), "xcpdb -R$R"); + my $cmd = [@xcpdb, "-R$R", $ibx->{inboxdir}]; + push @$cmd, '--compact' if $R == 1; + ok(run_script($cmd), "xcpdb -R$R"); my @new_shards = grep(m!/\d+\z!, glob("$ibx->{inboxdir}/xap*/*")); is(scalar(@new_shards), $R, 'resharded to two shards'); my $msgs = $ibx->search->query('s:this'); @@ -63,8 +62,12 @@ for my $R (qw(2 4 1 3 3)) { # ensure docids in Xapian match NNTP article numbers my $tot = 0; my %tmp = %nums; + my $XapianDatabase = do { + no warnings 'once'; + $PublicInbox::Search::X{Database}; + }; foreach my $d (@new_shards) { - my $xdb = Search::Xapian::Database->new($d); + my $xdb = $XapianDatabase->new($d); $tot += $xdb->get_doccount; my $it = $xdb->postlist_begin(''); my $end = $xdb->postlist_end('');