TestElementsAnnotatedWith.java revision 0:9a66ca7c79fa
1/* 2 * Copyright 2006-2007 Sun Microsystems, Inc. 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara, 20 * CA 95054 USA or visit www.sun.com if you need additional information or 21 * have any questions. 22 */ 23 24/* 25 * @test 26 * @bug 6397298 6400986 6425592 6449798 6453386 6508401 27 * @summary Tests that getElementsAnnotatedWith works properly. 28 * @author Joseph D. Darcy 29 * @compile TestElementsAnnotatedWith.java 30 * @compile InheritedAnnotation.java 31 * @compile -processor TestElementsAnnotatedWith -proc:only SurfaceAnnotations.java 32 * @compile -processor TestElementsAnnotatedWith -proc:only BuriedAnnotations.java 33 * @compile -processor TestElementsAnnotatedWith -proc:only Part1.java Part2.java 34 * @compile -processor TestElementsAnnotatedWith -proc:only TestElementsAnnotatedWith.java 35 * @compile -processor TestElementsAnnotatedWith -proc:only C2.java 36 */ 37 38import java.lang.annotation.Annotation; 39import java.util.Collections; 40import java.util.Set; 41import java.util.HashSet; 42import java.util.Arrays; 43import javax.annotation.processing.*; 44import javax.lang.model.SourceVersion; 45import javax.lang.model.element.*; 46import javax.lang.model.util.*; 47import static javax.lang.model.util.ElementFilter.*; 48 49/** 50 * This processor verifies that the information returned by 51 * getElementsAnnotatedWith is consistent with the expected results 52 * stored in an AnnotatedElementInfo annotation. 53 */ 54@SupportedAnnotationTypes("*") 55@AnnotatedElementInfo(annotationName="java.lang.SuppressWarnings", expectedSize=0, names={}) 56public class TestElementsAnnotatedWith extends AbstractProcessor { 57 58 public boolean process(Set<? extends TypeElement> annotations, 59 RoundEnvironment roundEnvironment) { 60 Elements elementUtils = processingEnv.getElementUtils(); 61 62 TypeElement annotatedElementInfoElement = 63 elementUtils.getTypeElement("AnnotatedElementInfo"); 64 Set<? extends Element> resultsMeta = Collections.emptySet(); 65 Set<? extends Element> resultsBase = Collections.emptySet(); 66 67 if (!roundEnvironment.processingOver()) { 68 testNonAnnotations(roundEnvironment); 69 70 // Verify AnnotatedElementInfo is present on the first 71 // specified type. 72 73 TypeElement firstType = typesIn(roundEnvironment.getRootElements()).iterator().next(); 74 75 AnnotatedElementInfo annotatedElementInfo = firstType.getAnnotation(AnnotatedElementInfo.class); 76 77 boolean failed = false; 78 79 if (annotatedElementInfo == null) 80 throw new IllegalArgumentException("Missing AnnotatedElementInfo annotation on " + 81 firstType); 82 else { 83 // Verify that the annotation information is as 84 // expected. 85 86 Set<String> expectedNames = new HashSet<String>(Arrays.asList(annotatedElementInfo.names())); 87 88 resultsMeta = 89 roundEnvironment. 90 getElementsAnnotatedWith(elementUtils. 91 getTypeElement(annotatedElementInfo. 92 annotationName())) ; 93 94 System.err.println("Results: " + resultsMeta); 95 96 if (resultsMeta.size() != annotatedElementInfo.expectedSize()) { 97 failed = true; 98 System.err.printf("Bad number of elements; expected %d, got %d%n", 99 annotatedElementInfo.expectedSize(), resultsMeta.size()); 100 } else { 101 for(Element element : resultsMeta) { 102 String simpleName = element.getSimpleName().toString(); 103 if (!expectedNames.contains(simpleName) ) { 104 failed = true; 105 System.err.println("Name ``" + simpleName + "'' not expected."); 106 } 107 } 108 } 109 } 110 111 resultsBase = computeResultsBase(roundEnvironment, annotatedElementInfo.annotationName()); 112 113 if (!resultsMeta.equals(resultsBase)) { 114 failed = true; 115 System.err.println("Base and Meta sets unequal;\n meta: " + resultsMeta + 116 "\nbase: " + resultsBase); 117 } 118 119 if (failed) { 120 System.err.println("AnnotatedElementInfo: " + annotatedElementInfo); 121 throw new RuntimeException(); 122 } 123 } else { 124 // If processing is over without an error, the specified 125 // elements should be empty so an empty set should be returned. 126 resultsMeta = roundEnvironment.getElementsAnnotatedWith(annotatedElementInfoElement); 127 resultsBase = roundEnvironment.getElementsAnnotatedWith(AnnotatedElementInfo.class); 128 if (!resultsMeta.isEmpty()) 129 throw new RuntimeException("Nonempty resultsMeta: " + resultsMeta); 130 if (!resultsBase.isEmpty()) 131 throw new RuntimeException("Nonempty resultsBase: " + resultsBase); 132 133 } 134 return true; 135 } 136 137 private Set<? extends Element> computeResultsBase(RoundEnvironment roundEnvironment, String name) { 138 try { 139 return roundEnvironment. 140 getElementsAnnotatedWith(Class.forName(name).asSubclass(Annotation.class)); 141 } catch(ClassNotFoundException cnfe) { 142 throw new RuntimeException(cnfe); 143 } 144 } 145 146 /** 147 * Verify non-annotation types result in 148 * IllegalArgumentExceptions. 149 */ 150 private void testNonAnnotations(RoundEnvironment roundEnvironment) { 151 try { 152 Set<? extends Element> elements = roundEnvironment.getElementsAnnotatedWith((Class)Object.class ); 153 throw new RuntimeException("Illegal argument exception not thrown"); 154 } catch(IllegalArgumentException iae) {} 155 156 try { 157 Set<? extends Element> elements = roundEnvironment.getElementsAnnotatedWith(processingEnv. 158 getElementUtils(). 159 getTypeElement("java.lang.Object") ); 160 throw new RuntimeException("Illegal argument exception not thrown"); 161 } catch(IllegalArgumentException iae) {} 162 } 163 164 @Override 165 public SourceVersion getSupportedSourceVersion() { 166 return SourceVersion.latest(); 167 } 168} 169