• file
• revisions
• annotate
• diff
• comparison
• raw

test/tools/javac/api/TestTreePath.java@959103a6100f (annotated)

test/tools/javac/api/TestTreePath.java

Wed, 27 Apr 2016 01:34:52 +0800

author

aoqi

date

Wed, 27 Apr 2016 01:34:52 +0800

changeset 0

959103a6100f

child 2525

2eb010b6cb22

permissions

-rw-r--r--

Initial load
http://hg.openjdk.java.net/jdk8u/jdk8u/langtools/
changeset: 2573:53ca196be1ae
tag: jdk8u25-b17

aoqi@0	1	/*
aoqi@0	2	* Copyright (c) 2006, Oracle and/or its affiliates. All rights reserved.
aoqi@0	3	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0	4	*
aoqi@0	5	* This code is free software; you can redistribute it and/or modify it
aoqi@0	6	* under the terms of the GNU General Public License version 2 only, as
aoqi@0	7	* published by the Free Software Foundation.
aoqi@0	8	*
aoqi@0	9	* This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0	10	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0	11	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
aoqi@0	12	* version 2 for more details (a copy is included in the LICENSE file that
aoqi@0	13	* accompanied this code).
aoqi@0	14	*
aoqi@0	15	* You should have received a copy of the GNU General Public License version
aoqi@0	16	* 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0	17	* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0	18	*
aoqi@0	19	* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0	20	* or visit www.oracle.com if you need additional information or have any
aoqi@0	21	* questions.
aoqi@0	22	*/
aoqi@0	23
aoqi@0	24	/*
aoqi@0	25	* @test
aoqi@0	26	* @bug 6473148
aoqi@0	27	* @summary TreePath.iterator() throws NPE
aoqi@0	28	*/
aoqi@0	29	import java.io.*;
aoqi@0	30	import java.util.Arrays;
aoqi@0	31	import java.util.Iterator;
aoqi@0	32	import java.util.Set;
aoqi@0	33
aoqi@0	34	import javax.annotation.processing.*;
aoqi@0	35	import javax.lang.model.SourceVersion;
aoqi@0	36	import javax.lang.model.element.Element;
aoqi@0	37	import javax.lang.model.element.TypeElement;
aoqi@0	38	import javax.lang.model.util.ElementFilter;
aoqi@0	39	import javax.tools.JavaCompiler;
aoqi@0	40	import javax.tools.JavaFileObject;
aoqi@0	41	import javax.tools.StandardJavaFileManager;
aoqi@0	42	import javax.tools.ToolProvider;
aoqi@0	43
aoqi@0	44	import com.sun.source.tree.Tree;
aoqi@0	45	import com.sun.source.util.*;
aoqi@0	46
aoqi@0	47	@SupportedAnnotationTypes("*")
aoqi@0	48	public class TestTreePath extends AbstractProcessor {
aoqi@0	49
aoqi@0	50	@Override
aoqi@0	51	public boolean process(Set<? extends TypeElement> annotations,
aoqi@0	52	RoundEnvironment roundEnv) {
aoqi@0	53	final Trees trees = Trees.instance(this.processingEnv);
aoqi@0	54	for (Element element : ElementFilter.typesIn(roundEnv.getRootElements())) {
aoqi@0	55	checkTreePath(trees, element, 2);
aoqi@0	56	for (Element member : element.getEnclosedElements())
aoqi@0	57	checkTreePath(trees, member, 3);
aoqi@0	58	}
aoqi@0	59	return true;
aoqi@0	60	}
aoqi@0	61
aoqi@0	62	private void checkTreePath(Trees trees, Element element, int expectedLength) {
aoqi@0	63	TreePath path = trees.getPath(element);
aoqi@0	64	assert path != null;
aoqi@0	65
aoqi@0	66	int enhancedLength = 0;
aoqi@0	67	for (Tree tree : path)
aoqi@0	68	++enhancedLength;
aoqi@0	69
aoqi@0	70	if (enhancedLength != expectedLength)
aoqi@0	71	throw new RuntimeException("found path length is wrong");
aoqi@0	72
aoqi@0	73	int normalLoopLength = 0;
aoqi@0	74	Iterator<Tree> iter = path.iterator();
aoqi@0	75	while (iter.hasNext()) {
aoqi@0	76	iter.next();
aoqi@0	77	++normalLoopLength;
aoqi@0	78	}
aoqi@0	79	if (normalLoopLength != expectedLength)
aoqi@0	80	throw new RuntimeException("found path length is wrong");
aoqi@0	81
aoqi@0	82	TreePath curr = path;
aoqi@0	83	// using getParent
aoqi@0	84	int whileLoopLength = 0;
aoqi@0	85	while (curr != null) {
aoqi@0	86	++whileLoopLength;
aoqi@0	87	curr = curr.getParentPath();
aoqi@0	88	}
aoqi@0	89	if (whileLoopLength != expectedLength)
aoqi@0	90	throw new RuntimeException("found path length is wrong");
aoqi@0	91	}
aoqi@0	92
aoqi@0	93	@Override
aoqi@0	94	public SourceVersion getSupportedSourceVersion() {
aoqi@0	95	return SourceVersion.latest();
aoqi@0	96	}
aoqi@0	97
aoqi@0	98	File writeTestFile() throws IOException {
aoqi@0	99	File f = new File("Test.java");
aoqi@0	100	PrintWriter out = new PrintWriter(new FileWriter(f));
aoqi@0	101	out.println("class Test { void method() { } }");
aoqi@0	102	out.close();
aoqi@0	103	return f;
aoqi@0	104	}
aoqi@0	105
aoqi@0	106	public void run() throws IOException {
aoqi@0	107	JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
aoqi@0	108	StandardJavaFileManager fileManager
aoqi@0	109	= compiler.getStandardFileManager(null, null, null);
aoqi@0	110	Iterable<? extends JavaFileObject> tests
aoqi@0	111	= fileManager.getJavaFileObjects(writeTestFile());
aoqi@0	112
aoqi@0	113	JavaCompiler.CompilationTask task =
aoqi@0	114	ToolProvider.getSystemJavaCompiler().getTask(
aoqi@0	115	null, null, null,
aoqi@0	116	Arrays.asList("-processor", this.getClass().getName()), null,
aoqi@0	117	tests);
aoqi@0	118	task.call();
aoqi@0	119	}
aoqi@0	120
aoqi@0	121	public static void main(String[] args) throws IOException {
aoqi@0	122	new TestTreePath().run();
aoqi@0	123	}
aoqi@0	124	}