test/tools/javac/processing/filer/TestPackageInfo.java

Tue, 24 Aug 2010 15:09:21 -0700

author
jjg
date
Tue, 24 Aug 2010 15:09:21 -0700
changeset 655
f3323b1c65ee
parent 554
9d9f26857129
child 699
d2aaaec153e8
permissions
-rw-r--r--

6929404: Filer.getResource(SOURCE_PATH, ...) does not work when -sourcepath contains >1 entry
Reviewed-by: darcy

duke@1 1 /*
ohair@554 2 * Copyright (c) 2006, Oracle and/or its affiliates. All rights reserved.
duke@1 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
duke@1 4 *
duke@1 5 * This code is free software; you can redistribute it and/or modify it
duke@1 6 * under the terms of the GNU General Public License version 2 only, as
duke@1 7 * published by the Free Software Foundation.
duke@1 8 *
duke@1 9 * This code is distributed in the hope that it will be useful, but WITHOUT
duke@1 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
duke@1 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
duke@1 12 * version 2 for more details (a copy is included in the LICENSE file that
duke@1 13 * accompanied this code).
duke@1 14 *
duke@1 15 * You should have received a copy of the GNU General Public License version
duke@1 16 * 2 along with this work; if not, write to the Free Software Foundation,
duke@1 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
duke@1 18 *
ohair@554 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@554 20 * or visit www.oracle.com if you need additional information or have any
ohair@554 21 * questions.
duke@1 22 */
duke@1 23
duke@1 24 /*
duke@1 25 * @test
duke@1 26 * @bug 6380018 6392177
duke@1 27 * @summary Test the ability to create and process package-info.java files
duke@1 28 * @author Joseph D. Darcy
duke@1 29 * @compile TestPackageInfo.java
duke@1 30 * @compile -processor TestPackageInfo -proc:only foo/bar/package-info.java TestPackageInfo.java
duke@1 31 */
duke@1 32
duke@1 33 import java.util.Set;
duke@1 34 import java.util.HashSet;
duke@1 35 import java.util.Map;
duke@1 36 import javax.annotation.processing.*;
duke@1 37 import javax.lang.model.SourceVersion;
duke@1 38 import static javax.lang.model.SourceVersion.*;
duke@1 39 import javax.lang.model.element.*;
duke@1 40 import javax.lang.model.util.*;
duke@1 41 import static javax.lang.model.util.ElementFilter.*;
duke@1 42 import static javax.tools.Diagnostic.Kind.*;
duke@1 43 import static javax.tools.StandardLocation.*;
duke@1 44
duke@1 45 import java.io.*;
duke@1 46
duke@1 47 /**
duke@1 48 * Test the ability to process annotations on package-info.java files:
duke@1 49 * 1) Visibility of package-info files from the command line
duke@1 50 * 2) Visibility of generated package-info.java source files
duke@1 51 */
duke@1 52 @SupportedAnnotationTypes("*")
duke@1 53 public class TestPackageInfo extends AbstractProcessor {
duke@1 54 private Elements eltUtils;
duke@1 55 private Messager messager;
duke@1 56 private Filer filer;
duke@1 57 private Map<String,String> options;
duke@1 58
duke@1 59 private int round = 0;
duke@1 60
duke@1 61 public boolean process(Set<? extends TypeElement> annotations,
duke@1 62 RoundEnvironment roundEnv) {
duke@1 63 round++;
duke@1 64
duke@1 65 // Verify annotations are as expected
duke@1 66 Set<TypeElement> expectedAnnotations = new HashSet<TypeElement>();
duke@1 67 if (round == 1)
duke@1 68 expectedAnnotations.add(eltUtils.
duke@1 69 getTypeElement("javax.annotation.processing.SupportedAnnotationTypes"));
duke@1 70 expectedAnnotations.add(eltUtils.
duke@1 71 getTypeElement("java.lang.SuppressWarnings"));
duke@1 72
duke@1 73 if (!roundEnv.processingOver()) {
duke@1 74 System.out.println("\nRound " + round);
duke@1 75 int rootElementSize = roundEnv.getRootElements().size();
duke@1 76
duke@1 77 for(Element elt : roundEnv.getRootElements()) {
duke@1 78 System.out.printf("%nElement %s\tkind: %s%n", elt.getSimpleName(), elt.getKind());
duke@1 79 eltUtils.printElements(new PrintWriter(System.out), elt);
duke@1 80 }
duke@1 81
duke@1 82 switch (round) {
duke@1 83 case 1:
duke@1 84 if (rootElementSize != 2)
duke@1 85 throw new RuntimeException("Unexpected root element size " + rootElementSize);
duke@1 86
duke@1 87 // Note that foo.bar.FuBar, an element of package
duke@1 88 // foo.bar, contains @Deprecated which should *not* be
duke@1 89 // included in the set of annotations to process
duke@1 90 if (!expectedAnnotations.equals(annotations)) {
duke@1 91 throw new RuntimeException("Unexpected annotations: " + annotations);
duke@1 92 }
duke@1 93
duke@1 94 try {
duke@1 95 try {
duke@1 96 filer.createClassFile("package-info");
duke@1 97 throw new RuntimeException("Created class file for \"package-info\".");
duke@1 98 } catch(FilerException fe) {}
duke@1 99
duke@1 100 PrintWriter pw = new PrintWriter(filer.createSourceFile("foo.package-info").openWriter());
duke@1 101 pw.println("@SuppressWarnings(\"\")");
duke@1 102 pw.println("package foo;");
duke@1 103 pw.close();
duke@1 104
duke@1 105 } catch(IOException ioe) {
duke@1 106 throw new RuntimeException(ioe);
duke@1 107 }
duke@1 108 break;
duke@1 109
duke@1 110 case 2:
duke@1 111 // Expect foo.package-info
duke@1 112
duke@1 113 Set<Element> expectedElement = new HashSet<Element>();
duke@1 114 expectedElement.add(eltUtils.getPackageElement("foo"));
duke@1 115 if (!expectedElement.equals(roundEnv.getRootElements()))
duke@1 116 throw new RuntimeException("Unexpected root element set " + roundEnv.getRootElements());
duke@1 117
duke@1 118 if (!expectedAnnotations.equals(annotations)) {
duke@1 119 throw new RuntimeException("Unexpected annotations: " + annotations);
duke@1 120 }
duke@1 121
duke@1 122 break;
duke@1 123
duke@1 124 default:
duke@1 125 throw new RuntimeException("Unexpected round number " + round);
duke@1 126 }
duke@1 127 }
duke@1 128 return false;
duke@1 129 }
duke@1 130
duke@1 131 public SourceVersion getSupportedSourceVersion() {
duke@1 132 return SourceVersion.latest();
duke@1 133 }
duke@1 134
duke@1 135 public void init(ProcessingEnvironment processingEnv) {
duke@1 136 super.init(processingEnv);
duke@1 137 eltUtils = processingEnv.getElementUtils();
duke@1 138 messager = processingEnv.getMessager();
duke@1 139 filer = processingEnv.getFiler();
duke@1 140 options = processingEnv.getOptions();
duke@1 141 }
duke@1 142 }

mercurial