1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/test/tools/javac/6902720/Test.java Thu Nov 19 11:49:27 2009 -0800 1.3 @@ -0,0 +1,91 @@ 1.4 +/* 1.5 + * Copyright 2009 Sun Microsystems, Inc. All Rights Reserved. 1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 1.7 + * 1.8 + * This code is free software; you can redistribute it and/or modify it 1.9 + * under the terms of the GNU General Public License version 2 only, as 1.10 + * published by the Free Software Foundation. 1.11 + * 1.12 + * This code is distributed in the hope that it will be useful, but WITHOUT 1.13 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 1.14 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 1.15 + * version 2 for more details (a copy is included in the LICENSE file that 1.16 + * accompanied this code). 1.17 + * 1.18 + * You should have received a copy of the GNU General Public License version 1.19 + * 2 along with this work; if not, write to the Free Software Foundation, 1.20 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 1.21 + * 1.22 + * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara, 1.23 + * CA 95054 USA or visit www.sun.com if you need additional information or 1.24 + * have any questions. 1.25 + */ 1.26 + 1.27 +import java.io.*; 1.28 +import java.net.*; 1.29 +import javax.tools.*; 1.30 +import java.util.*; 1.31 + 1.32 +import com.sun.source.tree.CompilationUnitTree; 1.33 +import com.sun.source.util.JavacTask; 1.34 +import com.sun.tools.javac.api.JavacTool; 1.35 +import com.sun.tools.javac.tree.JCTree; 1.36 +import com.sun.tools.javac.tree.Pretty; 1.37 + 1.38 +/** 1.39 + * @test 1.40 + * @bug 6902720 1.41 + * @summary javac pretty printer does not handle enums correctly 1.42 + */ 1.43 + 1.44 +public class Test { 1.45 + 1.46 + public static void main(String[] args) throws Exception { 1.47 + Test t = new Test(); 1.48 + t.run("E1.java", "E2.java"); 1.49 + } 1.50 + 1.51 + void run(String... args) throws Exception { 1.52 + File testSrcDir = new File(System.getProperty("test.src")); 1.53 + for (String arg: args) { 1.54 + test(new File(testSrcDir, arg)); 1.55 + } 1.56 + } 1.57 + 1.58 + void test(File test) throws Exception { 1.59 + JavacTool tool1 = JavacTool.create(); 1.60 + StandardJavaFileManager fm = tool1.getStandardFileManager(null, null, null); 1.61 + Iterable<? extends JavaFileObject> files = fm.getJavaFileObjects(test); 1.62 + 1.63 + // parse test file into a tree, and write it out to a stringbuffer using Pretty 1.64 + JavacTask t1 = tool1.getTask(null, fm, null, null, null, files); 1.65 + StringWriter sw = new StringWriter(); 1.66 + PrintWriter pw = new PrintWriter(sw); 1.67 + Iterable<? extends CompilationUnitTree> trees = t1.parse(); 1.68 + for (CompilationUnitTree tree: trees) { 1.69 + new Pretty(pw, true).printExpr((JCTree) tree); 1.70 + } 1.71 + pw.close(); 1.72 + 1.73 + final String out = sw.toString(); 1.74 + System.err.println("generated code:\n" + out + "\n"); 1.75 + 1.76 + // verify the generated code is valid Java by compiling it 1.77 + JavacTool tool2 = JavacTool.create(); 1.78 + JavaFileObject fo = new SimpleJavaFileObject(URI.create("output"), JavaFileObject.Kind.SOURCE) { 1.79 + @Override 1.80 + public CharSequence getCharContent(boolean ignoreEncodingErrors) { 1.81 + return out; 1.82 + } 1.83 + }; 1.84 + JavacTask t2 = tool2.getTask(null, fm, null, null, null, Collections.singleton(fo)); 1.85 + boolean ok = t2.call(); 1.86 + if (!ok) 1.87 + throw new Exception("compilation of generated code failed"); 1.88 + 1.89 + File expectedClass = new File(test.getName().replace(".java", ".class")); 1.90 + if (!expectedClass.exists()) 1.91 + throw new Exception(expectedClass + " not found"); 1.92 + } 1.93 +} 1.94 +