protected void tearDown() throws Exception {
if (runTestsForDatastore()) {
clean(SamplePoint.class);
clean(SampleLineString.class);
clean(SamplePolygon.class);
clean(SampleGeometryCollection.class);
}
super.tearDown();
}
protected void setUp() throws Exception {
super.setUp();
if (runTestsForDatastore()) {
PersistenceManager pm = pmf.getPersistenceManager();
Transaction tx = pm.currentTransaction();
try {
tx.begin();
pm.newQuery(SamplePoint.class).deletePersistentAll();
pm.newQuery(SampleLineString.class).deletePersistentAll();
pm.newQuery(SamplePolygon.class).deletePersistentAll();
pm.newQuery(SampleGeometryCollection.class).deletePersistentAll();
} finally {
tx.commit();
}
Connection sqlConn = null;
try {
tx.begin();
sqlConn = (Connection) pm.getDataStoreConnection();
String connUrl = pmf.getConnectionURL().toLowerCase();
String fileName = "sample_jts_postgis.sql";
if (connUrl.contains("mysql") || connUrl.contains("mariadb")) {
fileName = "sample_jts_mysql.sql";
}
File file =
new File(
PgGeometryMySQLTest.class
.getResource("/org/datanucleus/samples/data/" + fileName)
.toURI());
String s = "";
InputStream is = new FileInputStream(file);
int c;
while ((c = is.read()) != -1) {
s += (char) c;
}
String ss[] = StringUtils.split(s, ";");
for (int i = 0; i < ss.length; i++) {
sqlConn.createStatement().execute(ss[i]);
}
is.close();
} finally {
sqlConn.close();
tx.commit();
}
}
}