diff options
Diffstat (limited to 'javaparser-testing/src/test/java/com/github/javaparser/builders/EnumDeclarationBuildersTest.java')
-rw-r--r-- | javaparser-testing/src/test/java/com/github/javaparser/builders/EnumDeclarationBuildersTest.java | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/javaparser-testing/src/test/java/com/github/javaparser/builders/EnumDeclarationBuildersTest.java b/javaparser-testing/src/test/java/com/github/javaparser/builders/EnumDeclarationBuildersTest.java new file mode 100644 index 000000000..048307d75 --- /dev/null +++ b/javaparser-testing/src/test/java/com/github/javaparser/builders/EnumDeclarationBuildersTest.java @@ -0,0 +1,56 @@ +/* + * 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.EnumDeclaration; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; + +import java.util.function.Function; + +import static com.github.javaparser.utils.Utils.EOL; +import static org.junit.Assert.assertEquals; + +public class EnumDeclarationBuildersTest { + private final CompilationUnit cu = new CompilationUnit(); + + @Test + public void testAddImplements() { + EnumDeclaration testEnum = cu.addEnum("test"); + testEnum.addImplementedType(Function.class); + assertEquals(1, cu.getImports().size()); + assertEquals("import " + Function.class.getName() + ";" + EOL, + cu.getImport(0).toString()); + assertEquals(1, testEnum.getImplementedTypes().size()); + assertEquals(Function.class.getSimpleName(), testEnum.getImplementedTypes(0).getNameAsString()); + } + + @Test + public void testAddEnumConstant() { + EnumDeclaration testEnum = cu.addEnum("test"); + testEnum.addEnumConstant("MY_ENUM_CONSTANT"); + assertEquals(1, testEnum.getEntries().size()); + assertEquals("MY_ENUM_CONSTANT", testEnum.getEntry(0).getNameAsString()); + } +} |