1 /*
   2  * Copyright (c) 2006, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25  * @test
  26  * @bug 6453386
  27  * @summary Test Elements.getPackageOf
  28  * @author  Joseph D. Darcy
  29  * @build TestGetPackageOf
  30  * @compile -processor TestGetPackageOf -proc:only TestGetPackageOf.java
  31  */
  32 
  33 import java.util.Set;
  34 import javax.annotation.processing.*;
  35 import javax.lang.model.SourceVersion;
  36 import static javax.lang.model.SourceVersion.*;
  37 import javax.lang.model.element.*;
  38 import javax.lang.model.util.*;
  39 import static javax.lang.model.util.ElementFilter.*;
  40 import static javax.tools.Diagnostic.Kind.*;
  41 import static javax.tools.StandardLocation.*;
  42 
  43 /**
  44  * Test basic workings of Elements.getPackageOf
  45  */
  46 @SupportedAnnotationTypes("*")
  47 public class TestGetPackageOf extends AbstractProcessor {
  48     private Elements eltUtils;
  49 
  50     /**
  51      * Check expected behavior on classes and packages.
  52      */
  53     public boolean process(Set<? extends TypeElement> annotations,
  54                            RoundEnvironment roundEnv) {
  55         if (!roundEnv.processingOver()) {
  56             TypeElement    stringElt   = eltUtils.getTypeElement("java.lang.String");
  57             PackageElement javaLangPkg = eltUtils.getPackageElement("java.lang");
  58             PackageElement unnamedPkg  = eltUtils.getPackageElement("");
  59             PackageElement pkg = null;
  60 
  61             if (!javaLangPkg.equals(pkg=eltUtils.getPackageOf(stringElt) ) )
  62                 throw new RuntimeException("Unexpected package for String: " + pkg);
  63 
  64             if (!javaLangPkg.equals(pkg=eltUtils.getPackageOf(javaLangPkg) ) )
  65                 throw new RuntimeException("Unexpected package for java.lang: " + pkg);
  66 
  67             if (!unnamedPkg.equals(pkg=eltUtils.getPackageOf(unnamedPkg) ) )
  68                 throw new RuntimeException("Unexpected package for unnamed pkg: " + pkg);
  69         }
  70         return true;
  71     }
  72 
  73     public SourceVersion getSupportedSourceVersion() {
  74         return SourceVersion.latest();
  75     }
  76 
  77     public void init(ProcessingEnvironment processingEnv) {
  78         super.init(processingEnv);
  79         eltUtils = processingEnv.getElementUtils();
  80     }
  81 }