X-Git-Url: http://git.shiar.net/perl/file-rewrite.git/blobdiff_plain/6ad6da685e4d484f4937dfd8784195cbf6d3470f..6d348a616fe89141e6ea66155bc0a1cb94003b40:/t/10-fileset.t

diff --git a/t/10-fileset.t b/t/10-fileset.t
index c5a5709..ff43498 100644
--- a/t/10-fileset.t
+++ b/t/10-fileset.t
@@ -3,42 +3,72 @@
 use strict;
 use warnings;
 
-use Test::More tests => 13;
+use Test::More tests => 21;
 use Test::NoWarnings;
 
 use autodie;
 use File::Temp qw(tempdir);
 use File::Path qw(remove_tree);
 
-BEGIN { use_ok('File::Rewrite', 'fileset'); }
+BEGIN { use_ok('File::Rewrite', 'fileset') or BAIL_OUT('No fileset()'); }
 
 my $target = tempdir();
 
 sub testfileset {
-	my ($name, $input, $params, $result) = @_;
+	my ($name, $input, $params, $expect, $compare) = @_;
 
 	my $filename = "$target/$name";
 	open my $file, '>', $filename;
 	print {$file} $input;
 	close $file;
 
-	is(fileset($filename, @$params), $result, $name)
-		or diag(explain $input);
+	my $result = eval { fileset($filename, @$params) };
+	if (defined $result and defined $compare) {
+		my $output = eval {
+			local $/;
+			open my $confirm, '<', $filename;
+			readline $confirm;
+		};
+		if ($compare ne $output) {
+			fail($name);
+			diag(sprintf(
+				'  Output mismatch (got %d of %d lines with return value %s)',
+				$output =~ tr/\n//,
+				$compare =~ tr/\n//,
+				explain($result),
+			));
+			diag('         got: "', join('\n', split $/, $output ), '"');
+			diag('    expected: "', join('\n', split $/, $compare), '"');
+			return;
+		}
+	}
+	is($result, $expect, $name);
 }
 
 my $BASICBODY = "replace$/noremove$/remove$/.$/";
 
-testfileset('no match',      $BASICBODY,    [undef, 'nomatch'], 0);
-testfileset('remove string', $BASICBODY,    [undef, 'remove'],  1);
-testfileset('remove empty',  "$/.$/ $/$/",  [undef, ''],        2);
-testfileset('remove regex',  $BASICBODY,    [undef, qr/^re/],   2);
-testfileset('remove all',    $BASICBODY,    [undef, qr/./],     4, '');
+testfileset('no match',      $BASICBODY,    ['no'      => undef], 0, $BASICBODY);
+testfileset('remove string', $BASICBODY,    ['remove'  => undef], 1, "replace$/noremove$/.$/");
+testfileset('remove empty',  "$/.$/ $/$/",  [''        => undef], 2, ".$/ $/");
+testfileset('remove regex',  $BASICBODY,    [qr/^re/   => undef], 2, "noremove$/.$/");
+testfileset('remove all',    $BASICBODY,    [qr/./     => undef], 4, '');
 
+testfileset('add unconditionally', "$/0$/", [undef()   => '0'], 1, "$/0$/0$/");
 testfileset('add string',     "foo$/",      ['bar'],            1, "foo$/bar$/");
 testfileset('keep string',    "foo$/bar$/", ['foo'],            0, "foo$/bar$/");
-testfileset('replace string', "1$/2$/3$/",  [4, 1],             2, "2$/3$/4$/");
-testfileset('replace all',    $BASICBODY,   ['', qr/^re/],      3, "noremove$/.$/$/");
-testfileset('keep first',     "foo$/bar$/foo$/", ['foo'],       1, "foo$/bar$/");
+testfileset('keep first',     "foo$/FOO$/foo$/", ['foo'],       1, "foo$/FOO$/");
+testfileset('replace string', "$/0$/1$/2$/", [0 => 3],          2, "$/1$/2$/3$/");
+testfileset('keep regex',     "foo$/hi$/$/", [qr/./ => 'hi'],   1, "hi$/$/");
 
-is(remove_tree($target), 11, 'no unexpected files');
+testfileset('add new array',  ".$/",        ['' => ['foo','.']],  2, ".$/foo$/.$/");
+testfileset('replace by array', "$/foo$/.$/", ['' => ['foo','']], 3, "foo$/.$/foo$/$/");
+testfileset('partial add',    "foo$/$/",    ['foo' => ['foo','.']], 1, "foo$/$/.$/");
+
+my $SAMPLE2 = "$/foo$/and$/bar$/.$/";
+testfileset('keep pair',      $SAMPLE2, [['foo','bar']], 0, $SAMPLE2);
+testfileset('keep order',     $SAMPLE2, [['bar','foo']], 2, "$/and$/bar$/.$/foo$/");
+testfileset('replace pair',   $SAMPLE2, [['bar','foo'] => ['fooo','barr']], 4, "$/and$/.$/fooo$/barr$/");
+testfileset('mixed arrays',   $SAMPLE2, [['ignore',qr/a/] => ['bar','foo','.']], 3, "$/foo$/bar$/.$/foo$/.$/");
+
+is(remove_tree($target), 19, 'no unexpected files');