diff options
Diffstat (limited to 'javaparser-testing/src/test/java/com/github/javaparser/builders/FieldDeclarationBuildersTest.java')
-rw-r--r-- | javaparser-testing/src/test/java/com/github/javaparser/builders/FieldDeclarationBuildersTest.java | 120 |
1 files changed, 120 insertions, 0 deletions
diff --git a/javaparser-testing/src/test/java/com/github/javaparser/builders/FieldDeclarationBuildersTest.java b/javaparser-testing/src/test/java/com/github/javaparser/builders/FieldDeclarationBuildersTest.java new file mode 100644 index 000000000..7824addd1 --- /dev/null +++ b/javaparser-testing/src/test/java/com/github/javaparser/builders/FieldDeclarationBuildersTest.java @@ -0,0 +1,120 @@ +/* + * Copyright (C) 2007-2010 JĂșlio Vilmar Gesser. + * Copyright (C) 2011, 2013-2016 The JavaParser Team. + * + * This file is part of JavaParser. + * + * JavaParser can be used either under the terms of + * a) the GNU Lesser General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * b) the terms of the Apache License + * + * You should have received a copy of both licenses in LICENCE.LGPL and + * LICENCE.APACHE. Please refer to those files for details. + * + * JavaParser is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Lesser General Public License for more details. + */ + +package com.github.javaparser.builders; + +import com.github.javaparser.ast.CompilationUnit; +import com.github.javaparser.ast.body.*; +import com.github.javaparser.ast.stmt.ExpressionStmt; +import com.github.javaparser.ast.stmt.ReturnStmt; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; + +import java.util.List; + +import static com.github.javaparser.ast.type.PrimitiveType.intType; +import static org.junit.Assert.assertEquals; + +public class FieldDeclarationBuildersTest { + private final CompilationUnit cu = new CompilationUnit(); + private ClassOrInterfaceDeclaration testClass = cu.addClass("testClass"); + private EnumDeclaration testEnum = cu.addEnum("testEnum"); + + @Test(expected = IllegalStateException.class) + public void testOrphanFieldGetter() { + new FieldDeclaration().createGetter(); + } + + @Test(expected = IllegalStateException.class) + public void testOrphanFieldSetter() { + new FieldDeclaration().createSetter(); + } + + @Test + public void testCreateGetterInAClass() { + testClass.addPrivateField(int.class, "myField").createGetter(); + assertEquals(2, testClass.getMembers().size()); + assertEquals(MethodDeclaration.class, testClass.getMember(1).getClass()); + List<MethodDeclaration> methodsWithName = testClass.getMethodsByName("getMyField"); + assertEquals(1, methodsWithName.size()); + MethodDeclaration getter = methodsWithName.get(0); + assertEquals("getMyField", getter.getNameAsString()); + assertEquals("int", getter.getType().toString()); + assertEquals(ReturnStmt.class, getter.getBody().get().getStatement(0).getClass()); + } + + @Test + public void testCreateSetterInAClass() { + testClass.addPrivateField(int.class, "myField").createSetter(); + assertEquals(2, testClass.getMembers().size()); + assertEquals(MethodDeclaration.class, testClass.getMember(1).getClass()); + List<MethodDeclaration> methodsWithName = testClass.getMethodsByName("setMyField"); + assertEquals(1, methodsWithName.size()); + MethodDeclaration setter = methodsWithName.get(0); + assertEquals("setMyField", setter.getNameAsString()); + assertEquals("int", setter.getParameter(0).getType().toString()); + assertEquals(ExpressionStmt.class, setter.getBody().get().getStatement(0).getClass()); + assertEquals("this.myField = myField;", setter.getBody().get().getStatement(0).toString()); + } + + @Test + public void testCreateGetterInEnum() { + testEnum.addPrivateField(int.class, "myField").createGetter(); + assertEquals(2, testEnum.getMembers().size()); + assertEquals(MethodDeclaration.class, testEnum.getMember(1).getClass()); + List<MethodDeclaration> methodsWithName = testEnum.getMethodsByName("getMyField"); + assertEquals(1, methodsWithName.size()); + MethodDeclaration getter = methodsWithName.get(0); + assertEquals("getMyField", getter.getNameAsString()); + assertEquals("int", getter.getType().toString()); + assertEquals(ReturnStmt.class, getter.getBody().get().getStatement(0).getClass()); + } + + @Test + public void testCreateSetterInEnum() { + testEnum.addPrivateField(int.class, "myField").createSetter(); + assertEquals(2, testEnum.getMembers().size()); + assertEquals(MethodDeclaration.class, testEnum.getMember(1).getClass()); + List<MethodDeclaration> methodsWithName = testEnum.getMethodsByName("setMyField"); + assertEquals(1, methodsWithName.size()); + MethodDeclaration setter = methodsWithName.get(0); + assertEquals("setMyField", setter.getNameAsString()); + assertEquals("int", setter.getParameter(0).getType().toString()); + assertEquals(ExpressionStmt.class, setter.getBody().get().getStatement(0).getClass()); + assertEquals("this.myField = myField;", setter.getBody().get().getStatement(0).toString()); + } + + @Test(expected = IllegalStateException.class) + public void testCreateGetterWithANonValidField() { + FieldDeclaration myPrivateField = testClass.addPrivateField(int.class, "myField"); + myPrivateField.getVariables().add(new VariableDeclarator(intType(), "secondField")); + myPrivateField.createGetter(); + } + + @Test(expected = IllegalStateException.class) + public void testCreateSetterWithANonValidField() { + FieldDeclaration myPrivateField = testClass.addPrivateField(int.class, "myField"); + myPrivateField.getVariables().add(new VariableDeclarator(intType(), "secondField")); + myPrivateField.createSetter(); + } + +} |