Add note in line printing function

[colorize.git] / test.pl

diff --git a/test.pl b/test.pl
index 5329e1ff6b17e9a7a004ef3c7cc7c9f339f8e5cb..373a4bcde96d74f4ab284470406e3c63a24d4ab4 100755 (executable)
--- a/test.pl
+++ b/test.pl
@@ -5,11 +5,16 @@ use warnings;
 use constant true => 1;
 
 use File::Temp qw(tempfile tmpnam);
-use Test::More tests => 9;
+use Test::More;
 
-my $BUF_SIZE = 1024;
+my $tests = 21;
+
+my %BUF_SIZE = (
+   normal => 1024,
+   short  => 10,
+);
 my $source = 'colorize.c';
-my $warning_flags = '-Wall -Wextra -Wformat -Wswitch-default -Wuninitialized -Wunused -Wno-unused-parameter';
+my $warning_flags = '-Wall -Wextra -Wformat -Wswitch-default -Wuninitialized -Wunused -Wno-unused-function -Wno-unused-parameter';
 
 my $write_to_tmpfile = sub
 {
@@ -22,14 +27,13 @@ my $write_to_tmpfile = sub
     return $tmpfile;
 };
 
-my $program;
+plan tests => $tests;
 
 SKIP: {
-    skip "$source does not exist", 9 unless -e $source;
-
-    $program = tmpnam();
+    skip "$source does not exist", $tests unless -e $source;
 
-    skip 'compiling failed', 9 unless system("gcc -DTEST -DBUF_SIZE=$BUF_SIZE $warning_flags -o $program $source") == 0;
+    my $program = tmpnam();
+    skip 'compiling failed (normal)', $tests unless system("gcc -DTEST -DBUF_SIZE=$BUF_SIZE{normal} $warning_flags -o $program $source") == 0;
 
     is(system("$program --help >/dev/null 2>&1"), 0, 'exit value for help screen');
 
@@ -43,17 +47,59 @@ SKIP: {
     is_deeply([split /\n/, qx(cat $infile1 | $program none/none)], [split /\n/, $text], 'text read from stdin');
     is_deeply([split /\n/, qx($program none/none $infile1)],       [split /\n/, $text], 'text read from file');
 
+    my $check_clean = sub
+    {
+        my ($type) = @_;
+
+        my $switch = "--$type";
+
+        is(qx(echo -n "\e[35mhello\e[0m \e[36mworld\e[0m" | $program $switch),       'hello world', "$type colored words");
+        is(qx(echo -n "hello world" | $program Magenta | $program $switch),          'hello world', "$type colored line");
+        is_deeply([split /\n/, qx($program cyan $infile1 | $program $switch)], [split /\n/, $text], "$type colored text");
+
+        ok(qx(echo -n "\e[\e[33m" | $program $switch) eq "\e[", "$type with invalid sequence");
+    };
+
+    $check_clean->($_) foreach qw(clean clean-all);
+
+    is(qx(echo -n "\e[4munderline\e[24m" | $program --clean-all), 'underline', 'clean-all color sequences');
+
+    my $check_clean_buf = sub
+    {
+        my ($program_buf, $type) = @_;
+
+        my $switch = "--$type";
+
+        # Check that line chunks are merged when cleaning text
+        my $short_text = 'Linux dev 2.6.32-5-openvz-686 #1 SMP Sun Sep 23 11:40:07 UTC 2012 i686 GNU/Linux';
+        is(qx(echo -n "$short_text" | $program_buf $switch), $short_text, "merge ${\length $short_text} bytes (BUF_SIZE=$BUF_SIZE{short}, $type)");
+    };
+
+    SKIP: {
+        my $program_buf = tmpnam();
+        skip 'compiling failed (short buffer)', 2 unless system("gcc -DTEST -DBUF_SIZE=$BUF_SIZE{short} $warning_flags -o $program_buf $source") == 0;
+        $check_clean_buf->($program_buf, $_) foreach qw(clean clean-all);
+        unlink $program_buf;
+    }
+
     my $repeated = join "\n", ($text) x 7;
     my $infile2  = $write_to_tmpfile->($repeated);
 
-    is_deeply([split /\n/, qx(cat $infile2 | $program none/none)], [split /\n/, $repeated], "read ${\length $repeated} bytes (BUF_SIZE=$BUF_SIZE)");
+    is_deeply([split /\n/, qx(cat $infile2 | $program none/none)], [split /\n/, $repeated], "read ${\length $repeated} bytes (BUF_SIZE=$BUF_SIZE{normal})");
+
+    my $colored_text = qx(echo "foo bar baz" | $program red);
+    my $sequences = 0;
+    $sequences++ while $colored_text =~ /\e\[\d+m/g;
+    is($sequences, 2, 'count of sequences printed');
 
     is(qx(echo -n "hello\nworld\r\n" | $program none/none), "hello\nworld\r\n", 'stream mode');
 
     is(system("echo \"hello world\" | $program random --exclude-random=black >/dev/null 2>&1"), 0, 'switch exclude-random');
 
-    skip 'valgrind not found', 1 unless system('which valgrind >/dev/null 2>&1') == 0;
-    like(qx(valgrind $program none/none $infile1 2>&1 >/dev/null), qr/no leaks are possible/, 'valgrind memleaks');
+    SKIP: {
+        skip 'valgrind not found', 1 unless system('which valgrind >/dev/null 2>&1') == 0;
+        like(qx(valgrind $program none/none $infile1 2>&1 >/dev/null), qr/no leaks are possible/, 'valgrind memleaks');
+    }
 
     print <<'EOT';
 Colors
@@ -65,9 +111,9 @@ EOT
         my $bold_color = ucfirst $color;
         system("echo $bold_color | $program $bold_color");
     }
-};
 
-unlink $program if defined $program;
+    unlink $program;
+};
 
 __DATA__
 Lorem ipsum dolor sit amet, consectetur adipiscing elit. Vivamus urna mauris, ultricies faucibus placerat sit amet, rutrum eu