/*
* Copyright 2011, Mysema Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.mysema.query.sql.codegen;
import javax.tools.JavaCompiler;
import java.io.File;
import java.io.IOException;
import java.sql.*;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Set;
import com.mysema.codegen.SimpleCompiler;
import com.mysema.query.codegen.BeanSerializer;
import com.mysema.query.codegen.Serializer;
import com.mysema.util.FileUtils;
import junit.framework.Assert;
import org.junit.*;
import static org.junit.Assert.*;
public class MetaDataExporterTest {
private static final List<Serializer> BEAN_SERIALIZERS = Arrays.<Serializer>asList(
new BeanSerializer());
private static Connection connection;
private Statement statement;
private Serializer beanSerializer;
private boolean clean = true;
private boolean exportColumns = false;
private boolean schemaToPackage = false;
private DatabaseMetaData metadata;
private JavaCompiler compiler = new SimpleCompiler();
@BeforeClass
public static void setUpClass() throws ClassNotFoundException, SQLException{
Class.forName("org.h2.Driver");
String url = "jdbc:h2:mem:testdb" + System.currentTimeMillis();
connection = DriverManager.getConnection(url, "sa", "");
Statement stmt = connection.createStatement();
try{
// reserved words
stmt.execute("create table reserved (id int, while int)");
// underscore
stmt.execute("create table underscore (e_id int, c_id int)");
// bean generation
stmt.execute("create table beangen1 (\"SEP_Order\" int)");
// default instance clash
stmt.execute("create table definstance (id int, definstance int, definstance1 int)");
// class with pk and fk classes
stmt.execute("create table pkfk (id int primary key, pk int, fk int)");
// camel case
stmt.execute("create table \"camelCase\" (id int)");
stmt.execute("create table \"vwServiceName\" (id int)");
// simple types
stmt.execute("create table date_test (d date)");
stmt.execute("create table date_time_test (dt datetime)");
// complex type
stmt.execute("create table survey (id int, name varchar(30))");
// new line
stmt.execute("create table \"new\nline\" (id int)");
stmt.execute("create table newline2 (id int, \"new\nline\" int)");
stmt.execute("create table employee("
+ "id INT, "
+ "firstname VARCHAR(50), "
+ "lastname VARCHAR(50), "
+ "salary DECIMAL(10, 2), "
+ "datefield DATE, "
+ "timefield TIME, "
+ "superior_id int, "
+ "survey_id int, "
+ "survey_name varchar(30), "
+ "CONSTRAINT PK_employee PRIMARY KEY (id), "
+ "CONSTRAINT FK_superior FOREIGN KEY (superior_id) REFERENCES employee(id))");
// multi key
stmt.execute("create table multikey(id INT, id2 VARCHAR, id3 INT, CONSTRAINT pk_multikey PRIMARY KEY (id, id2, id3) )");
// M_PRODUCT_BOM_ID
stmt.execute("create table product(id int, "
+ "m_product_bom_id int, "
+ "m_productbom_id int, "
+ "constraint product_bom foreign key (m_productbom_id) references product(id))");
}finally{
stmt.close();
}
}
@AfterClass
public static void tearDownClass() throws SQLException{
connection.close();
}
@Before
public void setUp() throws ClassNotFoundException, SQLException {
statement = connection.createStatement();
metadata = connection.getMetaData();
}
@After
public void tearDown() throws SQLException{
statement.close();
}
private static final NamingStrategy defaultNaming = new DefaultNamingStrategy();
private static final NamingStrategy originalNaming = new OriginalNamingStrategy();
private String beanPackageName = null;
@Test
public void NormalSettings_Repetition() throws SQLException {
test("Q", "", "", "", defaultNaming, "target/1", false, false, false);
File file = new File("target/1/test/QEmployee.java");
long lastModified = file.lastModified();
assertTrue(file.exists());
clean = false;
test("Q", "", "", "", defaultNaming, "target/1", false, false, false);
assertEquals(lastModified, file.lastModified());
}
@Test
public void Multiple() throws SQLException {
// TODO : refactor this to use new JUnit constructs
List<String> emptyString = Collections.singletonList("");
boolean[] trueAndFalse = new boolean[]{true, false};
int counter = 0;
for (boolean withBeans : trueAndFalse) {
for (boolean withInnerClasses : trueAndFalse) {
for (boolean schemaToPackage : trueAndFalse) {
for (boolean exportColumns : trueAndFalse) {
for (boolean withOriginalPositioning : trueAndFalse) {
for (NamingStrategy ns : Arrays.asList(defaultNaming, originalNaming)) {
for (String namePrefix : Arrays.asList("", "Q", "Query")) {
for (String nameSuffix : Arrays.asList("", "Type")) {
for (String beanPrefix : withBeans ? Arrays.asList("", "Bean") : emptyString) {
for (String beanSuffix : withBeans ? Arrays.asList("", "Bean") : emptyString) {
for (String beanPackage : withBeans ? Arrays.asList("test2", null) : emptyString) {
for (Serializer beanSerializer : BEAN_SERIALIZERS) {
counter++;
this.beanPackageName = beanPackage;
this.schemaToPackage = schemaToPackage;
this.exportColumns = exportColumns;
this.beanSerializer = beanSerializer;
test(namePrefix, nameSuffix, beanPrefix, beanSuffix,
ns, "target/multiple_"+counter, withBeans, withInnerClasses, withOriginalPositioning);
}}}}}}}}}}}}
}
@Test
public void Explicit_Configuration() throws SQLException{
MetaDataExporter exporter = new MetaDataExporter();
exporter.setSchemaPattern("PUBLIC");
exporter.setNamePrefix("Q");
exporter.setPackageName("test");
exporter.setTargetFolder(new File("target/7"));
exporter.setNamingStrategy(new DefaultNamingStrategy());
exporter.setBeanSerializer(new BeanSerializer());
exporter.setBeanPackageName("test2");
exporter.export(metadata);
assertTrue(new File("target/7/test/QDateTest.java").exists());
assertTrue(new File("target/7/test2/DateTest.java").exists());
}
@Test
public void Minimal_Configuration() throws SQLException{
MetaDataExporter exporter = new MetaDataExporter();
exporter.setSchemaPattern("PUBLIC");
exporter.setPackageName("test");
exporter.setTargetFolder(new File("target/8"));
exporter.export(metadata);
assertTrue(new File("target/8/test/QDateTest.java").exists());
}
@Test
public void Minimal_Configuration_with_tables() throws SQLException{
MetaDataExporter exporter = new MetaDataExporter();
exporter.setSchemaPattern("PUBLIC");
exporter.setTableNamePattern("RESERVED,UNDERSCORE,BEANGEN1");
exporter.setPackageName("test");
exporter.setTargetFolder(new File("target/82"));
exporter.export(metadata);
assertTrue(new File("target/82/test/QBeangen1.java").exists());
assertTrue(new File("target/82/test/QReserved.java").exists());
assertTrue(new File("target/82/test/QUnderscore.java").exists());
assertFalse(new File("target/82/test/QDefinstance.java").exists());
}
@Test
public void Minimal_Configuration_with_Suffix() throws SQLException{
MetaDataExporter exporter = new MetaDataExporter();
exporter.setSchemaPattern("PUBLIC");
exporter.setPackageName("test");
exporter.setNamePrefix("");
exporter.setNameSuffix("Type");
exporter.setTargetFolder(new File("target/9"));
exporter.export(metadata);
assertTrue(new File("target/9/test/DateTestType.java").exists());
}
@Test
public void Minimal_Configuration_without_keys() throws SQLException{
MetaDataExporter exporter = new MetaDataExporter();
exporter.setSchemaPattern("PUBLIC");
exporter.setPackageName("test");
exporter.setNamePrefix("");
exporter.setNameSuffix("Type");
exporter.setTargetFolder(new File("target/10"));
exporter.setExportForeignKeys(false);
exporter.export(metadata);
assertTrue(new File("target/10/test/DateTestType.java").exists());
}
@Test
public void Minimal_Configuration_with_Bean_prefix() throws SQLException{
MetaDataExporter exporter = new MetaDataExporter();
exporter.setSchemaPattern("PUBLIC");
exporter.setPackageName("test");
exporter.setNamePrefix("");
exporter.setBeanPrefix("Bean");
exporter.setBeanSerializer(new BeanSerializer());
exporter.setTargetFolder(new File("target/a"));
exporter.export(metadata);
assertTrue(new File("target/a/test/DateTest.java").exists());
assertTrue(new File("target/a/test/BeanDateTest.java").exists());
}
@Test
public void Minimal_Configuration_with_Bean_suffix() throws SQLException{
MetaDataExporter exporter = new MetaDataExporter();
exporter.setSchemaPattern("PUBLIC");
exporter.setPackageName("test");
exporter.setNamePrefix("");
exporter.setBeanSuffix("Bean");
exporter.setBeanSerializer(new BeanSerializer());
exporter.setTargetFolder(new File("target/b"));
exporter.export(metadata);
assertTrue(new File("target/b/test/DateTest.java").exists());
assertTrue(new File("target/b/test/DateTestBean.java").exists());
}
private void test(String namePrefix, String nameSuffix, String beanPrefix, String beanSuffix,
NamingStrategy namingStrategy, String target, boolean withBeans,
boolean withInnerClasses, boolean withOrdinalPositioning) throws SQLException{
File targetDir = new File(target);
if (clean) {
try {
if (targetDir.exists()) {
FileUtils.delete(targetDir);
}
} catch (IOException e) {
throw new RuntimeException(e);
}
}
MetaDataExporter exporter = new MetaDataExporter();
exporter.setColumnAnnotations(exportColumns);
exporter.setSchemaPattern("PUBLIC");
exporter.setNamePrefix(namePrefix);
exporter.setNameSuffix(nameSuffix);
exporter.setBeanPrefix(beanPrefix);
exporter.setBeanSuffix(beanSuffix);
exporter.setInnerClassesForKeys(withInnerClasses);
exporter.setPackageName("test");
exporter.setBeanPackageName(beanPackageName);
exporter.setTargetFolder(targetDir);
exporter.setNamingStrategy(namingStrategy);
exporter.setSchemaToPackage(schemaToPackage);
if (withBeans) {
exporter.setBeanSerializer(beanSerializer);
}
if (withOrdinalPositioning) {
exporter.setColumnComparatorClass(OrdinalPositionComparator.class);
}
exporter.export(metadata);
Set<String> classes = exporter.getClasses();
int compilationResult = compiler.run(null, System.out, System.err,
classes.toArray(new String[classes.size()]));
if (compilationResult != 0) {
Assert.fail("Compilation Failed for " + target);
}
}
}