diff -r a344cff45d06 -r d3d578d22cc7 langtools/test/tools/javac/tree/ClassTreeTest.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/langtools/test/tools/javac/tree/ClassTreeTest.java Fri Aug 27 17:21:17 2010 -0700 @@ -0,0 +1,102 @@ +/* + * Copyright (c) 2010, 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 + */ + +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(); + StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null); + List opts = Collections.emptyList(); + File testSrc = new File(System.getProperty("test.src")); + File thisFile = new File(testSrc, ClassTreeTest.class.getSimpleName() + ".java"); + Iterable 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 s = new TreeScanner() { + @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 { } +}