1 /* 2 * Copyright (C) 2007-2010 Júlio Vilmar Gesser. 3 * Copyright (C) 2011, 2013-2016 The JavaParser Team. 4 * 5 * This file is part of JavaParser. 6 * 7 * JavaParser can be used either under the terms of 8 * a) the GNU Lesser General Public License as published by 9 * the Free Software Foundation, either version 3 of the License, or 10 * (at your option) any later version. 11 * b) the terms of the Apache License 12 * 13 * You should have received a copy of both licenses in LICENCE.LGPL and 14 * LICENCE.APACHE. Please refer to those files for details. 15 * 16 * JavaParser is distributed in the hope that it will be useful, 17 * but WITHOUT ANY WARRANTY; without even the implied warranty of 18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 19 * GNU Lesser General Public License for more details. 20 */ 21 22 package com.github.javaparser.builders; 23 24 import com.github.javaparser.ast.CompilationUnit; 25 import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration; 26 import com.github.javaparser.ast.expr.NormalAnnotationExpr; 27 import com.github.javaparser.ast.expr.SingleMemberAnnotationExpr; 28 import org.junit.After; 29 import org.junit.Before; 30 import org.junit.Test; 31 32 import static org.junit.Assert.assertEquals; 33 import static org.junit.Assert.assertTrue; 34 35 public class NodeWithAnnotationsBuildersTest { 36 private CompilationUnit cu = new CompilationUnit(); 37 private ClassOrInterfaceDeclaration testClass = cu.addClass("testClass"); ; 38 39 @interface hey { 40 41 } 42 43 @Test testAddAnnotation()44 public void testAddAnnotation() { 45 NormalAnnotationExpr annotation = testClass.addAndGetAnnotation(hey.class); 46 assertEquals("import com.github.javaparser.builders.NodeWithAnnotationsBuildersTest.hey;", cu.getImport(0).toString().trim()); 47 assertEquals(1, testClass.getAnnotations().size()); 48 assertEquals(annotation, testClass.getAnnotation(0)); 49 assertEquals(NormalAnnotationExpr.class, testClass.getAnnotation(0).getClass()); 50 } 51 52 @Test testAddMarkerAnnotation()53 public void testAddMarkerAnnotation() { 54 testClass.addMarkerAnnotation("test"); 55 assertEquals(1, testClass.getAnnotations().size()); 56 } 57 58 @Test testAddSingleMemberAnnotation()59 public void testAddSingleMemberAnnotation() { 60 testClass.addSingleMemberAnnotation("test", "value"); 61 assertEquals(1, testClass.getAnnotations().size()); 62 assertEquals("value", testClass.getAnnotation(0).asSingleMemberAnnotationExpr().getMemberValue().toString()); 63 } 64 65 @Test testIsAnnotationPresent()66 public void testIsAnnotationPresent() { 67 testClass.addMarkerAnnotation(hey.class); 68 assertTrue(testClass.isAnnotationPresent(hey.class)); 69 } 70 71 @Test testGetAnnotationByName()72 public void testGetAnnotationByName() { 73 NormalAnnotationExpr annotation = testClass.addAndGetAnnotation(hey.class); 74 assertEquals(annotation, testClass.getAnnotationByName("hey").get()); 75 } 76 77 @Test testGetAnnotationByClass()78 public void testGetAnnotationByClass() { 79 NormalAnnotationExpr annotation = testClass.addAndGetAnnotation(hey.class); 80 assertEquals(annotation, testClass.getAnnotationByClass(hey.class).get()); 81 } 82 } 83