/* * Copyright (c) 2011, 2015, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */ /** * @test * @bug 8003280 * @summary Add lambda tests * Test SAM conversion of lambda expressions in combinations of different contexts, * lambda body types(statement/expression), explict/implicit target type etc, to verify * SAM conversion being conducted successfully as expected. * @modules jdk.compiler */ import com.sun.source.util.JavacTask; import java.net.URI; import java.util.Arrays; import javax.tools.Diagnostic; import javax.tools.JavaCompiler; import javax.tools.JavaFileManager; import javax.tools.JavaFileObject; import javax.tools.SimpleJavaFileObject; import javax.tools.ToolProvider; public class SamConversionComboTest { enum FInterface { A("A", "interface A { Integer m(int i); }"), B("B", "interface B { int m(Integer i); }"), C("C", "interface C { int m(Integer i) throws Exception; }"); String interfaceType; String interfaceDef; FInterface(String interfaceType, String interfaceDef) { this.interfaceType = interfaceType; this.interfaceDef = interfaceDef; } String getParameterType() { switch(this) { case A: return "int"; case B: case C: return "Integer"; default: return null; } } } enum Context { ASSIGNMENT("#FType f = #LBody;"), METHOD_CALL("void method1(#FType f) { }\n" + " void method2() {\n" + " method1(#LBody);\n" + " }"), CONSTRUCTOR("X x = new X(#LBody);"), RETURN_OF_METHOD("#FType method1() {\n" + " return #LBody;\n" + "}"), ARRAY_INITIALIZER("Object[] oarray = {\"a\", 1, (#FType)#LBody};"), LAMBDA_BODY("#FType f = n -> ((#FType)#LBody).m(n);"), CAST("void test() throws Exception { int n = ((#FType)#LBody).m(1); }"), CONDITIONAL_EXPRESSION("#FType f = 2 > 1 ? #LBody : null;"); String context; Context(String context) { this.context = context; } String getContext(FInterface f, LambdaKind lk, LambdaBody lb, ReturnValue rv) { return context.replace("#FType", f.interfaceType).replace("#LBody", lb.getLambdaBody(f, lk, rv)); } } enum LambdaKind { EXPRESSION("#VAL"), STATEMENT("{return #VAL;}"), EXCEPTION_STMT("{throw new Exception();}"); String stmt; LambdaKind(String stmt) { this.stmt = stmt; } } enum ReturnValue { INT("i + 1"), INTEGER("new Integer(i+1)"), INT2("i.intValue() + 1"), STRING("i + \"\""), DOUBLE("i * 1.0"); String rValue; ReturnValue(String rValue) { this.rValue = rValue; } } enum LambdaBody { IMPLICIT("i -> #RET"),//type inferred EXPLICIT("(#Type i) -> #RET");//explicit type String bodyStr; LambdaBody(String bodyStr) { this.bodyStr = bodyStr; } String getLambdaBody(FInterface fi, LambdaKind lk, ReturnValue rv) { return bodyStr.replace("#Type", fi.getParameterType()).replace("#RET", lk.stmt.replace("#VAL", rv.rValue)); } } boolean checkSamConversion() { if(lambdaKind != LambdaKind.EXCEPTION_STMT && (returnValue == ReturnValue.DOUBLE || returnValue == ReturnValue.STRING)) //return type mismatch return false; if(context != Context.CONSTRUCTOR) {//context other than construcotr argument if(fInterface != FInterface.C && lambdaKind == LambdaKind.EXCEPTION_STMT) return false; if(fInterface == FInterface.A && returnValue == ReturnValue.INT2) return false; } else { //constructor argument context //match X(A a) or X(B b) or X(C c) if (lambdaKind == LambdaKind.EXCEPTION_STMT) { return false; //ambiguous target type } else if(lambdaBody == LambdaBody.IMPLICIT) { return false; } else { //explicit parameter type if(fInterface.getParameterType().equals("Integer")) //ambiguous target type //e.g. X x = new X((Integer i) -> i + 1); return false; if(returnValue == ReturnValue.INT2) //e.g. X x = new X(int i -> i.intValue() + 1); return false; } } return true; } SourceFile samSourceFile = new SourceFile("FInterface.java", "#C") { public String toString() { String interfaces = ""; for(FInterface fi : FInterface.values()) interfaces += fi.interfaceDef + "\n"; return template.replace("#C", interfaces); } }; String clientTemplate = "class Client {\n" + " #Context\n" + "}\n\n" + "class X {\n" + " int value = 0;\n\n" + " X(A a) {\n" + " value = a.m(6);\n" + " }\n\n" + " X(B b) {\n" + " value = b.m(7);\n" + " }\n\n" + " X(C c) {\n" + " try {\n" + " value = c.m(8);\n" + " } catch (Exception e){}\n" + " }\n" + "}"; SourceFile clientSourceFile = new SourceFile("Client.java", clientTemplate) { public String toString() { return template.replace("#Context", context.getContext(fInterface, lambdaKind, lambdaBody, returnValue)); } }; void test() throws Exception { System.out.println("\n===================================="); System.out.println(fInterface + ", " + context + ", " + lambdaKind + ", " + lambdaBody + ", " + returnValue); System.out.println(samSourceFile + "\n"); String clientFileStr = clientSourceFile.toString(); System.out.println(clientFileStr.substring(0, clientFileStr.indexOf("\n\n"))); DiagnosticChecker dc = new DiagnosticChecker(); JavacTask ct = (JavacTask)comp.getTask(null, fm, dc, null, null, Arrays.asList(samSourceFile, clientSourceFile)); try { ct.analyze(); } catch (Exception e) { throw new AssertionError("failing SAM source file \n" + samSourceFile + "\n\n" + "failing client source file \n"+ clientSourceFile); } if (dc.errorFound == checkSamConversion()) { throw new AssertionError(samSourceFile + "\n\n" + clientSourceFile); } count++; } abstract class SourceFile extends SimpleJavaFileObject { protected String template; public SourceFile(String filename, String template) { super(URI.create("myfo:/" + filename), JavaFileObject.Kind.SOURCE); this.template = template; } @Override public CharSequence getCharContent(boolean ignoreEncodingErrors) { return toString(); } public abstract String toString(); } static class DiagnosticChecker implements javax.tools.DiagnosticListener<JavaFileObject> { boolean errorFound = false; public void report(Diagnostic<? extends JavaFileObject> diagnostic) { if (diagnostic.getKind() == Diagnostic.Kind.ERROR) { errorFound = true; } } } FInterface fInterface; Context context; LambdaBody lambdaBody; LambdaKind lambdaKind; ReturnValue returnValue; static int count = 0; static JavaCompiler comp = ToolProvider.getSystemJavaCompiler(); static JavaFileManager fm = comp.getStandardFileManager(null, null, null); SamConversionComboTest(FInterface f, Context c, LambdaBody lb, LambdaKind lk, ReturnValue rv) { fInterface = f; context = c; lambdaKind = lk; lambdaBody = lb; returnValue = rv; } public static void main(String[] args) throws Exception { try { for(Context ct : Context.values()) { for (FInterface fi : FInterface.values()) { for (LambdaKind lk: LambdaKind.values()) { for (LambdaBody lb : LambdaBody.values()) { for(ReturnValue rv : ReturnValue.values()) { new SamConversionComboTest(fi, ct, lb, lk, rv).test(); } } } } } System.out.println("total tests: " + count); } finally { fm.close(); } } }