From fbbe1b868eb4692a9ae5e47bcda5138ca1615b00 Mon Sep 17 00:00:00 2001
From: Jesse Luehrs <doy@tozt.net>
Date: Mon, 5 Mar 2012 23:38:15 -0600
Subject: test for the role form

---
 t/role.t | 132 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 132 insertions(+)
 create mode 100644 t/role.t

diff --git a/t/role.t b/t/role.t
new file mode 100644
index 0000000..824a14b
--- /dev/null
+++ b/t/role.t
@@ -0,0 +1,132 @@
+#!/usr/bin/env perl
+use strict;
+use warnings;
+use Test::More;
+
+use IO::Pty::Easy;
+use IO::Select;
+
+my $script = <<'SCRIPT';
+use strict;
+use warnings;
+{
+    package My::Term::Filter;
+    use Moose;
+    with 'Term::Filter';
+
+    sub setup {
+        my $self = shift;
+        my @cmd = @_;
+        my $ref = ref($self);
+        print "SETUP: $self ($ref): @cmd\n";
+    }
+
+    sub cleanup {
+        my $self = shift;
+        my $ref = ref($self);
+        print "CLEANUP: $self ($ref)\n";
+    }
+
+    sub munge_input {
+        my $self = shift;
+        my ($buf) = @_;
+        my $ref = ref($self);
+        print "MUNGE_INPUT: $self ($ref): $buf\n";
+        $buf = "\n" if $buf =~ /exit/i;
+        return uc($buf);
+    }
+
+    sub munge_output {
+        my $self = shift;
+        my ($buf) = @_;
+        my $ref = ref($self);
+        print "MUNGE_OUTPUT: $self ($ref): $buf\n";
+        return lc($buf);
+    }
+}
+
+my $term = My::Term::Filter->new;
+print "$term\n";
+$term->run($^X, '-ple', q[last if /^$/]);
+print "done\n";
+SCRIPT
+
+my $crlf = qr/\x0d\x0a/;
+
+# just in case
+alarm 60;
+
+{
+    my $pty = IO::Pty::Easy->new(handle_pty_size => 0);
+    $pty->spawn($^X, '-Ilib', '-e', $script);
+
+    my $setup_str = full_read($pty);
+
+    my ($term_str, $ref) = $setup_str =~ m{
+        ^
+        ((.*)=.*)
+        \n
+        SETUP: \s \1 \s \(\2\):\s
+        \Q$^X\E .* \Q-ple\E .* last\ if\ /\^\$/ .*
+        \n
+        $
+    }sx;
+
+    is($ref, 'My::Term::Filter', "setup callback got a My::Term::Filter object");
+
+    $pty->write("fOo\n");
+
+    like(
+        full_read($pty),
+        qr{
+            ^
+            MUNGE_INPUT: \s \Q$term_str\E \s \($ref\): \s fOo\n
+            \n
+            MUNGE_OUTPUT: \s \Q$term_str\E \s \($ref\): \s FOO$crlf
+            \n
+            foo$crlf
+            MUNGE_OUTPUT: \s \Q$term_str\E \s \($ref\): \s FOO$crlf
+            \n
+            foo$crlf
+            $
+        }sx,
+        "munge_input and munge_output got the right arguments"
+    );
+
+    $pty->write("EXIT\n");
+
+    like(
+        full_read($pty),
+        qr{
+            ^
+            MUNGE_INPUT: \s \Q$term_str\E \s \($ref\): \s EXIT\n
+            \n
+            MUNGE_OUTPUT: \s \Q$term_str\E \s \($ref\): \s $crlf
+            \n
+            $crlf
+            CLEANUP: \s \Q$term_str\E \s \($ref\)\n
+            done\n
+            $
+        }sx,
+        "cleanup got the right arguments"
+    );
+}
+
+sub full_read {
+    my ($pty) = @_;
+
+    my $select = IO::Select->new($pty);
+    return if $select->has_exception(0.1);
+
+    my $ret;
+    while ($select->can_read(1)) {
+        my $new = $pty->read;
+        last unless defined($new) && length($new);
+        $ret .= $new;
+        return $ret if $select->has_exception(0.1);
+    }
+
+    return $ret;
+}
+
+done_testing;
-- 
cgit v1.2.3-54-g00ecf