X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=t%2F27filter_generated.t;h=fc768c21e63a9d677c57af04a82735e16062742e;hb=master;hp=22eb77223e5c286c31a6119b99355796ecb04d12;hpb=dde95f2c6188ae6182da0f32320e7f1c9da0a90a;p=dbsrgits%2FDBIx-Class-Schema-Loader.git

diff --git a/t/27filter_generated.t b/t/27filter_generated.t
index 22eb772..fc768c2 100644
--- a/t/27filter_generated.t
+++ b/t/27filter_generated.t
@@ -1,46 +1,93 @@
 use strict;
-use File::Slurp qw(slurp);
+use warnings;
+use DBIx::Class::Schema::Loader;
+use DBIx::Class::Schema::Loader::Utils 'slurp_file';
 use File::Path;
-use Test::More tests => 4;
+use Test::More tests => 19;
 use Test::Exception;
 use lib qw(t/lib);
 use make_dbictest_db;
 use dbixcsl_test_dir qw/$tdir/;
 
-use DBIx::Class::Schema::Loader;
-
 my $dump_path = "$tdir/dump";
 
 my %original_class_data;
 
+my ($schema_file_count, $result_file_count);
+
 {
     package DBICTest::Schema::1;
-    use base qw/ DBIx::Class::Schema::Loader /;
+    use Test::More;
+    use base 'DBIx::Class::Schema::Loader';
     __PACKAGE__->loader_options(
         dump_directory => $dump_path,
-        filter_generated_text => sub {
-            my ($class, $text) = @_;
+        quiet => 1,
+        filter_generated_code => sub {
+            my ($type, $class, $text) = @_;
+
+            like $type, qr/^(?:schema|result)\z/,
+                'got correct file type';
+
+            if ($type eq 'schema') {
+                $schema_file_count++;
+                is $class, 'DBICTest::Schema::1',
+                    'correct class for schema type file passed to filter';
+            }
+            elsif ($type eq 'result') {
+                $result_file_count++;
+                like $class, qr/^DBICTest::Schema::1::Result::(?:Foo|Bar)\z/,
+                    'correct class for result type file passed to filter';
+            }
+            else {
+                die 'invalid file type passed to filter';
+            }
+
             $original_class_data{$class} = $text;
-	    if ($class =~ /::1$/) {
+            if ($class =~ /::1$/) {
                 $text = "No Gotcha!";
-	    }
+            }
             else {
-	        $text .= q{"Kilroy was here";};
-	    }
+                $text .= q{my $foo = "Kilroy was here";};
+            }
             return $text;
         },
     );
 }
 
+{
+    package DBICTest::Schema::2;
+    use base 'DBIx::Class::Schema::Loader';
+    __PACKAGE__->loader_options(
+        dump_directory => $dump_path,
+        quiet => 1,
+        filter_generated_code => qq{"$^X" t/bin/simple_filter},
+    );
+}
+
 DBICTest::Schema::1->connect($make_dbictest_db::dsn);
 
-my $foo = slurp("$dump_path/DBICTest/Schema/1/Result/Foo.pm");
-ok(! -e "$dump_path/DBICTest/Schema/1.pm",
-     "No package means no file written");
-ok($original_class_data{"DBICTest::Schema::1"},
-     "Even though we processed the missing class");
-like($foo, qr/# Created by .* THE FIRST PART/s,
-     "We get the whole autogenerated text");
-like($foo, qr/Kilroy was here/, "Can insert text");
+# schema is generated in 2 passes
+
+is $schema_file_count, 2,
+    'correct number of schema files passed to filter';
+
+is $result_file_count, 4,
+    'correct number of result files passed to filter';
+
+my $foo = slurp_file "$dump_path/DBICTest/Schema/1/Result/Foo.pm";
+ok ! -e "$dump_path/DBICTest/Schema/1.pm",
+    "No package means no file written";
+ok $original_class_data{"DBICTest::Schema::1"},
+    "Even though we processed the missing class";
+like $foo, qr/# Created by .* THE FIRST PART/s,
+    "We get the whole autogenerated text";
+like $foo, qr/Kilroy was here/, "Can insert text";
+
+DBICTest::Schema::2->connect($make_dbictest_db::dsn);
+
+$foo = slurp_file "$dump_path/DBICTest/Schema/2/Result/Foo.pm";
+
+like $foo, qr/Kilroy was here/,
+    "Can insert text via command filter";
 
 END { rmtree($dump_path, 1, 1); }