/*
* Copyright (C) 2010 The Project Lombok Authors.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
package lombok.ast.grammar;
import static org.junit.Assert.*;
import java.nio.charset.Charset;
import java.util.List;
import javax.tools.JavaFileManager;
import lombok.ast.Node;
import lombok.ast.javac.JcTreeBuilder;
import lombok.ast.javac.JcTreePrinter;
import org.junit.Test;
import org.junit.runner.RunWith;
import com.sun.tools.javac.main.JavaCompiler;
import com.sun.tools.javac.tree.JCTree;
import com.sun.tools.javac.util.Context;
@RunWith(RunForEachFileInDirRunner.class)
public class JcTreeBuilderTest extends TreeBuilderRunner<JCTree> {
public JcTreeBuilderTest() {
super(true);
}
@Test
public boolean testJcTreeBuilder(Source source) throws Exception {
return testCompiler(source);
}
protected String convertToString(JCTree tree) {
JcTreePrinter printer = JcTreePrinter.printerWithPositions();
printer.visit(tree);
String string = printer.toString();
return string;
}
protected JCTree parseWithLombok(Source source) {
List<Node> nodes = source.getNodes();
assertEquals(1, nodes.size());
Context context = new Context();
boolean success = false;
Throwable failTrace = null;
try {
Class<?> dfm = Class.forName("com.sun.tools.javac.util.DefaultFileManager");
JavaFileManager instance = (JavaFileManager) dfm.getConstructor(Context.class, boolean.class, Charset.class).newInstance(context, true, Charset.forName("UTF-8"));
context.put(JavaFileManager.class, instance);
success = true;
} catch (Throwable t) {
//Either DFM, or its replacement JFM, exists (or possibly both in odd classpath configurations). If something is wrong, NoMethodDefErrors and the like occur.
failTrace = t;
}
try {
Class<?> jfm = Class.forName("com.sun.tools.javac.util.JavacFileManager");
JavaFileManager instance = (JavaFileManager) jfm.getConstructor(Context.class, boolean.class, Charset.class).newInstance(context, true, Charset.forName("UTF-8"));
context.put(JavaFileManager.class, instance);
success = true;
} catch (Throwable t) {
//Either DFM, or its replacement JFM, exists (or possibly both in odd classpath configurations). If something is wrong, NoMethodDefErrors and the like occur.
failTrace = t;
}
if (!success) {
if (failTrace instanceof Error) throw (Error)failTrace;
throw new RuntimeException("Neither com.sun.tools.javac.util.JavacFileManager nor com.sun.tools.javac.util.DefaultFileManager could be configured", failTrace);
}
JcTreeBuilder builder = new JcTreeBuilder(source, context);
builder.visit(nodes.get(0));
return builder.get();
}
protected JCTree parseWithTargetCompiler(Source source) throws Exception {
Context context = new Context();
JavaCompiler compiler = new JavaCompiler(context);
compiler.genEndPos = true;
JCTree result = compiler.parse(new ContentBasedJavaFileObject(source.getName(), source.getRawInput()));
return compiler.errorCount() > 0 ? null : result;
}
}