1 /*
   2  * Copyright 2006 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     6418666 6423973 6453386
  27  * @summary Test the NoTypes: VOID, PACKAGE, NONE
  28  * @author  Scott Seligman
  29  * @compile -g NoTypes.java
  30  * @compile -processor NoTypes -proc:only NoTypes.java
  31  */
  32 
  33 import java.util.Set;
  34 import javax.annotation.processing.*;
  35 import javax.lang.model.SourceVersion;
  36 import javax.lang.model.element.*;
  37 import javax.lang.model.type.*;
  38 import javax.lang.model.util.*;
  39 
  40 import static javax.lang.model.type.TypeKind.*;
  41 
  42 @SupportedAnnotationTypes("*")
  43 public class NoTypes extends AbstractProcessor {
  44 
  45     Elements elements;
  46     Types types;
  47 
  48     public void init(ProcessingEnvironment penv) {
  49         super.init(penv);
  50         elements = penv.getElementUtils();
  51         types =  penv.getTypeUtils();
  52     }
  53 
  54     public boolean process(Set<? extends TypeElement> annoTypes,
  55                            RoundEnvironment round) {
  56         if (!round.processingOver())
  57             doit(annoTypes, round);
  58         return true;
  59     }
  60 
  61     @Override
  62     public SourceVersion getSupportedSourceVersion() {
  63         return SourceVersion.latest();
  64     }
  65 
  66     private void doit(Set<? extends TypeElement> annoTypes,
  67                       RoundEnvironment round) {
  68 
  69         // The superclass of Object is NONE.
  70         TypeElement object = elements.getTypeElement("java.lang.Object");
  71         verifyKind(NONE, object.getSuperclass());
  72 
  73         // The enclosing type of a top-level class is NONE
  74         verifyKind(NONE, ((DeclaredType)object.asType()).getEnclosingType());
  75 
  76         // The superclass of an interface is NONE.
  77         TypeElement i = elements.getTypeElement("NoTypes.I");
  78         verifyKind(NONE, i.getSuperclass());
  79 
  80         // The type of a package is PACKAGE.
  81         Element pkg = i.getEnclosingElement().getEnclosingElement();
  82         verifyKind(PACKAGE, pkg.asType());
  83 
  84         // A package isn't enclosed.  Not yet, anyway.
  85         if (pkg.getEnclosingElement() != null)
  86             throw new AssertionError();
  87 
  88         verifyKind(VOID, types.getNoType(VOID));
  89         verifyKind(NONE, types.getNoType(NONE));
  90 
  91         // The return type of a constructor or void method is VOID.
  92         class Scanner extends ElementScanner6<Void, Void> {
  93             @Override
  94             public Void visitExecutable(ExecutableElement e, Void p) {
  95                 verifyKind(VOID, e.getReturnType());
  96                 ExecutableType t = (ExecutableType) e.asType();
  97                 verifyKind(VOID, t.getReturnType());
  98                 return null;
  99             }
 100         }
 101         TypeElement c = elements.getTypeElement("NoTypes.C");
 102         new Scanner().scan(c);
 103     }
 104 
 105     /**
 106      * Verify that a NoType instance is of a particular kind,
 107      * and that TypeKindVisitor6 properly dispatches on it.
 108      */
 109     private void verifyKind(TypeKind kind, TypeMirror type) {
 110         class Vis extends TypeKindVisitor6<TypeKind, Void> {
 111             @Override
 112             public TypeKind visitNoTypeAsVoid(NoType t, Void p) {
 113                 return VOID;
 114             }
 115             @Override
 116             public TypeKind visitNoTypeAsPackage(NoType t, Void p) {
 117                 return PACKAGE;
 118             }
 119             @Override
 120             public TypeKind visitNoTypeAsNone(NoType t, Void p) {
 121                 return NONE;
 122             }
 123         }
 124         if (kind != type.getKind() || kind != new Vis().visit(type))
 125             throw new AssertionError();
 126     }
 127 
 128 
 129     // Fodder for the tests
 130 
 131     interface I {
 132     }
 133 
 134     class C {
 135         C() {}
 136         void m() {}
 137     }
 138 }