diff -r 000000000000 -r 9a66ca7c79fa test/tools/javac/processing/environment/round/TestElementsAnnotatedWith.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/test/tools/javac/processing/environment/round/TestElementsAnnotatedWith.java Sat Dec 01 00:00:00 2007 +0000 @@ -0,0 +1,168 @@ +/* + * Copyright 2006-2007 Sun Microsystems, Inc. All Rights Reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. + * + * This code 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 General Public License + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara, + * CA 95054 USA or visit www.sun.com if you need additional information or + * have any questions. + */ + +/* + * @test + * @bug 6397298 6400986 6425592 6449798 6453386 6508401 + * @summary Tests that getElementsAnnotatedWith works properly. + * @author Joseph D. Darcy + * @compile TestElementsAnnotatedWith.java + * @compile InheritedAnnotation.java + * @compile -processor TestElementsAnnotatedWith -proc:only SurfaceAnnotations.java + * @compile -processor TestElementsAnnotatedWith -proc:only BuriedAnnotations.java + * @compile -processor TestElementsAnnotatedWith -proc:only Part1.java Part2.java + * @compile -processor TestElementsAnnotatedWith -proc:only TestElementsAnnotatedWith.java + * @compile -processor TestElementsAnnotatedWith -proc:only C2.java + */ + +import java.lang.annotation.Annotation; +import java.util.Collections; +import java.util.Set; +import java.util.HashSet; +import java.util.Arrays; +import javax.annotation.processing.*; +import javax.lang.model.SourceVersion; +import javax.lang.model.element.*; +import javax.lang.model.util.*; +import static javax.lang.model.util.ElementFilter.*; + +/** + * This processor verifies that the information returned by + * getElementsAnnotatedWith is consistent with the expected results + * stored in an AnnotatedElementInfo annotation. + */ +@SupportedAnnotationTypes("*") +@AnnotatedElementInfo(annotationName="java.lang.SuppressWarnings", expectedSize=0, names={}) +public class TestElementsAnnotatedWith extends AbstractProcessor { + + public boolean process(Set annotations, + RoundEnvironment roundEnvironment) { + Elements elementUtils = processingEnv.getElementUtils(); + + TypeElement annotatedElementInfoElement = + elementUtils.getTypeElement("AnnotatedElementInfo"); + Set resultsMeta = Collections.emptySet(); + Set resultsBase = Collections.emptySet(); + + if (!roundEnvironment.processingOver()) { + testNonAnnotations(roundEnvironment); + + // Verify AnnotatedElementInfo is present on the first + // specified type. + + TypeElement firstType = typesIn(roundEnvironment.getRootElements()).iterator().next(); + + AnnotatedElementInfo annotatedElementInfo = firstType.getAnnotation(AnnotatedElementInfo.class); + + boolean failed = false; + + if (annotatedElementInfo == null) + throw new IllegalArgumentException("Missing AnnotatedElementInfo annotation on " + + firstType); + else { + // Verify that the annotation information is as + // expected. + + Set expectedNames = new HashSet(Arrays.asList(annotatedElementInfo.names())); + + resultsMeta = + roundEnvironment. + getElementsAnnotatedWith(elementUtils. + getTypeElement(annotatedElementInfo. + annotationName())) ; + + System.err.println("Results: " + resultsMeta); + + if (resultsMeta.size() != annotatedElementInfo.expectedSize()) { + failed = true; + System.err.printf("Bad number of elements; expected %d, got %d%n", + annotatedElementInfo.expectedSize(), resultsMeta.size()); + } else { + for(Element element : resultsMeta) { + String simpleName = element.getSimpleName().toString(); + if (!expectedNames.contains(simpleName) ) { + failed = true; + System.err.println("Name ``" + simpleName + "'' not expected."); + } + } + } + } + + resultsBase = computeResultsBase(roundEnvironment, annotatedElementInfo.annotationName()); + + if (!resultsMeta.equals(resultsBase)) { + failed = true; + System.err.println("Base and Meta sets unequal;\n meta: " + resultsMeta + + "\nbase: " + resultsBase); + } + + if (failed) { + System.err.println("AnnotatedElementInfo: " + annotatedElementInfo); + throw new RuntimeException(); + } + } else { + // If processing is over without an error, the specified + // elements should be empty so an empty set should be returned. + resultsMeta = roundEnvironment.getElementsAnnotatedWith(annotatedElementInfoElement); + resultsBase = roundEnvironment.getElementsAnnotatedWith(AnnotatedElementInfo.class); + if (!resultsMeta.isEmpty()) + throw new RuntimeException("Nonempty resultsMeta: " + resultsMeta); + if (!resultsBase.isEmpty()) + throw new RuntimeException("Nonempty resultsBase: " + resultsBase); + + } + return true; + } + + private Set computeResultsBase(RoundEnvironment roundEnvironment, String name) { + try { + return roundEnvironment. + getElementsAnnotatedWith(Class.forName(name).asSubclass(Annotation.class)); + } catch(ClassNotFoundException cnfe) { + throw new RuntimeException(cnfe); + } + } + + /** + * Verify non-annotation types result in + * IllegalArgumentExceptions. + */ + private void testNonAnnotations(RoundEnvironment roundEnvironment) { + try { + Set elements = roundEnvironment.getElementsAnnotatedWith((Class)Object.class ); + throw new RuntimeException("Illegal argument exception not thrown"); + } catch(IllegalArgumentException iae) {} + + try { + Set elements = roundEnvironment.getElementsAnnotatedWith(processingEnv. + getElementUtils(). + getTypeElement("java.lang.Object") ); + throw new RuntimeException("Illegal argument exception not thrown"); + } catch(IllegalArgumentException iae) {} + } + + @Override + public SourceVersion getSupportedSourceVersion() { + return SourceVersion.latest(); + } +}