Add loader_roles option

[dbsrgits/DBIx-Class-Schema-Loader.git] / t / 24loader_subclass.t

diff --git a/t/24loader_subclass.t b/t/24loader_subclass.t
index 66d24df..2d3b595 100644 (file)
--- a/t/24loader_subclass.t
+++ b/t/24loader_subclass.t
@@ -4,14 +4,17 @@ use Test::More;
 use lib qw(t/lib);
 use make_dbictest_db;
 
-my %loader_class = ( 'TestLoaderSubclass' => 'TestLoaderSubclass',
-                     '::DBI::SQLite'      => 'DBIx::Class::Schema::Loader::DBI::SQLite'
-                   );
+my %loader_class = (
+    'TestLoaderSubclass'           => 'TestLoaderSubclass',
+    'TestLoaderSubclass_NoRebless' => 'TestLoaderSubclass_NoRebless',
+    '::DBI::SQLite'                => 'DBIx::Class::Schema::Loader::DBI::SQLite'
+);
 
 my %invocations = (
     loader_class => sub {
         package DBICTest::Schema::1;
         use base qw/ DBIx::Class::Schema::Loader /;
+        __PACKAGE__->_loader_invoked(0);
         __PACKAGE__->naming('current');
         __PACKAGE__->loader_class(shift);
         __PACKAGE__->connect($make_dbictest_db::dsn);
@@ -19,6 +22,7 @@ my %invocations = (
     connect_info => sub {
         package DBICTeset::Schema::2;
         use base qw/ DBIx::Class::Schema::Loader /;
+        __PACKAGE__->_loader_invoked(0);
         __PACKAGE__->naming('current');
         __PACKAGE__->connect($make_dbictest_db::dsn, { loader_class => shift });
     },
@@ -39,6 +43,6 @@ while (my ($style,$subref) = each %invocations) {
     while (my ($arg, $class) = each %loader_class) {
         my $schema = $subref->($arg);
         $schema = $schema->clone unless ref $schema;
-        isa_ok($schema->_loader, $class, "$style($arg)");
+        isa_ok($schema->loader, $class, "$style($arg)");
     }
 }