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