code style improvements.
[utils] / test / enterprise / src / test / java / org / wamblee / support / persistence / MyEntityExampleTestBase.java
index 8bf094b3fa451c132d9f866ec4c4b54d807e2226..649508e658b76f0431284a5984d75568a7a743e1 100644 (file)
  */ 
 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,13 @@ 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();
     }
 
     @After
     public void tearDown() {
+        dbutils.stop();
         jpaTester.stop();
     }
 
@@ -85,9 +78,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;
             }
         });