persistenceUnit = new MyPersistenceUnit();
- dbutils = new DatabaseUtils(dataSource, persistenceUnit.getTables());
- dbutils.dropTables();
+ dbutils = new DatabaseUtils(dataSource);
+ dbutils.dropTables(new MyTables());
dbutils.dropTables(JpaCustomizerBuilder.getCustomizer().getJpaTables());
builder = new JpaBuilder(db.getJdbcUrl(), db.getUsername(), db.getPassword(), persistenceUnit);
builder.start();
assertEquals(0, db.getActiveConnections());
- dbtester = dbutils.createDbTester();
+ dbtester = dbutils.createDbTester(new MyTables());
}
@After
@Test
public void testTablesCorrect() throws Exception {
- String[] tables = dbutils.getTableNames();
+ String[] tables = dbutils.getTableNames(new MyTables());
assertEquals(1, tables.length);
assertEquals("XYZ_MYENTITY", tables[0]);
}
@Test
public void testDeleteTables() throws Exception {
- String[] tables = dbutils.getTableNames();
+ String[] tables = dbutils.getTableNames(new MyTables());
assertEquals(1, tables.length);
assertEquals("XYZ_MYENTITY", tables[0]);
assertEquals(1, table.getRowCount());
// Clean the database
- dbutils.cleanDatabase();
+ dbutils.cleanDatabase(new MyTables());
table = dbtester.getDataSet().getTable("XYZ_MYENTITY");
assertEquals(0, table.getRowCount());
// Now drop the database
- dbutils.dropTables();
+ dbutils.dropTables(new MyTables());
dbutils.dropTables(JpaCustomizerBuilder.getCustomizer().getJpaTables());
- tables = dbutils.getTableNames();
+ tables = dbutils.getTableNames(new MyTables());
assertEquals(0, tables.length);
}