2 * Copyright 2005-2010 the original author or authors.
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
16 package org.wamblee.test.persistence;
18 import static junit.framework.Assert.assertEquals;
20 import javax.persistence.EntityManager;
21 import javax.sql.DataSource;
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;
36 public class DatabaseUtilsTestBase {
38 private DataSource dataSource;
39 private PersistenceUnitDescription persistenceUnit;
40 private JpaBuilder builder;
41 private DatabaseUtils dbutils;
42 private IDatabaseTester dbtester;
45 public void setUp() throws Exception {
46 db = DatabaseBuilder.getDatabase();
47 dataSource = db.start();
49 persistenceUnit = new MyPersistenceUnit();
51 dbutils = new DatabaseUtils(dataSource);
52 dbutils.dropTables(new MyTables());
53 dbutils.dropTables(JpaCustomizerBuilder.getCustomizer().getJpaTables());
55 builder = new JpaBuilder(db.getJdbcUrl(), db.getUsername(), db
56 .getPassword(), persistenceUnit);
59 assertEquals(0, db.getActiveConnections());
60 dbtester = dbutils.createDbTester(new MyTables());
64 public void tearDown() throws Exception {
65 dbtester.getConnection().close();
68 assertEquals(0, db.getActiveConnections());
73 public void testTablesCorrect() throws Exception {
74 String[] tables = dbutils.getTableNames(new MyTables());
75 assertEquals(1, tables.length);
76 assertEquals("XYZ_MYENTITY", tables[0]);
80 public void testDeleteTables() throws Exception {
81 String[] tables = dbutils.getTableNames(new MyTables());
82 assertEquals(1, tables.length);
83 assertEquals("XYZ_MYENTITY", tables[0]);
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");
94 // Verify one row is written (using Db unit)
95 ITable table = dbtester.getDataSet().getTable("XYZ_MYENTITY");
96 assertEquals(1, table.getRowCount());
99 dbutils.cleanDatabase(new MyTables());
100 table = dbtester.getDataSet().getTable("XYZ_MYENTITY");
101 assertEquals(0, table.getRowCount());
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);