diff -r 843d3b191773 -r 01c9d4161882 test/tools/javac/lambda/methodReference/SamConversion.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/test/tools/javac/lambda/methodReference/SamConversion.java Sat Nov 17 19:01:03 2012 +0000 @@ -0,0 +1,337 @@ +/* + * Copyright (c) 2012, 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 method references in contexts of assignment, method/constructor argument, + * return statement, array initializer, lambda expression body, conditional expression and cast. + * @compile SamConversion.java + * @run main SamConversion + */ + +public class SamConversion { + + static int assertionCount = 0; + + static interface Foo { + Integer m(int i); + } + + static interface Bar { + int m(Integer i) throws MyException; + } + + private static void assertTrue(boolean b) { + assertionCount++; + if(!b) + throw new AssertionError(); + } + + private static int test1(Foo foo) { + return foo.m(1); + } + + private static void test2(Bar bar, int result) { + try { + assertTrue(bar.m(1) == result); + } catch (Exception e){ + assertTrue(false); + } + } + + private static Bar test3(int i) { + switch (i) { + case 0: + return A::method1; + case 1: + return new A()::method2; + case 2: + return A::method3; + case 3: + return new A()::method4; + case 4: + return new A()::method5; + case 5: + return A::method6; + default: + return null; + } + } + + /** + * Test SAM conversion of method reference in assignment context + */ + private static void testAssignment() { + Foo foo = A::method1; //static reference, parameter type matching and return type matching + assertTrue(foo.m(1) == 2); + + foo = new A()::method2; //instance reference, parameter type unboxing and return type boxing + assertTrue(foo.m(1) == 3); + + foo = A::method3; //static reference, parameter type matching and return type boxing + assertTrue(foo.m(1) == 4); + + foo = new A()::method4; //instance reference, parameter type unboxing and return type matching + assertTrue(foo.m(1) == 5); + + foo = new A()::method5; //instance reference, parameter type unboxing and return type matching + assertTrue(foo.m(1) == 6); + + Bar bar = A::method1; + try { + assertTrue(bar.m(1) == 2); + } catch (Exception e) { + assertTrue(false); + } + + bar = new A()::method2; + try { + assertTrue(bar.m(1) == 3); + } catch (Exception e) { + assertTrue(false); + } + + bar = A::method3; + try { + assertTrue(bar.m(1) == 4); + } catch (Exception e) { + assertTrue(false); + } + + bar = new A()::method4; + try { + assertTrue(bar.m(1) == 5); + } catch (Exception e) { + assertTrue(false); + } + + bar = new A()::method5; + try { + assertTrue(bar.m(1) == 6); + } catch (Exception e) { + assertTrue(false); + } + + bar = new A()::method6; + try { + bar.m(1); + assertTrue(false); + } catch (MyException e) { + } catch (Exception e) { + assertTrue(false); + } + } + + /** + * Test SAM conversion of method reference in method/constructor argument context + */ + private static void testMethodArgument() { + assertTrue(test1(A::method1) == 2); + assertTrue(test1(new A()::method2) == 3); + assertTrue(test1(A::method3) == 4); + assertTrue(test1(new A()::method4) == 5); + assertTrue(test1(new A()::method5) == 6); + test2(A::method1, 2); + test2(new A()::method2, 3); + test2(A::method3, 4); + test2(new A()::method4, 5); + test2(new A()::method5, 6); + A a = new A(A::method1); //A(Foo f) called + assertTrue(a.method2(1) == 11); + assertTrue(a.method4(1) == 11); + assertTrue(a.method5(1) == 11); + A a2 = new A(new A()::method2); //A(Bar b) called + assertTrue(a2.method2(1) == 12); + assertTrue(a2.method4(1) == 12); + assertTrue(a2.method5(1) == 12); + } + + /** + * Test SAM conversion of method reference in return statement context + */ + private static void testReturnStatement() { + Bar bar = test3(0); + try { + assertTrue(bar.m(1) == 2); + } catch (Exception e) { + assertTrue(false); + } + bar = test3(1); + try { + assertTrue(bar.m(1) == 3); + } catch (Exception e) { + assertTrue(false); + } + bar = test3(2); + try { + assertTrue(bar.m(1) == 4); + } catch (Exception e) { + assertTrue(false); + } + bar = test3(3); + try { + assertTrue(bar.m(1) == 5); + } catch (Exception e) { + assertTrue(false); + } + bar = test3(4); + try { + assertTrue(bar.m(1) == 6); + } catch (Exception e) { + assertTrue(false); + } + bar = test3(5); + try { + bar.m(1); + assertTrue(false); + } catch (MyException e) { + } catch (Exception e) { + assertTrue(false); + } + } + + /** + * Test SAM conversion of method reference in cast context + */ + private static void testCast() { + assertTrue(((Foo)A::method1).m(1) == 2); + try { + assertTrue(((Bar)new A()::method2).m(1) == 3); + } catch (Exception e) { + assertTrue(false); + } + } + + /** + * Test SAM conversion of method reference in array initializer context + */ + private static void testArrayInitializer() { + Object[] oarray = {"a", 1, (Foo)A::method3}; //last element need a cast + Object[] oarray2 = {"a", 1, (Bar)new A()::method4}; //last element need a cast + Foo[] farray = {A::method1, new A()::method2, A::method3, new A()::method4, new A()::method5}; + Bar[] barray = {A::method1, new A()::method2, A::method3, new A()::method4, new A()::method5, A::method6}; + } + + /** + * Test SAM conversion of method reference in conditional expression context + */ + private static void testConditionalExpression(boolean b) { + Foo f = b ? A::method3 : new A()::method5; + if(b) + assertTrue(f.m(1) == 4); + else + assertTrue(f.m(1) == 6); + + Bar bar = b ? A::method1 : A::method6; + if(b) { + try { + assertTrue(bar.m(1) == 2); + } catch (Exception e) { + assertTrue(false); + } + } + else { + try { + bar.m(1); + assertTrue(false); + } catch (MyException e) { + } catch (Exception e) { + assertTrue(false); + } + } + } + + /** + * Test SAM conversion of method reference in lambda expression body + */ + private static void testLambdaExpressionBody() { + Foo f = n -> ((Foo)A::method3).m(n); + assertTrue(f.m(1) == 4); + + Bar b = n -> { return ((Foo)new A()::method5).m(n); }; + try { + assertTrue(b.m(1) == 6); + } catch (Exception e) { + assertTrue(false); + } + } + + public static void main(String[] args) { + testAssignment(); + testMethodArgument(); + testReturnStatement(); + testCast(); + testArrayInitializer(); + testConditionalExpression(true); + testConditionalExpression(false); + testLambdaExpressionBody(); + + assertTrue(assertionCount == 38); + } + + static class MyException extends Exception {} + + static class A { + + int value = 0; + + A() { + } + + A(Foo f) { + value = f.m(9); + } + + A(Bar b) { + try { + value = b.m(9); + } catch (MyException e){} + } + + static Integer method1(int n) { + return n + 1; + } + + int method2(Integer n) { + return value == 0 ? n + 2 : n + value; + } + + static int method3(int n) { + return n + 3; + } + + Integer method4(Integer n) { + return value == 0 ? n + 4 : n + value; + } + + Integer method5(Integer n) { + return value == 0 ? new Integer(n + 5) : new Integer(n + value); + } + + static int method6(Integer n) throws MyException{ + throw new MyException(); + } + } +}