T4994049.java revision 3294:9adfb22ff08f
1193326Sed/* 2193326Sed * Copyright (c) 2005, 2016, Oracle and/or its affiliates. All rights reserved. 3353358Sdim * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4353358Sdim * 5353358Sdim * This code is free software; you can redistribute it and/or modify it 6193326Sed * under the terms of the GNU General Public License version 2 only, as 7193326Sed * published by the Free Software Foundation. 8193326Sed * 9198092Srdivacky * This code is distributed in the hope that it will be useful, but WITHOUT 10193326Sed * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11193326Sed * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12193326Sed * version 2 for more details (a copy is included in the LICENSE file that 13193326Sed * accompanied this code). 14249423Sdim * 15206084Srdivacky * You should have received a copy of the GNU General Public License version 16249423Sdim * 2 along with this work; if not, write to the Free Software Foundation, 17206084Srdivacky * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18234353Sdim * 19193326Sed * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20249423Sdim * or visit www.oracle.com if you need additional information or have any 21193326Sed * questions. 22193326Sed */ 23193326Sed 24261991Sdim/** 25249423Sdim * @test 26249423Sdim * @bug 4994049 27249423Sdim * @summary Unit test for SourcePosition.column with respect to tab expansion 28193326Sed * @author Peter von der Ah\u00e9 29193326Sed * @modules jdk.javadoc/jdk.javadoc.internal.tool 30193326Sed * @run main T4994049 FileWithTabs.java 31251662Sdim */ 32251662Sdim 33251662Sdimimport java.io.*; 34251662Sdimimport java.util.ArrayList; 35224145Sdimimport java.util.Collections; 36193326Sedimport java.util.List; 37193326Sedimport java.util.Locale; 38193326Sedimport java.util.Set; 39276479Sdim 40202379Srdivackyimport javax.lang.model.SourceVersion; 41202379Srdivackyimport javax.lang.model.element.ElementKind; 42202379Srdivackyimport javax.lang.model.element.ExecutableElement; 43202379Srdivackyimport javax.lang.model.element.TypeElement; 44193326Sed 45193326Sedimport com.sun.source.tree.CompilationUnitTree; 46323112Sdimimport com.sun.source.tree.LineMap; 47323112Sdimimport com.sun.source.util.DocTrees; 48323112Sdimimport com.sun.source.util.SourcePositions; 49323112Sdimimport com.sun.source.util.TreePath; 50224145Sdimimport jdk.javadoc.doclet.Doclet; 51224145Sdimimport jdk.javadoc.doclet.Reporter; 52226633Sdimimport jdk.javadoc.doclet.DocletEnvironment; 53234353Sdim 54221345Sdimimport static jdk.javadoc.internal.tool.Main.execute; 55221345Sdim 56341825Sdimpublic class T4994049 implements Doclet { 57221345Sdim 58221345Sdim public boolean run(DocletEnvironment root) { 59221345Sdim DocTrees trees = root.getDocTrees(); 60221345Sdim 61221345Sdim SourcePositions sourcePositions = trees.getSourcePositions(); 62221345Sdim for (TypeElement klass : root.getIncludedClasses()) { 63249423Sdim for (ExecutableElement method : getMethods(klass)) { 64221345Sdim if (method.getSimpleName().toString().equals("tabbedMethod")) { 65221345Sdim TreePath path = trees.getPath(method); 66221345Sdim CompilationUnitTree cu = path.getCompilationUnit(); 67221345Sdim long pos = sourcePositions.getStartPosition(cu, path.getLeaf()); 68221345Sdim LineMap lineMap = cu.getLineMap(); 69221345Sdim long columnNumber = lineMap.getColumnNumber(pos); 70249423Sdim if (columnNumber == 9) { 71221345Sdim System.out.println(columnNumber + ": OK!"); 72221345Sdim return true; 73221345Sdim } else { 74221345Sdim System.err.println(columnNumber + ": wrong tab expansion"); 75210299Sed return false; 76221345Sdim } 77221345Sdim } 78221345Sdim } 79224145Sdim } 80221345Sdim return false; 81221345Sdim } 82193326Sed 83193326Sed public static void main(String... args) throws Exception { 84193326Sed File testSrc = new File(System.getProperty("test.src")); 85193326Sed File tmpSrc = new File("tmpSrc"); 86280031Sdim initTabs(testSrc, tmpSrc); 87224145Sdim 88198092Srdivacky 89193326Sed for (String file : args) { 90203955Srdivacky File source = new File(tmpSrc, file); 91193326Sed String[] array = { 92198092Srdivacky "-docletpath", System.getProperty("test.classes", "."), 93193326Sed "-doclet", "T4994049", 94198092Srdivacky source.getPath() 95198092Srdivacky }; 96224145Sdim int rc = execute(array); 97198092Srdivacky if (rc != 0) 98224145Sdim throw new Error("Unexpected return code from javadoc: " + rc); 99224145Sdim } 100224145Sdim } 101224145Sdim 102341825Sdim static void initTabs(File from, File to) throws IOException { 103224145Sdim for (File f: from.listFiles()) { 104224145Sdim File t = new File(to, f.getName()); 105193326Sed if (f.isDirectory()) { 106193326Sed initTabs(f, t); 107224145Sdim } else if (f.getName().endsWith(".java")) { 108224145Sdim write(t, read(f).replace("\\t", "\t")); 109224145Sdim } 110224145Sdim } 111224145Sdim } 112224145Sdim 113224145Sdim static String read(File f) throws IOException { 114224145Sdim StringBuilder sb = new StringBuilder(); 115341825Sdim try (BufferedReader in = new BufferedReader(new FileReader(f))) { 116224145Sdim String line; 117224145Sdim while ((line = in.readLine()) != null) { 118224145Sdim sb.append(line); 119224145Sdim sb.append("\n"); 120280031Sdim } 121280031Sdim } 122280031Sdim return sb.toString(); 123224145Sdim } 124341825Sdim 125224145Sdim static void write(File f, String s) throws IOException { 126224145Sdim f.getParentFile().mkdirs(); 127341825Sdim try (Writer out = new FileWriter(f)) { 128224145Sdim out.write(s); 129224145Sdim } 130224145Sdim } 131341825Sdim 132224145Sdim List<ExecutableElement> getMethods(TypeElement klass) { 133224145Sdim List<ExecutableElement> methods = new ArrayList<>(); 134224145Sdim klass.getEnclosedElements() 135224145Sdim .stream() 136224145Sdim .filter((e) -> (e.getKind() == ElementKind.METHOD)) 137276479Sdim .forEach((e) -> { 138360784Sdim methods.add((ExecutableElement) e); 139360784Sdim }); 140224145Sdim return methods; 141224145Sdim } 142341825Sdim 143224145Sdim @Override 144224145Sdim public String getName() { 145276479Sdim return "Test"; 146224145Sdim } 147224145Sdim 148341825Sdim @Override 149224145Sdim public Set<Option> getSupportedOptions() { 150224145Sdim return Collections.emptySet(); 151193326Sed } 152193326Sed 153224145Sdim @Override 154224145Sdim public SourceVersion getSupportedSourceVersion() { 155224145Sdim return SourceVersion.latest(); 156224145Sdim } 157224145Sdim 158224145Sdim @Override 159288943Sdim public void init(Locale locale, Reporter reporter) { 160288943Sdim return; 161288943Sdim } 162288943Sdim} 163224145Sdim