@Test
public void testSimple() {
String expectedClass =
"package com.optible.processor.model;\n\n"
+ "import org.junit.Test;\n\n"
+ "public class ClassModelTest {\n"
+ "}\n";
TopLevelModel model = new TopLevelModel("com.optible.processor.model", "ClassModelTest");
model.addModifiers(Modifier.PUBLIC);
model.addImports("org.junit.Test");
String resultClass = model.toString();
assertEquals(expectedClass, resultClass);
}
@Test
public void testWithGenericsExtends() {
String expectedClass =
"package com.optible.processor.model;\n\n"
+ "import org.junit.Test;\n\n"
+ "public class ClassModelTest extends BaseTest<View.view, Presenter.presenter> {\n"
+ "}\n";
TopLevelModel model = new TopLevelModel("com.optible.processor.model", "ClassModelTest");
model.addExtends("BaseTest<View.view, Presenter.presenter>");
model.addModifiers(Modifier.PUBLIC);
model.addImports("org.junit.Test");
String resultClass = model.toString();
assertEquals(expectedClass, resultClass);
}
@Test
public void testWithExtendsWithPath() {
String expectedClass =
"package com.optible.processor.model;\n\n"
+ "import test.BaseTest;\nimport org.junit.Test;\n\n"
+ "public class ClassModelTest extends BaseTest {\n"
+ "}\n";
TopLevelModel model = new TopLevelModel("com.optible.processor.model", "ClassModelTest");
model.addExtends(new TypeModel("test", "BaseTest"));
model.addModifiers(Modifier.PUBLIC);
model.addImports("org.junit.Test");
String resultClass = model.toString();
assertEquals(expectedClass, resultClass);
}
@Test
public void testWithAnnotations() {
String expectedClass =
"package com.optible.processor.model;\n\n"
+ "import org.junit.Test;\n\n"
+ "@org.junit.Test\n@Generated\n"
+ "public class ClassModelTest {\n"
+ "}\n";
TopLevelModel model = new TopLevelModel("com.optible.processor.model", "ClassModelTest");
model.addModifiers(Modifier.PUBLIC);
model.addImports("org.junit.Test");
model.addAnnotations(new AnnotationModel("org.junit.Test"), new AnnotationModel("Generated"));
String resultClass = model.toString();
assertEquals(expectedClass, resultClass);
}
@Test
public void testWithFieldWithPackage() {
String expectedClass =
"package com.optible.processor.model;\n\n"
+ "import org.junit.Test;\n\n"
+ "@org.junit.Test\n"
+ "public class ClassModelTest {\n\n"
+ " public Long id;\n Long id2;\n"
+ "}\n";
TopLevelModel model = new TopLevelModel("com.optible.processor.model", "ClassModelTest");
model.addModifiers(Modifier.PUBLIC);
model.addImports("org.junit.Test");
model.addAnnotations(new AnnotationModel("org.junit.Test"));
FieldModel methodModel = new FieldModel("java.lang", "Long", "id");
methodModel.addModifiers(Modifier.PUBLIC);
model.addFields(methodModel);
FieldModel methodModel2 = new FieldModel("java.lang", "Long", "id2");
model.addFields(methodModel2);
String resultClass = model.toString();
assertEquals(expectedClass, resultClass);
}
@Override
public void addMethod(MethodModel method) {
method.setIncludeBody(false);
super.addMethod(method);
}
