Fix building on perls with no . in @INC
[dbsrgits/DBIx-Class.git] / t / storage / dbh_do.t
index 0beb858..07453ea 100644 (file)
@@ -1,13 +1,17 @@
+BEGIN { do "./t/lib/ANFANG.pm" or die ( $@ || $! ) }
+
 use strict;
 use warnings;
 
 use Test::More;
-use lib qw(t/lib);
+
 use DBICTest;
 
 
 my $schema = DBICTest->init_schema();
 my $storage = $schema->storage;
+$storage = $storage->master
+  if $storage->isa('DBIx::Class::Storage::DBI::Replicated');
 
 # test (re)connection
 for my $disconnect (0, 1) {
@@ -43,7 +47,7 @@ is_deeply (
 
 # test nested aliasing
 my $res = 'original';
-$storage->dbh_do (sub {
+$schema->storage->dbh_do (sub {
   shift->dbh_do(sub { $_[3] = 'changed' }, @_)
 }, $res);