use strict;
-use File::Slurp qw(slurp);
+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$/) {
$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 => "$^X t/bin/simple_filter",
+ );
+}
+
DBICTest::Schema::1->connect($make_dbictest_db::dsn);
-my $foo = slurp("$dump_path/DBICTest/Schema/1/Result/Foo.pm");
+# 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"},
"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); }