wamblee.org Git - utils/blob - test/enterprise/src/test/java/org/wamblee/test/persistence/DatabaseUtilsTestBase.java

   1 /*
   2  * Copyright 2005-2010 the original author or authors.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package org.wamblee.test.persistence;
  17
  18 import static junit.framework.Assert.assertEquals;
  19
  20 import javax.persistence.EntityManager;
  21 import javax.sql.DataSource;
  22
  23 import org.dbunit.IDatabaseTester;
  24 import org.dbunit.dataset.ITable;
  25 import org.junit.After;
  26 import org.junit.Before;
  27 import org.junit.Test;
  28 import org.wamblee.test.persistence.Database;
  29 import org.wamblee.test.persistence.DatabaseBuilder;
  30 import org.wamblee.test.persistence.DatabaseUtils;
  31 import org.wamblee.test.persistence.JpaBuilder;
  32 import org.wamblee.test.persistence.JpaCustomizerBuilder;
  33 import org.wamblee.test.persistence.PersistenceUnitDescription;
  34 import org.wamblee.test.persistence.JpaBuilder.JpaUnitOfWork;
  35
  36 public class DatabaseUtilsTestBase {
  37     private Database db;
  38     private DataSource dataSource;
  39     private PersistenceUnitDescription persistenceUnit;
  40     private JpaBuilder builder;
  41     private DatabaseUtils dbutils;
  42     private IDatabaseTester dbtester;
  43
  44     @Before
  45     public void setUp() throws Exception {
  46         db = DatabaseBuilder.getDatabase();
  47         dataSource = db.start();
  48
  49         persistenceUnit = new MyPersistenceUnit();
  50
  51         dbutils = new DatabaseUtils(dataSource);
  52         dbutils.dropTables(new MyTables());
  53         dbutils.dropTables(JpaCustomizerBuilder.getCustomizer().getJpaTables());
  54
  55         builder = new JpaBuilder(db.getJdbcUrl(), db.getUsername(), db
  56             .getPassword(), persistenceUnit);
  57         builder.start();
  58
  59         assertEquals(0, db.getActiveConnections());
  60         dbtester = dbutils.createDbTester(new MyTables());
  61     }
  62
  63     @After
  64     public void tearDown() throws Exception {
  65         dbtester.getConnection().close();
  66         builder.stop();
  67         dbutils.stop();
  68         assertEquals(0, db.getActiveConnections());
  69         db.stop();
  70     }
  71
  72     @Test
  73     public void testTablesCorrect() throws Exception {
  74         String[] tables = dbutils.getTableNames(new MyTables());
  75         assertEquals(1, tables.length);
  76         assertEquals("XYZ_MYENTITY", tables[0]);
  77     }
  78
  79     @Test
  80     public void testDeleteTables() throws Exception {
  81         String[] tables = dbutils.getTableNames(new MyTables());
  82         assertEquals(1, tables.length);
  83         assertEquals("XYZ_MYENTITY", tables[0]);
  84
  85         // Put some data in the database.
  86         builder.execute(new JpaUnitOfWork<Void>() {
  87             public Void execute(EntityManager aEm) {
  88                 MyEntity entity = new MyEntity("a", "b");
  89                 aEm.persist(entity);
  90                 return null;
  91             }
  92         });
  93
  94         // Verify one row is written (using Db unit)
  95         ITable table = dbtester.getDataSet().getTable("XYZ_MYENTITY");
  96         assertEquals(1, table.getRowCount());
  97
  98         // Clean the database
  99         dbutils.cleanDatabase(new MyTables());
 100         table = dbtester.getDataSet().getTable("XYZ_MYENTITY");
 101         assertEquals(0, table.getRowCount());
 102
 103         // Now drop the database
 104         dbutils.dropTables(new MyTables());
 105         dbutils.dropTables(JpaCustomizerBuilder.getCustomizer().getJpaTables());
 106         tables = dbutils.getTableNames(new MyTables());
 107         assertEquals(0, tables.length);
 108     }
 109
 110 }