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

test/tools/javac/processing/model/util/filter/TestIterables.java@2eb010b6cb22 (annotated)

test/tools/javac/processing/model/util/filter/TestIterables.java

Thu, 31 Aug 2017 15:17:03 +0800

author

aoqi

date

Thu, 31 Aug 2017 15:17:03 +0800

changeset 2525

2eb010b6cb22

parent 1466

b52a38d4536c

parent 0

959103a6100f

permissions

-rw-r--r--

merge

aoqi@0	1	/*
aoqi@0	2	* Copyright (c) 2006, 2010, 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 6406164
aoqi@0	27	* @summary Test that ElementFilter iterable methods behave properly.
aoqi@0	28	* @author Joseph D. Darcy
aoqi@0	29	* @library /tools/javac/lib
aoqi@0	30	* @build JavacTestingAbstractProcessor
aoqi@0	31	* @compile TestIterables.java
aoqi@0	32	* @compile -processor TestIterables -proc:only Foo1.java
aoqi@0	33	* @compile Foo1.java
aoqi@0	34	* @compile -processor TestIterables Foo1 TestIterables.java
aoqi@0	35	*/
aoqi@0	36
aoqi@0	37	import java.util.Set;
aoqi@0	38	import java.util.HashSet;
aoqi@0	39	import java.util.Arrays;
aoqi@0	40	import java.util.Iterator;
aoqi@0	41	import javax.annotation.processing.*;
aoqi@0	42	import javax.lang.model.SourceVersion;
aoqi@0	43	import static javax.lang.model.SourceVersion.*;
aoqi@0	44	import javax.lang.model.element.*;
aoqi@0	45	import javax.lang.model.util.*;
aoqi@0	46	import static javax.tools.Diagnostic.Kind.*;
aoqi@0	47
aoqi@0	48	import static javax.lang.model.util.ElementFilter.*;
aoqi@0	49
aoqi@0	50	/**
aoqi@0	51	* Verify that the {@code ElementFilter} methods taking iterables
aoqi@0	52	* behave correctly by comparing ExpectedElementCounts with actual
aoqi@0	53	* results.
aoqi@0	54	*/
aoqi@0	55	@SupportedAnnotationTypes("ExpectedElementCounts")
aoqi@0	56	@ExpectedElementCounts(methods=2)
aoqi@0	57	public class TestIterables extends JavacTestingAbstractProcessor {
aoqi@0	58	public boolean process(Set<? extends TypeElement> annotations,
aoqi@0	59	RoundEnvironment roundEnv) {
aoqi@0	60	if (!roundEnv.processingOver()) {
aoqi@0	61	boolean error = false;
aoqi@0	62	int topLevelCount = 0;
aoqi@0	63
aoqi@0	64	for (TypeElement type : typesIn(roundEnv.getRootElements())) {
aoqi@0	65	topLevelCount++;
aoqi@0	66	ExpectedElementCounts elementCounts = type.getAnnotation(ExpectedElementCounts.class);
aoqi@0	67	if (elementCounts == null)
aoqi@0	68	throw new RuntimeException("Missing ExpectedElementCounts annotation!");
aoqi@0	69
aoqi@0	70	Iterable<? extends Element> irritable = type.getEnclosedElements();
aoqi@0	71
aoqi@0	72	int constructorCount = getCount(constructorsIn(irritable));
aoqi@0	73	int fieldCount = getCount(fieldsIn(irritable));
aoqi@0	74	int methodCount = getCount(methodsIn(irritable));
aoqi@0	75	int typeCount = getCount(typesIn(irritable));
aoqi@0	76
aoqi@0	77	System.out.println("\nType: " + type.getQualifiedName());
aoqi@0	78	System.out.format("Element Actual\tExpected%n");
aoqi@0	79	System.out.format("constructors %d\t\t%d%n", constructorCount, elementCounts.constructors());
aoqi@0	80	System.out.format("fields %d\t\t%d%n", fieldCount, elementCounts.fields());
aoqi@0	81	System.out.format("methods %d\t\t%d%n", methodCount, elementCounts.methods());
aoqi@0	82	System.out.format("types %d\t\t%d%n", typeCount, elementCounts.types());
aoqi@0	83
aoqi@0	84	if ((constructorCount != elementCounts.constructors()) ||
aoqi@0	85	(fieldCount != elementCounts.fields()) ||
aoqi@0	86	(methodCount != elementCounts.methods()) ||
aoqi@0	87	(typeCount != elementCounts.types()) )
aoqi@0	88	error = true;
aoqi@0	89	}
aoqi@0	90
aoqi@0	91	if (topLevelCount == 0)
aoqi@0	92	error = true;
aoqi@0	93
aoqi@0	94	if (error)
aoqi@0	95	throw new RuntimeException("A count mismatch has occured.");
aoqi@0	96	}
aoqi@0	97
aoqi@0	98	return true;
aoqi@0	99	}
aoqi@0	100
aoqi@0	101	private int getCount(Iterable<? extends Element> iter) {
aoqi@0	102	int count1 = 0;
aoqi@0	103	int count2 = 0;
aoqi@0	104
aoqi@0	105	Iterator<? extends Element> iterator = iter.iterator();
aoqi@0	106	while (iterator.hasNext() ) {
aoqi@0	107	count1++;
aoqi@0	108	iterator.next();
aoqi@0	109	}
aoqi@0	110
aoqi@0	111	iterator = iter.iterator();
aoqi@0	112	while (iterator.hasNext() ) {
aoqi@0	113	count2++;
aoqi@0	114	iterator.next();
aoqi@0	115	}
aoqi@0	116
aoqi@0	117	if (count1 != count2)
aoqi@0	118	throw new RuntimeException("Inconsistent counts from iterator.");
aoqi@0	119
aoqi@0	120	return count1;
aoqi@0	121	}
aoqi@0	122	}