More test hackage, some cleanup in ResultSet

[dbsrgits/DBIx-Class.git] / t / run / 04db.tl

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