public void testDrop() throws Exception {
_tool.createTestSchema();
createTestTable();
_tool.dropTestSchema();
// The table should no longer be there
try {
_tool.executeSql(
_tool.getTestCatalog(), _tool.getTestSchema(), "SELECT * FROM " + TEST_TABLE);
fail();
} catch (OpenGammaRuntimeException e) {
// Ok - no table should be there!
}
}
private void createTestTable() throws SQLException {
Table table = new Table(TEST_TABLE);
try {
String dropSql =
table.sqlDropString(_tool.getHibernateDialect(), null, _tool.getTestSchema());
_tool.executeSql(_tool.getTestCatalog(), _tool.getTestSchema(), dropSql);
} catch (OpenGammaRuntimeException e) {
// It might not exist, that's OK
}
String createSql =
"CREATE TABLE "
+ table.getQualifiedName(_tool.getHibernateDialect(), null, _tool.getTestSchema())
+ " (test_column char(50))";
_tool.executeSql(_tool.getTestCatalog(), _tool.getTestSchema(), createSql);
}
@BeforeMethod(alwaysRun = true)
public void setUp() throws Exception {
Properties props = TestProperties.getTestProperties();
String dbHost = props.getProperty("utildb.jdbc.url");
String user = props.getProperty("utildb.jdbc.username");
String password = props.getProperty("utildb.jdbc.password");
_tool = new DbTool(dbHost, user, password);
_tool.initialize();
}
public void testClear() throws Exception {
_tool.createTestSchema();
createTestTable();
_tool.executeSql(
_tool.getTestCatalog(),
_tool.getTestSchema(),
"INSERT INTO " + TEST_TABLE + " (test_column) VALUES ('test')");
_tool.clearTestTables();
Connection connection = getConnection();
Statement statement = connection.createStatement();
ResultSet rs = statement.executeQuery("SELECT COUNT(*) FROM " + TEST_TABLE);
if (rs.next()) {
int count = rs.getInt(1);
assertEquals(0, count);
} else {
fail();
}
rs.close();
statement.close();
connection.close();
_tool.dropTestSchema();
}
private Connection getConnection() throws SQLException {
return DriverManager.getConnection(
_tool.getDbServerHost() + "/" + _tool.getTestCatalog(),
_tool.getUser(),
_tool.getPassword());
}
