duke@1: /* ohair@554: * Copyright (c) 2006, Oracle and/or its affiliates. All rights reserved. duke@1: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. duke@1: * duke@1: * This code is free software; you can redistribute it and/or modify it duke@1: * under the terms of the GNU General Public License version 2 only, as duke@1: * published by the Free Software Foundation. duke@1: * duke@1: * This code is distributed in the hope that it will be useful, but WITHOUT duke@1: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or duke@1: * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License duke@1: * version 2 for more details (a copy is included in the LICENSE file that duke@1: * accompanied this code). duke@1: * duke@1: * You should have received a copy of the GNU General Public License version duke@1: * 2 along with this work; if not, write to the Free Software Foundation, duke@1: * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. duke@1: * ohair@554: * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA ohair@554: * or visit www.oracle.com if you need additional information or have any ohair@554: * questions. duke@1: */ duke@1: duke@1: /* duke@1: * @test duke@1: * @bug 6380018 6392177 duke@1: * @summary Test the ability to create and process package-info.java files duke@1: * @author Joseph D. Darcy duke@1: * @compile TestPackageInfo.java duke@1: * @compile -processor TestPackageInfo -proc:only foo/bar/package-info.java TestPackageInfo.java duke@1: */ duke@1: duke@1: import java.util.Set; duke@1: import java.util.HashSet; duke@1: import java.util.Map; duke@1: import javax.annotation.processing.*; duke@1: import javax.lang.model.SourceVersion; duke@1: import static javax.lang.model.SourceVersion.*; duke@1: import javax.lang.model.element.*; duke@1: import javax.lang.model.util.*; duke@1: import static javax.lang.model.util.ElementFilter.*; duke@1: import static javax.tools.Diagnostic.Kind.*; duke@1: import static javax.tools.StandardLocation.*; duke@1: duke@1: import java.io.*; duke@1: duke@1: /** duke@1: * Test the ability to process annotations on package-info.java files: duke@1: * 1) Visibility of package-info files from the command line duke@1: * 2) Visibility of generated package-info.java source files duke@1: */ duke@1: @SupportedAnnotationTypes("*") duke@1: public class TestPackageInfo extends AbstractProcessor { duke@1: private Elements eltUtils; duke@1: private Messager messager; duke@1: private Filer filer; duke@1: private Map options; duke@1: duke@1: private int round = 0; duke@1: duke@1: public boolean process(Set annotations, duke@1: RoundEnvironment roundEnv) { duke@1: round++; duke@1: duke@1: // Verify annotations are as expected duke@1: Set expectedAnnotations = new HashSet(); duke@1: if (round == 1) duke@1: expectedAnnotations.add(eltUtils. duke@1: getTypeElement("javax.annotation.processing.SupportedAnnotationTypes")); duke@1: expectedAnnotations.add(eltUtils. duke@1: getTypeElement("java.lang.SuppressWarnings")); duke@1: duke@1: if (!roundEnv.processingOver()) { duke@1: System.out.println("\nRound " + round); duke@1: int rootElementSize = roundEnv.getRootElements().size(); duke@1: duke@1: for(Element elt : roundEnv.getRootElements()) { duke@1: System.out.printf("%nElement %s\tkind: %s%n", elt.getSimpleName(), elt.getKind()); duke@1: eltUtils.printElements(new PrintWriter(System.out), elt); duke@1: } duke@1: duke@1: switch (round) { duke@1: case 1: duke@1: if (rootElementSize != 2) duke@1: throw new RuntimeException("Unexpected root element size " + rootElementSize); duke@1: duke@1: // Note that foo.bar.FuBar, an element of package duke@1: // foo.bar, contains @Deprecated which should *not* be duke@1: // included in the set of annotations to process duke@1: if (!expectedAnnotations.equals(annotations)) { duke@1: throw new RuntimeException("Unexpected annotations: " + annotations); duke@1: } duke@1: duke@1: try { duke@1: try { duke@1: filer.createClassFile("package-info"); duke@1: throw new RuntimeException("Created class file for \"package-info\"."); duke@1: } catch(FilerException fe) {} duke@1: duke@1: PrintWriter pw = new PrintWriter(filer.createSourceFile("foo.package-info").openWriter()); duke@1: pw.println("@SuppressWarnings(\"\")"); duke@1: pw.println("package foo;"); duke@1: pw.close(); duke@1: duke@1: } catch(IOException ioe) { duke@1: throw new RuntimeException(ioe); duke@1: } duke@1: break; duke@1: duke@1: case 2: duke@1: // Expect foo.package-info duke@1: duke@1: Set expectedElement = new HashSet(); duke@1: expectedElement.add(eltUtils.getPackageElement("foo")); duke@1: if (!expectedElement.equals(roundEnv.getRootElements())) duke@1: throw new RuntimeException("Unexpected root element set " + roundEnv.getRootElements()); duke@1: duke@1: if (!expectedAnnotations.equals(annotations)) { duke@1: throw new RuntimeException("Unexpected annotations: " + annotations); duke@1: } duke@1: duke@1: break; duke@1: duke@1: default: duke@1: throw new RuntimeException("Unexpected round number " + round); duke@1: } duke@1: } duke@1: return false; duke@1: } duke@1: duke@1: public SourceVersion getSupportedSourceVersion() { duke@1: return SourceVersion.latest(); duke@1: } duke@1: duke@1: public void init(ProcessingEnvironment processingEnv) { duke@1: super.init(processingEnv); duke@1: eltUtils = processingEnv.getElementUtils(); duke@1: messager = processingEnv.getMessager(); duke@1: filer = processingEnv.getFiler(); duke@1: options = processingEnv.getOptions(); duke@1: } duke@1: }