1 /* 2 * Copyright (c) 2006, 2014, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 /* 25 * @test 26 * @bug 6397298 6400986 6425592 6449798 6453386 6508401 6498938 6911854 8030049 8038080 27 * @summary Tests that getElementsAnnotatedWith works properly. 28 * @author Joseph D. Darcy 29 * @library /tools/javac/lib 30 * @build JavacTestingAbstractProcessor 31 * @compile TestElementsAnnotatedWith.java 32 * @compile InheritedAnnotation.java 33 * @compile TpAnno.java 34 * @compile Anno.java 35 * @compile -processor TestElementsAnnotatedWith -proc:only SurfaceAnnotations.java 36 * @compile -processor TestElementsAnnotatedWith -proc:only BuriedAnnotations.java 37 * @compile -processor TestElementsAnnotatedWith -proc:only Part1.java Part2.java 38 * @compile -processor TestElementsAnnotatedWith -proc:only C2.java 39 * @compile -processor TestElementsAnnotatedWith -proc:only Foo.java 40 * @compile -processor TestElementsAnnotatedWith -proc:only TypeParameterAnnotations.java 41 * @compile -processor TestElementsAnnotatedWith -proc:only ParameterAnnotations.java 42 * @compile/fail/ref=ErroneousAnnotations.out -processor TestElementsAnnotatedWith -proc:only -XDrawDiagnostics ErroneousAnnotations.java 43 * @compile Foo.java 44 * @compile/process -processor TestElementsAnnotatedWith -proc:only Foo 45 */ 46 47 import java.lang.annotation.Annotation; 48 import java.util.Collections; 49 import java.util.Set; 50 import java.util.HashSet; 51 import java.util.Arrays; 52 import javax.annotation.processing.*; 53 import javax.lang.model.element.*; 54 import static javax.lang.model.util.ElementFilter.*; 55 56 /** 57 * This processor verifies that the information returned by 58 * getElementsAnnotatedWith is consistent with the expected results 59 * stored in an AnnotatedElementInfo annotation. 60 */ 61 @AnnotatedElementInfo(annotationName="java.lang.SuppressWarnings", expectedSize=0, names={}) 62 public class TestElementsAnnotatedWith extends JavacTestingAbstractProcessor { 63 64 public boolean process(Set<? extends TypeElement> annotations, 65 RoundEnvironment roundEnvironment) { 66 TypeElement annotatedElementInfoElement = 67 elements.getTypeElement("AnnotatedElementInfo"); 68 Set<? extends Element> resultsMeta = Collections.emptySet(); 69 Set<? extends Element> resultsBase = Collections.emptySet(); 70 71 if (!roundEnvironment.processingOver()) { 72 testNonAnnotations(roundEnvironment); 73 74 // Verify AnnotatedElementInfo is present on the first 75 // specified type. 76 77 TypeElement firstType = typesIn(roundEnvironment.getRootElements()).iterator().next(); 78 79 AnnotatedElementInfo annotatedElementInfo = firstType.getAnnotation(AnnotatedElementInfo.class); 80 81 boolean failed = false; 82 83 if (annotatedElementInfo == null) 84 throw new IllegalArgumentException("Missing AnnotatedElementInfo annotation on " + 85 firstType); 86 else { 87 // Verify that the annotation information is as 88 // expected. 89 90 Set<String> expectedNames = new HashSet<String>(Arrays.asList(annotatedElementInfo.names())); 91 92 resultsMeta = 93 roundEnvironment. 94 getElementsAnnotatedWith(elements.getTypeElement(annotatedElementInfo.annotationName())); 95 96 System.err.println("Results: " + resultsMeta); 97 98 if (resultsMeta.size() != annotatedElementInfo.expectedSize()) { 99 failed = true; 100 System.err.printf("Bad number of elements; expected %d, got %d%n", 101 annotatedElementInfo.expectedSize(), resultsMeta.size()); 102 } else { 103 for(Element element : resultsMeta) { 104 String simpleName = element.getSimpleName().toString(); 105 if (!expectedNames.contains(simpleName) ) { 106 failed = true; 107 System.err.println("Name ``" + simpleName + "'' not expected."); 108 } 109 } 110 } 111 } 112 113 resultsBase = computeResultsBase(roundEnvironment, annotatedElementInfo.annotationName()); 114 115 if (!resultsMeta.equals(resultsBase)) { 116 failed = true; 117 System.err.println("Base and Meta sets unequal;\n meta: " + resultsMeta + 118 "\nbase: " + resultsBase); 119 } 120 121 if (failed) { 122 System.err.println("AnnotatedElementInfo: " + annotatedElementInfo); 123 throw new RuntimeException(); 124 } 125 } else { 126 // If processing is over without an error, the specified 127 // elements should be empty so an empty set should be returned. 128 resultsMeta = roundEnvironment.getElementsAnnotatedWith(annotatedElementInfoElement); 129 resultsBase = roundEnvironment.getElementsAnnotatedWith(AnnotatedElementInfo.class); 130 if (!resultsMeta.isEmpty()) 131 throw new RuntimeException("Nonempty resultsMeta: " + resultsMeta); 132 if (!resultsBase.isEmpty()) 133 throw new RuntimeException("Nonempty resultsBase: " + resultsBase); 134 135 } 136 return true; 137 } 138 139 private Set<? extends Element> computeResultsBase(RoundEnvironment roundEnvironment, String name) { 140 try { 141 return roundEnvironment. 142 getElementsAnnotatedWith(Class.forName(name).asSubclass(Annotation.class)); 143 } catch(ClassNotFoundException cnfe) { 144 throw new RuntimeException(cnfe); 145 } 146 } 147 148 /** 149 * Verify non-annotation types result in 150 * IllegalArgumentExceptions. 151 */ 152 private void testNonAnnotations(RoundEnvironment roundEnvironment) { 153 try { 154 Set<? extends Element> elements = roundEnvironment.getElementsAnnotatedWith((Class)Object.class ); 155 throw new RuntimeException("Illegal argument exception not thrown"); 156 } catch(IllegalArgumentException iae) {} 157 158 try { 159 Set<? extends Element> elements = 160 roundEnvironment.getElementsAnnotatedWith(processingEnv. 161 getElementUtils(). 162 getTypeElement("java.lang.Object") ); 163 throw new RuntimeException("Illegal argument exception not thrown"); 164 } catch(IllegalArgumentException iae) {} 165 } 166 }