/*
* 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.codegen;
import java.io.IOException;
import java.io.Serializable;
import java.io.StringWriter;
import java.io.Writer;
import java.util.Arrays;
import java.util.Date;
import com.mysema.codegen.JavaWriter;
import com.mysema.codegen.StringUtils;
import com.mysema.codegen.model.*;
import org.junit.Before;
import org.junit.Test;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
public class BeanSerializerTest {
private Type typeModel;
private EntityType type;
private final Writer writer = new StringWriter();
@Before
public void setUp() {
typeModel = new SimpleType(TypeCategory.ENTITY, "com.mysema.query.DomainClass", "com.mysema.query", "DomainClass", false,false);
type = new EntityType(typeModel);
}
@Test
public void Annotations() throws IOException{
type.addAnnotation(new QueryEntityImpl());
BeanSerializer serializer = new BeanSerializer();
serializer.serialize(type, SimpleSerializerConfig.DEFAULT, new JavaWriter(writer));
String str = writer.toString();
assertTrue(str.contains("import com.mysema.query.annotations.QueryEntity;"));
assertTrue(str.contains("@QueryEntity"));
}
@Test
public void Annotated_Property() throws IOException{
Property property = new Property(type, "entityField", type);
property.addAnnotation(new QueryEntityImpl());
type.addProperty(property);
BeanSerializer serializer = new BeanSerializer();
serializer.serialize(type, SimpleSerializerConfig.DEFAULT, new JavaWriter(writer));
String str = writer.toString();
assertTrue(str.contains("import com.mysema.query.annotations.QueryEntity;"));
assertTrue(str.contains("@QueryEntity"));
}
@Test
public void Annotated_Property_Not_Serialized() throws IOException{
Property property = new Property(type, "entityField", type);
property.addAnnotation(new QueryEntityImpl());
type.addProperty(property);
BeanSerializer serializer = new BeanSerializer(false);
serializer.serialize(type, SimpleSerializerConfig.DEFAULT, new JavaWriter(writer));
String str = writer.toString();
assertFalse(str.contains("import com.mysema.query.annotations.QueryEntity;"));
assertFalse(str.contains("@QueryEntity"));
}
@Test
public void Capitalization() throws IOException{
// property
type.addProperty(new Property(type, "cId", type));
BeanSerializer serializer = new BeanSerializer();
serializer.serialize(type, SimpleSerializerConfig.DEFAULT, new JavaWriter(writer));
assertTrue(writer.toString().contains("public DomainClass getcId() {"));
}
@Test
public void Interfaces() throws IOException {
BeanSerializer serializer = new BeanSerializer();
serializer.addInterface(new ClassType(Serializable.class));
serializer.serialize(type, SimpleSerializerConfig.DEFAULT, new JavaWriter(writer));
assertTrue(writer.toString().contains("public class DomainClass implements Serializable {"));
}
@Test
public void Interfaces2() throws IOException {
BeanSerializer serializer = new BeanSerializer();
serializer.addInterface(Serializable.class);
serializer.serialize(type, SimpleSerializerConfig.DEFAULT, new JavaWriter(writer));
assertTrue(writer.toString().contains("public class DomainClass implements Serializable {"));
}
@Test
public void ToString() throws IOException{
// property
type.addProperty(new Property(type, "entityField", type));
type.addProperty(new Property(type, "collection", new SimpleType(Types.COLLECTION, typeModel)));
type.addProperty(new Property(type, "listField", new SimpleType(Types.LIST, typeModel)));
type.addProperty(new Property(type, "setField", new SimpleType(Types.SET, typeModel)));
type.addProperty(new Property(type, "arrayField", new ClassType(TypeCategory.ARRAY, String[].class)));
type.addProperty(new Property(type, "mapField", new SimpleType(Types.MAP, typeModel, typeModel)));
BeanSerializer serializer = new BeanSerializer();
serializer.setAddToString(true);
serializer.serialize(type, SimpleSerializerConfig.DEFAULT, new JavaWriter(writer));
//System.out.println(writer.toString());
}
@Test
public void FullConstructor() throws IOException{
// property
type.addProperty(new Property(type, "entityField", type));
type.addProperty(new Property(type, "collection", new SimpleType(Types.COLLECTION, typeModel)));
type.addProperty(new Property(type, "listField", new SimpleType(Types.LIST, typeModel)));
type.addProperty(new Property(type, "setField", new SimpleType(Types.SET, typeModel)));
type.addProperty(new Property(type, "arrayField", new ClassType(TypeCategory.ARRAY, String[].class)));
type.addProperty(new Property(type, "mapField", new SimpleType(Types.MAP, typeModel, typeModel)));
BeanSerializer serializer = new BeanSerializer();
serializer.setAddFullConstructor(true);
serializer.serialize(type, SimpleSerializerConfig.DEFAULT, new JavaWriter(writer));
//System.out.println(writer.toString());
}
@Test
public void Properties() throws IOException{
// property
type.addProperty(new Property(type, "entityField", type));
type.addProperty(new Property(type, "collection", new SimpleType(Types.COLLECTION, typeModel)));
type.addProperty(new Property(type, "listField", new SimpleType(Types.LIST, typeModel)));
type.addProperty(new Property(type, "setField", new SimpleType(Types.SET, typeModel)));
type.addProperty(new Property(type, "arrayField", new ClassType(TypeCategory.ARRAY, String[].class)));
type.addProperty(new Property(type, "mapField", new SimpleType(Types.MAP, typeModel, typeModel)));
for (Class<?> cl : Arrays.<Class<?>>asList(Boolean.class, Comparable.class, Integer.class, Date.class, java.sql.Date.class, java.sql.Time.class)) {
Type classType = new ClassType(TypeCategory.get(cl.getName()), cl);
type.addProperty(new Property(type, StringUtils.uncapitalize(cl.getSimpleName()), classType));
}
BeanSerializer serializer = new BeanSerializer();
serializer.serialize(type, SimpleSerializerConfig.DEFAULT, new JavaWriter(writer));
String str = writer.toString();
//System.err.println(str);
for (String prop : Arrays.asList(
"String[] arrayField;",
"Boolean boolean$;",
"Collection<DomainClass> collection;",
"Comparable comparable;",
"java.util.Date date;",
"DomainClass entityField;",
"Integer integer;",
"List<DomainClass> listField;",
"Map<DomainClass, DomainClass> mapField;",
"Set<DomainClass> setField;",
"java.sql.Time time;")) {
assertTrue(prop + " was not contained", str.contains(prop));
}
}
}