| /* |
| * Copyright (c) 2010, 2015, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| /* |
| * @test |
| * @bug 6570730 |
| * @summary com.sun.source.tree.ModifiersTree.getFlags() should return class type |
| * @modules jdk.compiler/com.sun.tools.javac.api |
| * jdk.compiler/com.sun.tools.javac.file |
| */ |
| |
| import java.io.*; |
| import java.util.*; |
| import javax.tools.*; |
| import com.sun.source.tree.*; |
| import com.sun.source.util.*; |
| import com.sun.tools.javac.api.*; |
| |
| public class ClassTreeTest { |
| public static void main(String... args) throws Exception { |
| new ClassTreeTest().run(); |
| } |
| |
| void run() throws Exception { |
| JavacTool tool = JavacTool.create(); |
| try (StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null)) { |
| List<String> opts = Collections.<String>emptyList(); |
| File testSrc = new File(System.getProperty("test.src")); |
| File thisFile = new File(testSrc, ClassTreeTest.class.getSimpleName() + ".java"); |
| Iterable<? extends JavaFileObject> fos = fm.getJavaFileObjects(thisFile); |
| JavacTask task = tool.getTask(null, fm, null, opts, null, fos); |
| for (CompilationUnitTree cu: task.parse()) { |
| check(cu, "CLASS", Tree.Kind.CLASS); |
| check(cu, "INTERFACE", Tree.Kind.INTERFACE); |
| check(cu, "ENUM", Tree.Kind.ENUM); |
| check(cu, "ANNOTATION_TYPE", Tree.Kind.ANNOTATION_TYPE); |
| } |
| |
| int expected = 4; |
| if (checks != expected) |
| error("Unexpected number of checks performed; expected: " + expected + ", found: " + checks); |
| |
| if (errors > 0) |
| throw new Exception(errors + " errors found"); |
| } |
| } |
| |
| void check(CompilationUnitTree cu, String name, Tree.Kind k) { |
| checks++; |
| |
| TreeScanner<ClassTree,String> s = new TreeScanner<ClassTree,String>() { |
| @Override |
| public ClassTree visitClass(ClassTree c, String name) { |
| if (c.getSimpleName().toString().equals(name)) |
| return c; |
| else |
| return super.visitClass(c, name); |
| } |
| |
| @Override |
| public ClassTree reduce(ClassTree t1, ClassTree t2) { |
| return (t1 != null ? t1 : t2); |
| } |
| }; |
| |
| ClassTree c = s.scan(cu, name); |
| if (c == null) |
| error("Can't find node named " + name); |
| else if (c.getKind() != k) |
| error("Unexpected kind for node named " + name + ": expected: " + k + ", found: " + c.getKind()); |
| } |
| |
| void error(String msg) { |
| System.err.println("Error: " + msg); |
| errors++; |
| } |
| |
| int checks; |
| int errors; |
| |
| class CLASS { } |
| interface INTERFACE { } |
| enum ENUM { } |
| @interface ANNOTATION_TYPE { } |
| } |