X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=t%2Frun%2F04db.tl;fp=t%2Frun%2F04db.tl;h=e1da6b495e73327e0ad1932c15d21542d40bc0df;hb=1edaf6fef6739e4d809ae6adfbeaa6b660ca29b3;hp=1b4089dd5ae3a3b835a11cace31969d83a1cb306;hpb=9ce9ec1036c3a67b4087b3f4a90bca09463c1e03;p=dbsrgits%2FDBIx-Class.git

diff --git a/t/run/04db.tl b/t/run/04db.tl
index 1b4089d..e1da6b4 100644
--- a/t/run/04db.tl
+++ b/t/run/04db.tl
@@ -1,30 +1,31 @@
 sub run_tests {
+my $schema = shift;
   
 plan tests => 2;
 
 # add some rows inside a transaction and commit it
 # XXX: Is storage->dbh the only way to get a dbh?
-DBICTest->class("Artist")->txn_begin;
+$schema->class("Artist")->txn_begin;
 for (10..15) {
-    DBICTest->class("Artist")->create( { 
+    $schema->class("Artist")->create( { 
         artistid => $_,
         name => "artist number $_",
     } );
 }
-DBICTest->class("Artist")->txn_commit;
-my ($artist) = DBICTest->class("Artist")->find(15);
+$schema->class("Artist")->txn_commit;
+my ($artist) = $schema->class("Artist")->find(15);
 is($artist->name, 'artist number 15', "Commit ok");
 
 # add some rows inside a transaction and roll it back
-DBICTest->class("Artist")->txn_begin;
+$schema->class("Artist")->txn_begin;
 for (21..30) {
-    DBICTest->class("Artist")->create( {
+    $schema->class("Artist")->create( {
         artistid => $_,
         name => "artist number $_",
     } );
 }
-DBICTest->class("Artist")->txn_rollback;
-($artist) = DBICTest->class("Artist")->search( artistid => 25 );
+$schema->class("Artist")->txn_rollback;
+($artist) = $schema->class("Artist")->search( artistid => 25 );
 is($artist, undef, "Rollback ok");
 
 }