X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=test%2Fenterprise%2Fsrc%2Ftest%2Fjava%2Forg%2Fwamblee%2Fsupport%2Fpersistence%2FMyEntityExampleTestBase.java;h=649508e658b76f0431284a5984d75568a7a743e1;hb=dec278a67997ea8e85d10662e31548afd8890ed3;hp=ec6b268f98b25f513f9b84444f9fbc3894f76832;hpb=8de36ff0206c996baf3ee4adc3e2293b12ff5f39;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 ec6b268f..649508e6 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
@@ -1,24 +1,31 @@
+/*
+ * Copyright 2005-2010 the original author or authors.
+ * 
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */ 
 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.
  */
@@ -54,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();
     }
 
@@ -70,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;
             }
         });