# Copyright (c) 2009 by David Golden. All rights reserved.
# Licensed under Apache License, Version 2.0 (the "License").
# You may not use this file except in compliance with the License.
# A copy of the License was distributed with this file or you may obtain a
# copy of the License from http://www.apache.org/licenses/LICENSE-2.0
use strict;
use warnings;
use Test::More;
use lib 't/lib';
use IO::Handle;
use IO::File;
use File::Temp qw/tmpnam/;
use Utils qw/next_fd sig_num/;
use Capture::Tiny ':all';
use Config;
plan tests => 19;
local $ENV{PERL_CAPTURE_TINY_TIMEOUT} = 0; # no timeouts
my $builder = Test::More->builder;
binmode($builder->failure_output, ':utf8') if $] >= 5.008;
my $fd = next_fd;
my ($out, $err, $res, @res);
#--------------------------------------------------------------------------#
# capture to custom IO::File
#--------------------------------------------------------------------------#
my $temp_out = tmpnam();
my $temp_err = tmpnam();
ok( !-e $temp_out, "Temp out '$temp_out' doesn't exist" );
ok( !-e $temp_err, "Temp out '$temp_err' doesn't exist" );
my $out_fh = IO::File->new($temp_out, "w+");
my $err_fh = IO::File->new($temp_err, "w+");
capture {
print STDOUT "foo\n";
print STDERR "bar\n";
} stdout => $out_fh, stderr => $err_fh;
$out_fh->close;
$err_fh->close;
is( scalar do {local (@ARGV,$/) = $temp_out; <>} , "foo\n",
"captured STDOUT to custom handle (IO::File)"
);
is( scalar do {local (@ARGV,$/) = $temp_err; <>} , "bar\n",
"captured STDERR to custom handle (IO::File)"
);
unlink $_ for $temp_out, $temp_err;
#--------------------------------------------------------------------------#
# capture to GLOB handle
#--------------------------------------------------------------------------#
$temp_out = tmpnam();
$temp_err = tmpnam();
ok( !-e $temp_out, "Temp out '$temp_out' doesn't exist" );
ok( !-e $temp_err, "Temp out '$temp_err' doesn't exist" );
open $out_fh, "+>", $temp_out;
open $err_fh, "+>", $temp_err;
capture {
print STDOUT "foo\n";
print STDERR "bar\n";
} stdout => $out_fh, stderr => $err_fh;
$out_fh->close;
$err_fh->close;
is( scalar do {local (@ARGV,$/) = $temp_out; <>} , "foo\n",
"captured STDOUT to custom handle (GLOB)"
);
is( scalar do {local (@ARGV,$/) = $temp_err; <>} , "bar\n",
"captured STDERR to custom handle (GLOB)"
);
unlink $_ for $temp_out, $temp_err;
#--------------------------------------------------------------------------#
# append to custom IO::File
#--------------------------------------------------------------------------#
$temp_out = tmpnam();
$temp_err = tmpnam();
ok( !-e $temp_out, "Temp out '$temp_out' doesn't exist" );
ok( !-e $temp_err, "Temp out '$temp_err' doesn't exist" );
$out_fh = IO::File->new($temp_out, "w+");
$err_fh = IO::File->new($temp_err, "w+");
$out_fh->autoflush(1);
$err_fh->autoflush(1);
print $out_fh "Shouldn't see this in capture\n";
print $err_fh "Shouldn't see this in capture\n";
my ($got_out, $got_err) = capture {
print STDOUT "foo\n";
print STDERR "bar\n";
} stdout => $out_fh, stderr => $err_fh;
$out_fh->close;
$err_fh->close;
is( $got_out, "foo\n",
"captured appended STDOUT to custom handle"
);
is( $got_err, "bar\n",
"captured appended STDERR to custom handle"
);
unlink $_ for $temp_out, $temp_err;
#--------------------------------------------------------------------------#
# repeated append to custom IO::File with no output
#--------------------------------------------------------------------------#
$temp_out = tmpnam();
$temp_err = tmpnam();
ok( !-e $temp_out, "Temp out '$temp_out' doesn't exist" );
ok( !-e $temp_err, "Temp out '$temp_err' doesn't exist" );
$out_fh = IO::File->new($temp_out, "a+");
$err_fh = IO::File->new($temp_err, "a+");
($got_out, $got_err) = capture {
my $i = 0; $i++ for 1 .. 10; # no output, just busywork
} stdout => $out_fh, stderr => $err_fh;
is( $got_out, "",
"Try 1: captured empty appended STDOUT to custom handle"
);
is( $got_err, "",
"Try 1: captured empty appended STDERR to custom handle"
);
($got_out, $got_err) = capture {
my $i = 0; $i++ for 1 .. 10; # no output, just busywork
} stdout => $out_fh, stderr => $err_fh;
is( $got_out, "",
"Try 2: captured empty appended STDOUT to custom handle"
);
is( $got_err, "",
"Try 2: captured empty appended STDERR to custom handle"
);
unlink $_ for $temp_out, $temp_err;
#--------------------------------------------------------------------------#
# finish
#--------------------------------------------------------------------------#
close ARGV; # opened by reading from <>
is( next_fd, $fd, "no file descriptors leaked" );
exit 0;