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
38 import java.lang.annotation.Annotation;
39 import java.util.Collections;
40 import java.util.Set;
41 import java.util.HashSet;
42 import java.util.Arrays;
43 import javax.annotation.processing.*;
44 import javax.lang.model.SourceVersion;
45 import javax.lang.model.element.*;
46 import javax.lang.model.util.*;
47 import 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={})
56 public 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 }