X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=t%2Flib%2FDBICTest%2FSchema.pm;h=2e783a75912d6f8d1cd593f2b79e630cacac03e0;hb=2c649fafe996a7ee5b85b229a9ed2128b0e7ee1f;hp=5d2518accfbff62fd2766a41023f550f356ee0df;hpb=6918c70e3970b631dd6f4e298a87ae02476fbde1;p=dbsrgits%2FDBIx-Class.git

diff --git a/t/lib/DBICTest/Schema.pm b/t/lib/DBICTest/Schema.pm
index 5d2518a..2e783a7 100644
--- a/t/lib/DBICTest/Schema.pm
+++ b/t/lib/DBICTest/Schema.pm
@@ -5,10 +5,7 @@ use strict;
 use warnings;
 no warnings 'qw';
 
-use base 'DBIx::Class::Schema';
-
-use DBICTest::Util qw/populate_weakregistry assert_empty_weakregistry/;
-use namespace::clean;
+use base 'DBICTest::BaseSchema';
 
 __PACKAGE__->mk_group_accessors(simple => 'custom_attr');
 
@@ -54,7 +51,7 @@ __PACKAGE__->load_classes(qw/
     'CD_to_Producer',
     'Dummy',    # this is a real result class we remove in the hook below
   ),
-  qw/SelfRefAlias TreeLike TwoKeyTreeLike Event EventTZ NoPrimaryKey/,
+  qw/SelfRefAlias TreeLike TwoKeyTreeLike Event NoPrimaryKey/,
   qw/Collection CollectionObject TypedObject Owners BooksInLibrary/,
   qw/ForceForeign Encoded/,
 );
@@ -65,39 +62,4 @@ sub sqlt_deploy_hook {
   $sqlt_schema->drop_table('dummy');
 }
 
-my $weak_registry = {};
-
-sub clone {
-  my $self = shift->next::method(@_);
-  populate_weakregistry ( $weak_registry, $self )
-    if $INC{'Test/Builder.pm'};
-  $self;
-}
-
-sub connection {
-  my $self = shift->next::method(@_);
-
-  if ($INC{'Test/Builder.pm'}) {
-    populate_weakregistry ( $weak_registry, $self->storage );
-
-    my $cur_connect_call = $self->storage->on_connect_call;
-
-    $self->storage->on_connect_call([
-      (ref $cur_connect_call eq 'ARRAY'
-        ? @$cur_connect_call
-        : ($cur_connect_call || ())
-      ),
-      [sub {
-        populate_weakregistry( $weak_registry, shift->_dbh )
-      }],
-    ]);
-  }
-
-  $self;
-}
-
-END {
-  assert_empty_weakregistry($weak_registry, 'quiet');
-}
-
 1;