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); }