0N/A/*
797N/A * Copyright (c) 2006, 2010, Oracle and/or its affiliates. All rights reserved.
0N/A * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
0N/A *
0N/A * This code is free software; you can redistribute it and/or modify it
0N/A * under the terms of the GNU General Public License version 2 only, as
0N/A * published by the Free Software Foundation.
0N/A *
0N/A * This code is distributed in the hope that it will be useful, but WITHOUT
0N/A * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
0N/A * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
0N/A * version 2 for more details (a copy is included in the LICENSE file that
0N/A * accompanied this code).
0N/A *
0N/A * You should have received a copy of the GNU General Public License version
0N/A * 2 along with this work; if not, write to the Free Software Foundation,
0N/A * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
0N/A *
553N/A * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
553N/A * or visit www.oracle.com if you need additional information or have any
553N/A * questions.
0N/A */
0N/A
0N/A/*
0N/A * @test
0N/A * @bug 6392782
0N/A * @summary TreeScanner.visitImport returns null, not result of nested scan
0N/A */
0N/A
0N/Aimport java.io.*;
0N/Aimport java.util.*;
0N/Aimport javax.tools.*;
0N/Aimport com.sun.source.tree.*;
0N/Aimport com.sun.source.util.*;
0N/Aimport com.sun.tools.javac.api.*;
0N/A
0N/Apublic class T6392782 {
0N/A public static void main(String... args) throws IOException {
0N/A String testSrc = System.getProperty("test.src", ".");
0N/A JavacTool tool = JavacTool.create();
0N/A StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);
0N/A Iterable<? extends JavaFileObject> files =
0N/A fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrc, T6392782.class.getName()+".java")));
0N/A JavacTask task = tool.getTask(null, fm, null, null, null, files);
0N/A Iterable<? extends Tree> trees = task.parse();
0N/A TreeScanner<Integer,Void> scanner = new MyScanner();
0N/A check(scanner, 6, scanner.scan(trees, null));
0N/A
0N/A CountNodes nodeCounter = new CountNodes();
678N/A // 359 nodes with the regular parser; 360 nodes with EndPosParser
678N/A // We automatically switch to EndPosParser when calling JavacTask.parse()
678N/A check(nodeCounter, 360, nodeCounter.scan(trees, null));
0N/A
0N/A CountIdentifiers idCounter = new CountIdentifiers();
678N/A check(idCounter, 107, idCounter.scan(trees, null));
0N/A }
0N/A
0N/A private static void check(TreeScanner<?,?> scanner, int expect, int found) {
0N/A if (found != expect)
0N/A throw new AssertionError(scanner.getClass().getName() + ": expected: " + expect + " found: " + found);
0N/A }
0N/A
0N/A static class MyScanner extends TreeScanner<Integer,Void> {
0N/A @Override
0N/A public Integer visitImport(ImportTree tree, Void ignore) {
0N/A //System.err.println(tree);
0N/A return 1;
0N/A }
0N/A
0N/A @Override
0N/A public Integer reduce(Integer i1, Integer i2) {
0N/A return (i1 == null ? 0 : i1) + (i2 == null ? 0 : i2);
0N/A }
0N/A }
0N/A
0N/A static class CountNodes extends TreeScanner<Integer,Void> {
0N/A @Override
0N/A public Integer scan(Tree node, Void p) {
678N/A if (node == null)
678N/A return 0;
0N/A Integer n = super.scan(node, p);
0N/A return (n == null ? 0 : n) + 1;
0N/A }
0N/A @Override
0N/A public Integer reduce(Integer r1, Integer r2) {
0N/A return (r1 == null ? 0 : r1) + (r2 == null ? 0 : r2);
0N/A }
0N/A }
0N/A
0N/A // example from TreeScanner javadoc
0N/A static class CountIdentifiers extends TreeScanner<Integer,Void> {
0N/A @Override
0N/A public Integer visitIdentifier(IdentifierTree node, Void p) {
0N/A return 1;
0N/A }
0N/A @Override
0N/A public Integer reduce(Integer r1, Integer r2) {
0N/A return (r1 == null ? 0 : r1) + (r2 == null ? 0 : r2);
0N/A }
0N/A }
0N/A}