X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=test%2Fenterprise%2Fsrc%2Ftest%2Fjava%2Forg%2Fwamblee%2Fsupport%2Fpersistence%2FMyEntityExampleTestBase.java;h=c3771775363558a507f74f3a4f318946e332e615;hb=c63e76e400f06f51ba235b9e6658bfa3149c1d48;hp=8bf094b3fa451c132d9f866ec4c4b54d807e2226;hpb=17775e14ecfb286e59f67117e5cee7e21e95ab1f;p=utils

diff --git a/test/enterprise/src/test/java/org/wamblee/support/persistence/MyEntityExampleTestBase.java b/test/enterprise/src/test/java/org/wamblee/support/persistence/MyEntityExampleTestBase.java
index 8bf094b3..c3771775 100644
--- a/test/enterprise/src/test/java/org/wamblee/support/persistence/MyEntityExampleTestBase.java
+++ b/test/enterprise/src/test/java/org/wamblee/support/persistence/MyEntityExampleTestBase.java
@@ -15,25 +15,17 @@
  */ 
 package org.wamblee.support.persistence;
 
+import static junit.framework.Assert.*;
+
 import javax.persistence.EntityManager;
-import javax.persistence.Persistence;
-import javax.sql.DataSource;
 
-import org.dbunit.DataSourceDatabaseTester;
-import org.dbunit.DatabaseTestCase;
 import org.dbunit.IDatabaseTester;
 import org.dbunit.dataset.ITable;
-import org.dbunit.dataset.filter.ITableFilterSimple;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
-import org.wamblee.support.persistence.DatabaseUtils;
-import org.wamblee.support.persistence.JpaBuilder;
-import org.wamblee.support.persistence.JpaTester;
 import org.wamblee.support.persistence.JpaBuilder.JpaUnitOfWork;
 
-import static junit.framework.Assert.*;
-
 /**
  * This class shows an example of how to test an entity using jpa.
  */
@@ -69,12 +61,14 @@ public class MyEntityExampleTestBase {
         // but we don't need it. We can use datbase utils if we want to execute
         // straight JDBC calls.
         builder = jpaTester.getJpaBuilder();
-        dbtester = jpaTester.getDbTester();
+        dbtester = jpaTester.getDbUtils().createDbTester(new MyTables());
         dbutils = jpaTester.getDbUtils();
+        dbutils.start();
     }
 
     @After
     public void tearDown() {
+        dbutils.stop();
         jpaTester.stop();
     }
 
@@ -85,9 +79,9 @@ public class MyEntityExampleTestBase {
         // as and execute the
         // unit of work.
         builder.execute(new JpaUnitOfWork<Void>() {
-            public Void execute(EntityManager em) {
+            public Void execute(EntityManager aEm) {
                 MyEntity entity = new MyEntity("a", "b");
-                em.persist(entity);
+                aEm.persist(entity);
                 return null;
             }
         });