#! perl
# Copyright (C) 2010, Parrot Foundation.
# $Id:$

use strict;
use warnings;
use lib qw( . lib ../lib ../../lib );

use Test::More tests => 3;
use Parrot::Test;
use Parrot::Config;

use Parrot::Test::Util 'create_tempfile';

=head1 NAME

t/pmc/stdin-io.t - IO Ops from stdin

=head1 SYNOPSIS

    % prove t/pmc/stdin-io.t

=head1 DESCRIPTION

Tests the Parrot IO operations from stdin.

=cut

sub pir_stdin_output_is {
    my ($input_string, $code, $expected_output, $test_name) = @_;

    my $stuff = sub {
	# Put the string on a file.
	my $string = shift;

	my (undef, $file) = create_tempfile(UNLINK => 1);
	open(my $out, ">$file") or die "bug";
	print $out $string;
	return $file;
    };

    # Write the input and code strings.
    my $input_file = $stuff->($input_string);
    my $code_file = $stuff->($code);

    # Slurp and compare the output.
    my $result = do {
	local $/;
	open(my $in, "./parrot '$code_file' < '$input_file' |")
	    or die "bug";
	<$in>;
    };
    Test::More::is($result, $expected_output, $test_name);
}

pir_stdin_output_is(<<'INPUT', <<'CODE', <<'OUTPUT', 'simple stdin use');
foo
INPUT
.sub test :main
	.local pmc stdin
	stdin = getstdin
	read_str(stdin, 1)
	read_str(stdin, 2)
	read_str(stdin, 5)
	read_str(stdin, 5)
.end
.sub read_str
	.param pmc stream
	.param int n_chars
	.local string input_string
	.local int len
	input_string = stream.'read'(n_chars)
	len = length input_string
	print "got len "
	print len
	print ':  "'
	print input_string
	print "\"\n"
.end
CODE
got len 1:  "f"
got len 2:  "oo"
got len 1:  "
"
got len 0:  ""
OUTPUT

pir_stdin_output_is(<<'INPUT', <<'CODE', <<'OUTPUT', 'readline from stdin');
one
two
three
INPUT
.sub test :main
	.local pmc stdin
	stdin = getstdin
	read_str(stdin)
	read_str(stdin)
	read_str(stdin)
	read_str(stdin)
.end
.sub read_str
	.param pmc stream
	.local string input_string
	.local int len
	input_string = stream.'readline'()
	len = length input_string
	print "got len "
	print len
	print ':  "'
	print input_string
	print "\"\n"
.end
CODE
got len 4:  "one
"
got len 4:  "two
"
got len 6:  "three
"
got len 0:  ""
OUTPUT

pir_stdin_output_is(<<'INPUT', <<'CODE', <<'OUTPUT', 'readall from stdin');
one
two
three
INPUT
.sub test :main
	.local pmc stdin
	stdin = getstdin
	read_str(stdin)
	read_str(stdin)
.end
.sub read_str
	.param pmc stream
	.local string input_string
	.local int len
	input_string = stream.'readall'()
	len = length input_string
	print "got len "
	print len
	print ':  "'
	print input_string
	print "\"\n"
.end
CODE
got len 14:  "one
two
three
"
got len 0:  ""
OUTPUT