# Copyright (C) 2018-2020 all contributors
# License: AGPL-3.0+
use strict;
use warnings;
use Test::More;
use PublicInbox::TestCommon;
use File::Path qw(remove_tree);
use Cwd qw(abs_path);
require_git(2.6);
local $ENV{HOME} = abs_path('t');
# Integration tests for HTTP cloning + mirroring
require_mods(qw(Plack::Util Plack::Builder
HTTP::Date HTTP::Status Search::Xapian DBD::SQLite));
use IO::Socket;
use POSIX qw(dup2);
use_ok 'PublicInbox::V2Writable';
use PublicInbox::InboxWritable;
use PublicInbox::Eml;
use PublicInbox::Config;
# FIXME: too much setup
my ($tmpdir, $for_destroy) = tmpdir();
my $pi_config = "$tmpdir/config";
{
open my $fh, '>', $pi_config or die "open($pi_config): $!";
print $fh <<"" or die "print $pi_config: $!";
[publicinbox "v2"]
inboxdir = $tmpdir/in
address = test\@example.com
close $fh or die "close($pi_config): $!";
}
local $ENV{PI_CONFIG} = $pi_config;
my $cfg = PublicInbox::Config->new($pi_config);
my $ibx = $cfg->lookup('test@example.com');
ok($ibx, 'inbox found');
$ibx->{version} = 2;
my $v2w = PublicInbox::V2Writable->new($ibx, 1);
ok $v2w, 'v2w loaded';
$v2w->{parallel} = 0;
my $mime = PublicInbox::Eml->new(<<'');
From: Me
To: You
Subject: a
Date: Thu, 01 Jan 1970 00:00:00 +0000
my $old_rotate_bytes = $v2w->{rotate_bytes};
$v2w->{rotate_bytes} = 500; # force rotating
for my $i (1..9) {
$mime->header_set('Message-ID', "<$i\@example.com>");
$mime->header_set('Subject', "subject = $i");
ok($v2w->add($mime), "add msg $i OK");
}
my $epoch_max = $v2w->{epoch_max};
ok($epoch_max > 0, "multiple epochs");
$v2w->done;
{
my $smsg = $ibx->over->get_art(1);
like($smsg->{lines}, qr/\A[0-9]+\z/, 'lines is a digit');
like($smsg->{bytes}, qr/\A[0-9]+\z/, 'bytes is a digit');
}
$ibx->cleanup;
my $sock = tcp_server();
ok($sock, 'sock created');
my $cmd = [ '-httpd', '-W0', "--stdout=$tmpdir/out", "--stderr=$tmpdir/err" ];
my $td = start_script($cmd, undef, { 3 => $sock });
my ($host, $port) = ($sock->sockhost, $sock->sockport);
$sock = undef;
my @cmd;
foreach my $i (0..$epoch_max) {
my $sfx = $i == 0 ? '.git' : '';
@cmd = (qw(git clone --mirror -q),
"http://$host:$port/v2/$i$sfx",
"$tmpdir/m/git/$i.git");
is(xsys(@cmd), 0, "cloned $i.git");
ok(-d "$tmpdir/m/git/$i.git", "mirror $i OK");
}
@cmd = ("-init", '-j1', '-V2', 'm', "$tmpdir/m", 'http://example.com/m',
'alt@example.com');
ok(run_script(\@cmd), 'initialized public-inbox -V2');
my @shards = glob("$tmpdir/m/xap*/?");
is(scalar(@shards), 1, 'got a single shard on init');
ok(run_script([qw(-index -j0), "$tmpdir/m"]), 'indexed');
my $mibx = { inboxdir => "$tmpdir/m", address => 'alt@example.com' };
$mibx = PublicInbox::Inbox->new($mibx);
is_deeply([$mibx->mm->minmax], [$ibx->mm->minmax], 'index synched minmax');
$v2w->{rotate_bytes} = $old_rotate_bytes;
for my $i (10..15) {
$mime->header_set('Message-ID', "<$i\@example.com>");
$mime->header_set('Subject', "subject = $i");
ok($v2w->add($mime), "add msg $i OK");
}
$v2w->done;
$ibx->cleanup;
my $fetch_each_epoch = sub {
foreach my $i (0..$epoch_max) {
my $dir = "$tmpdir/m/git/$i.git";
is(xsys('git', "--git-dir=$dir", 'fetch', '-q'), 0,
'fetch successful');
}
};
$fetch_each_epoch->();
my $mset = $mibx->search->reopen->mset('m:15@example.com');
is(scalar($mset->items), 0, 'new message not found in mirror, yet');
ok(run_script([qw(-index -j0), "$tmpdir/m"]), 'index updated');
is_deeply([$mibx->mm->minmax], [$ibx->mm->minmax], 'index synched minmax');
$mset = $mibx->search->reopen->mset('m:15@example.com');
is(scalar($mset->items), 1, 'found message in mirror');
# purge:
$mime->header_set('Message-ID', '<10@example.com>');
$mime->header_set('Subject', 'subject = 10');
{
my @warn;
local $SIG{__WARN__} = sub { push @warn, @_ };
ok($v2w->purge($mime), 'purge a message');
my $warn = join('', @warn);
like($warn, qr/purge rewriting/);
my @subj = ($warn =~ m/^# subject .*$/mg);
is_deeply(\@subj, ["# subject = 10"], "only rewrote one");
}
$v2w->done;
my $msgs = $mibx->over->get_thread('10@example.com');
my $to_purge = $msgs->[0]->{blob};
like($to_purge, qr/\A[a-f0-9]{40,}\z/, 'read blob to be purged');
$mset = $ibx->search->reopen->mset('m:10@example.com');
is(scalar($mset->items), 0, 'purged message gone from origin');
$fetch_each_epoch->();
{
$ibx->cleanup;
PublicInbox::InboxWritable::cleanup($mibx);
$v2w->done;
my $cmd = [ qw(-index --prune -j0), "$tmpdir/m" ];
my ($out, $err) = ('', '');
my $opt = { 1 => \$out, 2 => \$err };
ok(run_script($cmd, undef, $opt), '-index --prune');
like($err, qr/discontiguous range/, 'warned about discontiguous range');
unlike($err, qr/fatal/, 'no scary fatal error shown');
}
$mset = $mibx->search->reopen->mset('m:10@example.com');
is(scalar($mset->items), 0, 'purged message not found in mirror');
is_deeply([$mibx->mm->minmax], [$ibx->mm->minmax], 'minmax still synced');
for my $i ((1..9),(11..15)) {
$mset = $mibx->search->mset("m:$i\@example.com");
is(scalar($mset->items), 1, "$i\@example.com remains visible");
}
is($mibx->git->check($to_purge), undef, 'unindex+prune successful in mirror');
{
my @warn;
local $SIG{__WARN__} = sub { push @warn, @_ };
$v2w->index_sync;
is_deeply(\@warn, [], 'no warnings from index_sync after purge');
}
# deletes happen in a different fetch window
{
$mset = $mibx->search->reopen->mset('m:1@example.com');
is(scalar($mset->items), 1, '1@example.com visible in mirror');
$mime->header_set('Message-ID', '<1@example.com>');
$mime->header_set('Subject', 'subject = 1');
ok($v2w->remove($mime), 'removed <1@example.com> from source');
$v2w->done;
$ibx->cleanup;
$fetch_each_epoch->();
PublicInbox::InboxWritable::cleanup($mibx);
my $cmd = [ qw(-index -j0), "$tmpdir/m" ];
my ($out, $err) = ('', '');
my $opt = { 1 => \$out, 2 => \$err };
ok(run_script($cmd, undef, $opt), 'index ran');
is($err, '', 'no errors reported by index');
$mset = $mibx->search->reopen->mset('m:1@example.com');
is(scalar($mset->items), 0, '1@example.com no longer visible in mirror');
}
if ('sequential-shard') {
$mset = $mibx->search->mset('m:15@example.com');
is(scalar($mset->items), 1, 'large message not indexed');
remove_tree(glob("$tmpdir/m/xap*"), glob("$tmpdir/m/msgmap.*"));
my $cmd = [ qw(-index -j9 --sequential-shard), "$tmpdir/m" ];
ok(run_script($cmd), '--sequential-shard works');
my @shards = glob("$tmpdir/m/xap*/?");
is(scalar(@shards), 8, 'got expected shard count');
PublicInbox::InboxWritable::cleanup($mibx);
$mset = $mibx->search->mset('m:15@example.com');
is(scalar($mset->items), 1, 'search works after --sequential-shard');
}
if ('max size') {
$mime->header_set('Message-ID', '<2big@a>');
my $max = '2k';
$mime->body_str_set("z\n" x 1024);
ok($v2w->add($mime), "add big message");
$v2w->done;
$ibx->cleanup;
$fetch_each_epoch->();
PublicInbox::InboxWritable::cleanup($mibx);
my $cmd = [qw(-index -j0), "$tmpdir/m", "--max-size=$max" ];
my $opt = { 2 => \(my $err) };
ok(run_script($cmd, undef, $opt), 'indexed with --max-size');
like($err, qr/skipping [a-f0-9]{40,}/, 'warned about skipping message');
$mset = $mibx->search->reopen->mset('m:2big@a');
is(scalar($mset->items), 0, 'large message not indexed');
{
open my $fh, '>>', $pi_config or die;
print $fh <search->reopen->mset('m:2big@a');
is(scalar($mset->items), 0, 'large message not re-indexed');
}
ok($td->kill, 'killed httpd');
$td->join;
done_testing();
1;