Tested yet another way to manage transactions through the user transaction.
[utils] / 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.getPassword(), persistenceUnit);
56         builder.start();
57
58         assertEquals(0, db.getActiveConnections());
59         dbtester = dbutils.createDbTester(new MyTables());
60     }
61
62     @After
63     public void tearDown() throws Exception {
64         dbtester.getConnection().close();
65         builder.stop();
66         dbutils.stop();
67         assertEquals(0, db.getActiveConnections());
68         db.stop();
69     }
70
71     @Test
72     public void testTablesCorrect() throws Exception {
73         String[] tables = dbutils.getTableNames(new MyTables());
74         assertEquals(1, tables.length);
75         assertEquals("XYZ_MYENTITY", tables[0]);
76     }
77
78     @Test
79     public void testDeleteTables() throws Exception {
80         String[] tables = dbutils.getTableNames(new MyTables());
81         assertEquals(1, tables.length);
82         assertEquals("XYZ_MYENTITY", tables[0]);
83
84         // Put some data in the database.
85         builder.execute(new JpaUnitOfWork<Void>() {
86             public Void execute(EntityManager aEm) {
87                 MyEntity entity = new MyEntity("a", "b");
88                 aEm.persist(entity);
89                 return null;
90             }
91         });
92
93         // Verify one row is written (using Db unit)
94         ITable table = dbtester.getDataSet().getTable("XYZ_MYENTITY");
95         assertEquals(1, table.getRowCount());
96
97         // Clean the database
98         dbutils.cleanDatabase(new MyTables());
99         table = dbtester.getDataSet().getTable("XYZ_MYENTITY");
100         assertEquals(0, table.getRowCount());
101
102         // Now drop the database
103         dbutils.dropTables(new MyTables());
104         dbutils.dropTables(JpaCustomizerBuilder.getCustomizer().getJpaTables());
105         tables = dbutils.getTableNames(new MyTables());
106         assertEquals(0, tables.length);
107     }
108
109 }