github/jooq
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[#1105] Add integration tests for multi-schema source code generation and querying - partial commit

Browse Source
This commit is contained in:
Lukas Eder 2012-02-05 13:21:17 +00:00
parent 43b781c21d
commit 453d0e0b86
3 changed files with 102 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

144
jOOQ-test/src/org/jooq/test/jOOQAbstractTest.java
View File

@ -64,6 +64,7 @@ import static org.jooq.tools.unsigned.Unsigned.ushort;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
@ -254,27 +255,34 @@ public abstract class jOOQAbstractTest<
protected static boolean initialised;
protected static boolean reset;
protected static Connection connection;
protected static Connection connectionMultiSchema;
protected static boolean autocommit;
protected static String jdbcURL;
protected static String jdbcSchema;
protected static Map<String, String> scripts = new HashMap<String, String>();
protected void execute(String script) throws Exception {
protected void execute(String script, Connection con) throws Exception {
Statement stmt = null;
String allSQL = scripts.get(script);
if (allSQL == null) {
File file = new File(getClass().getResource(script).toURI());
allSQL = FileUtils.readFileToString(file);
try {
File file = new File(getClass().getResource(script).toURI());
allSQL = FileUtils.readFileToString(file);
testSQLWatch.splitDebug("Loaded SQL file");
}
catch (IOException ignore) {
allSQL = "";
}
scripts.put(script, allSQL);
testSQLWatch.splitDebug("Loaded SQL file");
}
for (String sql : allSQL.split("/")) {
try {
if (!StringUtils.isBlank(sql)) {
sql = sql.replace("{" + JDBC_SCHEMA + "}", jdbcSchema);
stmt = connection.createStatement();
stmt = con.createStatement();
stmt.execute(sql.trim());
testSQLWatch.splitDebug(StringUtils.abbreviate(sql.trim().replaceAll("[\\n\\r]|\\s+", " "), 25));
}
@ -377,16 +385,20 @@ public abstract class jOOQAbstractTest<
@Before
public void setUp() throws Exception {
connection = getConnection();
connectionMultiSchema = getConnectionMultiSchema();
autocommit = connection.getAutoCommit();
if (!initialised) {
initialised = true;
execute(getCreateScript());
execute(getCreateScript(), connection);
execute(getCreateMultiSchemaScript(), connectionMultiSchema);
}
if (!reset) {
reset = true;
execute(getResetScript());
execute(getResetScript(), connection);
execute(getResetMultiSchemaScript(), connectionMultiSchema);
}
}
@ -426,60 +438,72 @@ public abstract class jOOQAbstractTest<
protected final Connection getConnection() {
if (connection == null) {
try {
String property = System.getProperty("jdbc.properties");
if (property == null) {
log.error("No system property 'jdbc.properties' found");
log.error("-----------");
log.error("Please be sure property is set; example: -Djdbc.properties=/org/jooq/configuration/${env_var:USERNAME}/db2/library.properties");
throw new Error();
}
InputStream in = GenerationTool.class.getResourceAsStream(property);
if (in == null) {
log.error("Cannot find " + property);
log.error("-----------");
log.error("Please be sure it is located on the classpath and qualified as a classpath location.");
log.error("If it is located at the current working directory, try adding a '/' to the path");
throw new Error();
}
Properties properties = new Properties();
try {
properties.load(in);
}
finally {
in.close();
}
String driver = properties.getProperty(JDBC_DRIVER);
jdbcURL = properties.getProperty(JDBC_URL) + getSchemaSuffix();
String jdbcUser = properties.getProperty(JDBC_USER);
String jdbcPassword = properties.getProperty(JDBC_PASSWORD);
jdbcSchema = properties.getProperty(JDBC_SCHEMA) + getSchemaSuffix();
Class.forName(driver);
if (StringUtils.isBlank(jdbcUser)) {
Properties p = new Properties();
if (getClass().getSimpleName().toLowerCase().contains("ingres")) {
p.setProperty("timezone", "EUROPE-CENTRAL");
}
connection = DriverManager.getConnection(getJdbcURL(), p);
}
else {
connection = DriverManager.getConnection(getJdbcURL(), jdbcUser, jdbcPassword);
}
}
catch (Exception e) {
throw new Error(e);
}
connection = getConnection0(null, null);
}
return connection;
}
protected final Connection getConnectionMultiSchema() {
if (connectionMultiSchema == null) {
connectionMultiSchema = getConnection0("MULTI_SCHEMA", "MULTI_SCHEMA");
}
return connectionMultiSchema;
}
private final Connection getConnection0(String jdbcUser, String jdbcPassword) {
try {
String property = System.getProperty("jdbc.properties");
if (property == null) {
log.error("No system property 'jdbc.properties' found");
log.error("-----------");
log.error("Please be sure property is set; example: -Djdbc.properties=/org/jooq/configuration/${env_var:USERNAME}/db2/library.properties");
throw new Error();
}
InputStream in = GenerationTool.class.getResourceAsStream(property);
if (in == null) {
log.error("Cannot find " + property);
log.error("-----------");
log.error("Please be sure it is located on the classpath and qualified as a classpath location.");
log.error("If it is located at the current working directory, try adding a '/' to the path");
throw new Error();
}
Properties properties = new Properties();
try {
properties.load(in);
}
finally {
in.close();
}
String driver = properties.getProperty(JDBC_DRIVER);
jdbcURL = properties.getProperty(JDBC_URL) + getSchemaSuffix();
jdbcUser = jdbcUser != null ? jdbcUser : properties.getProperty(JDBC_USER);
jdbcPassword = jdbcPassword != null ? jdbcPassword : properties.getProperty(JDBC_PASSWORD);
jdbcSchema = properties.getProperty(JDBC_SCHEMA) + getSchemaSuffix();
Class.forName(driver);
if (StringUtils.isBlank(jdbcUser)) {
Properties p = new Properties();
if (getClass().getSimpleName().toLowerCase().contains("ingres")) {
p.setProperty("timezone", "EUROPE-CENTRAL");
}
return DriverManager.getConnection(getJdbcURL(), p);
}
else {
return DriverManager.getConnection(getJdbcURL(), jdbcUser, jdbcPassword);
}
}
catch (Exception e) {
throw new Error(e);
}
}
/**
* Gets the jdbc url.
*
@ -499,6 +523,14 @@ public abstract class jOOQAbstractTest<
return "/org/jooq/test/" + getDialect().getName().toLowerCase() + "/reset.sql";
}
protected final String getCreateMultiSchemaScript() throws Exception {
return "/org/jooq/test/" + getDialect().getName().toLowerCase() + "/create-multi-schema.sql";
}
protected final String getResetMultiSchemaScript() throws Exception {
return "/org/jooq/test/" + getDialect().getName().toLowerCase() + "/reset-multi-schema.sql";
}
protected abstract Table<T658> T658();
protected abstract Table<T725> T725();
protected abstract TableField<T725, Integer> T725_ID();

13
jOOQ-test/src/org/jooq/test/oracle/create-multi-schema.sql Normal file
View File

@ -0,0 +1,13 @@
DROP TABLE MULTI_SCHEMA.T_BOOK_SALE/
CREATE TABLE MULTI_SCHEMA.T_BOOK_SALE (
ID NUMBER(7) NOT NULL,
BOOK_ID NUMBER(7) NOT NULL,
BOOK_STORE_NAME VARCHAR2(400) NOT NULL,
SOLD_AT DATE NOT NULL,
SOLD_FOR NUMBER(10, 2) NOT NULL,
CONSTRAINT pk_t_book_sale PRIMARY KEY (ID),
CONSTRAINT fk_t_book_to_book_store FOREIGN KEY (BOOK_ID, BOOK_STORE_NAME) REFERENCES TEST.T_BOOK_TO_BOOK_STORE(BOOK_ID, BOOK_STORE_NAME) ON DELETE CASCADE
)
/

1
jOOQ-test/src/org/jooq/test/oracle/reset-multi-schema.sql Normal file
View File

@ -0,0 +1 @@
DELETE FROM MULTI_SCHEMA.T_BOOK_SALE/