X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=t%2Frow%2Finflate_result.t;h=ab35f86b919995e081f43ab3cd0a17a7d7cd6ac6;hb=d9c17594ab3be0b866c555750cdbd1ad6a1b34e6;hp=370c205b7e452b152b394bb5855c518d0b984a0a;hpb=de621449d983d7dbc139687a1d9f157524009942;p=dbsrgits%2FDBIx-Class.git

diff --git a/t/row/inflate_result.t b/t/row/inflate_result.t
index 370c205..ab35f86 100644
--- a/t/row/inflate_result.t
+++ b/t/row/inflate_result.t
@@ -2,7 +2,6 @@ package My::Schema::Result::User;
 
 use strict;
 use warnings;
-use DBIx::Class::ResultClass::HashRefInflator;
 use base qw/DBIx::Class::Core/;
 
 ### Define what our admin class is, for ensure_class_loaded()
@@ -21,7 +20,6 @@ __PACKAGE__->set_primary_key('user_id');
 sub inflate_result {
     my $self = shift;
     my $ret  = $self->next::method(@_);
-    
     if ( $ret->admin ) {    ### If this is an admin, rebless for extra functions
         $self->ensure_class_loaded($admin_class);
         bless $ret, $admin_class;
@@ -63,7 +61,6 @@ package main;
 
 use lib qw(t/lib);
 use DBICTest;
-use DBIx::Class::ResultClass::HashRefInflator;
 
 use Test::More;
 
@@ -79,7 +76,7 @@ my $admin_data = {
     admin    => 1
 };
 
-ok( my $schema = My::Schema->connection('dbi:SQLite:dbname=:memory:') );
+ok( my $schema = My::Schema->connection(DBICTest->_database) );
 
 ok(
     $schema->storage->dbh->do(
@@ -87,16 +84,13 @@ ok(
     )
 );
 
-my $rs = $schema->resultset('User');
-
 TODO: {
     local $TODO = 'New objects should also be inflated';
-    my $user  = $rs->create($user_data);
-    my $admin = $rs->create($admin_data);
+    my $user  = $schema->resultset('User')->create($user_data);
+    my $admin = $schema->resultset('User')->create($admin_data);
 
     is( ref $user,  'My::Schema::Result::User' );
     is( ref $admin, 'My::Schema::Result::User::Admin' );
-
 }
 
 my $user  = $schema->resultset('User')->single($user_data);