public boolean containsField(String name) {
for (FieldModel fieldModel : fields) {
if (fieldModel.getName().equals(name)) {
return true;
}
}
return false;
}
@Test
public void testWithFields() {
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"
+ "}\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("Long", "id");
methodModel.addModifiers(Modifier.PUBLIC);
model.addFields(methodModel);
String resultClass = model.toString();
assertEquals(expectedClass, resultClass);
}
