lei_to_mail: support for non-seekable outputs

[public-inbox.git] / t / lei_to_mail.t

diff --git a/t/lei_to_mail.t b/t/lei_to_mail.t
index 231cf543a8f954447e81d6283a0ba9a40095cbef..f3cc71adfeb35e7738e499f86b2a4ae199553480 100644 (file)
--- a/t/lei_to_mail.t
+++ b/t/lei_to_mail.t
@@ -6,6 +6,8 @@ use v5.10.1;
 use Test::More;
 use PublicInbox::TestCommon;
 use PublicInbox::Eml;
+use Fcntl qw(SEEK_SET);
+require_mods(qw(DBD::SQLite));
 use_ok 'PublicInbox::LeiToMail';
 my $from = "Content-Length: 10\nSubject: x\n\nFrom hell\n";
 my $noeol = "Subject: x\n\nFrom hell";
@@ -86,27 +88,74 @@ my $orig = do {
 
        local $lei->{opt} = { jobs => 2 };
        $wcb = PublicInbox::LeiToMail->write_cb("mboxcl2:$fn", $lei);
+       $lei->{dedupe}->prepare_dedupe;
        $wcb->(\($dup = $buf), 'deadbeef', [ qw(seen) ]);
        undef $wcb;
        open $fh, '<', $fn or BAIL_OUT $!;
        is($raw, do { local $/; <$fh> }, 'jobs > 1');
        $raw;
 };
-SKIP: {
-       use PublicInbox::Spawn qw(which);
-       my $gzip = which('gzip') or skip 'gzip not found', 1;
-       my $wcb = PublicInbox::LeiToMail->write_cb("mboxcl2:$fn.gz", $lei);
-       $wcb->(\(my $dup = $buf), 'deadbeef', [ qw(seen) ]);
-       undef $wcb;
-       my $uncompressed = xqx([$gzip, '-dc', "$fn.gz"]);
-       is($uncompressed, $orig, 'gzip works');
+for my $zsfx (qw(gz bz2 xz)) { # XXX should we support zst, zz, lzo, lzma?
+       my $zsfx2cmd = PublicInbox::LeiToMail->can('zsfx2cmd');
+       SKIP: {
+               my $cmd = eval { $zsfx2cmd->($zsfx, 0, $lei) };
+               skip $@, 3 if $@;
+               my $dc_cmd = eval { $zsfx2cmd->($zsfx, 1, $lei) };
+               ok($dc_cmd, "decompressor for .$zsfx");
+               my $f = "$fn.$zsfx";
+               my $dst = "mboxcl2:$f";
+               my $wcb = PublicInbox::LeiToMail->write_cb($dst, $lei);
+               $wcb->(\(my $dup = $buf), 'deadbeef', [ qw(seen) ]);
+               undef $wcb;
+               my $uncompressed = xqx([@$dc_cmd, $f]);
+               is($uncompressed, $orig, "$zsfx works unlocked");
 
-       local $lei->{opt} = { jobs => 2 };
-       unlink "$fn.gz" or die "unlink $!";
-       $wcb = PublicInbox::LeiToMail->write_cb("mboxcl2:$fn.gz", $lei);
-       $wcb->(\(my $dupe = $buf), 'deadbeef', [ qw(seen) ]);
-       undef $wcb;
-       is(xqx([$gzip, '-dc', "$fn.gz"]), $orig);
+               local $lei->{opt} = { jobs => 2 }; # for atomic writes
+               unlink $f or BAIL_OUT "unlink $!";
+               $wcb = PublicInbox::LeiToMail->write_cb($dst, $lei);
+               $lei->{dedupe}->prepare_dedupe;
+               $wcb->(\($dup = $buf), 'deadbeef', [ qw(seen) ]);
+               undef $wcb;
+               is(xqx([@$dc_cmd, $f]), $orig, "$zsfx matches with lock");
+       }
+}
+
+unlink $fn or BAIL_OUT $!;
+require PublicInbox::MboxReader;
+if ('default deduplication uses content_hash') {
+       my $wcb = PublicInbox::LeiToMail->write_cb("mboxo:$fn", $lei);
+       $wcb->(\(my $x = $buf), 'deadbeef', []) for (1..2);
+       undef $wcb; # undef to commit changes
+       my $cmp = '';
+       open my $fh, '<', $fn or BAIL_OUT $!;
+       PublicInbox::MboxReader->mboxo($fh, sub { $cmp .= shift->as_string });
+       is($cmp, $buf, 'only one message written');
+}
+
+{ # stdout support
+       open my $tmp, '+>', undef or BAIL_OUT $!;
+       local $lei->{1} = $tmp;
+       my $wcb = PublicInbox::LeiToMail->write_cb("mboxrd:/dev/stdout", $lei);
+       $wcb->(\(my $x = $buf), 'deadbeef', []);
+       undef $wcb; # commit
+       seek($tmp, 0, SEEK_SET) or BAIL_OUT $!;
+       my $cmp = '';
+       PublicInbox::MboxReader->mboxrd($tmp, sub { $cmp .= shift->as_string });
+       is($cmp, $buf, 'message written to stdout');
+}
+
+SKIP: { # FIFO support
+       use PublicInbox::Spawn qw(popen_rd which);
+       use POSIX qw(mkfifo);
+       my $fn = "$tmpdir/fifo";
+       mkfifo($fn, 0600) or skip("mkfifo not supported: $!", 1);
+       my $cat = popen_rd([which('cat'), $fn]);
+       my $wcb = PublicInbox::LeiToMail->write_cb("mboxo:$fn", $lei);
+       $wcb->(\(my $x = $buf), 'deadbeef', []);
+       undef $wcb; # commit
+       my $cmp = '';
+       PublicInbox::MboxReader->mboxo($cat, sub { $cmp .= shift->as_string });
+       is($cmp, $buf, 'message written to FIFO');
 }
 
 done_testing;