test/jdk/java/lang/invoke/MethodHandles/TestCatchException.java

/*
 * Copyright (c) 2013, 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 8027823
 * @run junit test.java.lang.invoke.TestCatchException
 */
package test.java.lang.invoke;

import java.lang.invoke.*;
import org.junit.*;
import static org.junit.Assert.*;

public class TestCatchException {
    static final MethodHandles.Lookup LOOKUP = MethodHandles.lookup();
    static final MethodType M_TYPE = MethodType.methodType(int.class, Object.class, Object.class, int.class, int.class, int.class, int.class, int.class, int.class, int.class);

    private static int noThrow(Object o1, Object o2, int i1, int i2, int i3, int i4, int i5, int i6, int i7) {
        return 42;
    }

    private static int throwEx(Object o1, Object o2, int i1, int i2, int i3, int i4, int i5, int i6, int i7) throws Exception {
        throw new Exception();
    }

    private static int handler(Exception e) {
        return 17;
    }

    @Test
    public void testNoThrowPath() throws Throwable {
        MethodHandle target = LOOKUP.findStatic(TestCatchException.class, "noThrow", M_TYPE);
        MethodHandle handler = LOOKUP.findStatic(TestCatchException.class, "handler", MethodType.methodType(int.class, Exception.class));

        MethodHandle h = MethodHandles.catchException(target, Exception.class, handler);

        int x = (int)h.invokeExact(new Object(), new Object(), 1, 2, 3, 4, 5, 6, 7);
        assertEquals(x, 42);
    }

    @Test
    public void testThrowPath() throws Throwable {
        MethodHandle target = LOOKUP.findStatic(TestCatchException.class, "throwEx", M_TYPE);
        MethodHandle handler = LOOKUP.findStatic(TestCatchException.class, "handler", MethodType.methodType(int.class, Exception.class));

        MethodHandle h = MethodHandles.catchException(target, Exception.class, handler);

        int x = (int)h.invokeExact(new Object(), new Object(), 1, 2, 3, 4, 5, 6, 7);
        assertEquals(x, 17);
    }

    static final Object masterParam = new Object();
    static final Object[] masterTail = new Object[] { "str" };
    static Exception masterEx = new Exception();

    public static Object m1(Object o1, Object o2, Object o3, Object o4, Object o5,
                            Object o6, Object o7, Object o8, Object... tail) {
        assertEquals(masterParam, o1);
        assertEquals(masterParam, o2);
        assertEquals(masterParam, o3);
        assertEquals(masterParam, o4);
        assertEquals(masterParam, o5);
        assertEquals(masterParam, o6);
        assertEquals(masterParam, o7);
        assertEquals(masterParam, o8);
        assertEquals(masterTail, tail);
        return tail;
    }

    public static Object m2(Exception e, Object o1, Object o2, Object o3, Object o4,
                            Object o5, Object o6, Object o7, Object o8, Object... tail) {
        assertEquals(masterEx, e);
        assertEquals(masterParam, o1);
        assertEquals(masterParam, o2);
        assertEquals(masterParam, o3);
        assertEquals(masterParam, o4);
        assertEquals(masterParam, o5);
        assertEquals(masterParam, o6);
        assertEquals(masterParam, o7);
        assertEquals(masterParam, o8);
        assertEquals(masterTail, tail);
        return tail;
    }

    public static Object throwEx(Object o1, Object o2, Object o3, Object o4, Object o5,
                                 Object o6, Object o7, Object o8, Object... tail) throws Exception {
        assertEquals(masterParam, o1);
        assertEquals(masterParam, o2);
        assertEquals(masterParam, o3);
        assertEquals(masterParam, o4);
        assertEquals(masterParam, o5);
        assertEquals(masterParam, o6);
        assertEquals(masterParam, o7);
        assertEquals(masterParam, o8);
        assertEquals(masterTail, tail);
        throw masterEx;
    }

    @Test
    public void testVarargsCollectorNoThrow() throws Throwable {
        MethodType t1 = MethodType.methodType(Object.class, Object.class, Object.class, Object.class, Object.class,
                Object.class, Object.class, Object.class, Object.class, Object[].class);

        MethodType t2 = t1.insertParameterTypes(0, Exception.class);

        MethodHandle target = LOOKUP.findStatic(TestCatchException.class, "m1", t1)
                                    .asVarargsCollector(Object[].class);
        MethodHandle catcher = LOOKUP.findStatic(TestCatchException.class, "m2", t2)
                                     .asVarargsCollector(Object[].class);
        MethodHandle gwc = MethodHandles.catchException(target, Exception.class, catcher);

        Object o = masterParam;
        Object[] obj1 = masterTail;

        Object r2 = gwc.invokeExact(o, o, o, o, o, o, o, o, obj1);
        assertEquals(r2, obj1);
    }

    @Test
    public void testVarargsCollectorThrow() throws Throwable {
        MethodType t1 = MethodType.methodType(Object.class, Object.class, Object.class, Object.class, Object.class,
                Object.class, Object.class, Object.class, Object.class, Object[].class);

        MethodType t2 = t1.insertParameterTypes(0, Exception.class);

        MethodHandle target = LOOKUP.findStatic(TestCatchException.class, "throwEx", t1)
                                    .asVarargsCollector(Object[].class);
        MethodHandle catcher = LOOKUP.findStatic(TestCatchException.class, "m2", t2)
                                     .asVarargsCollector(Object[].class);
        MethodHandle gwc = MethodHandles.catchException(target, Exception.class, catcher);

        Object o = masterParam;
        Object[] obj1 = masterTail;

        Object r2 = gwc.invokeExact(o, o, o, o, o, o, o, o, obj1);
        assertEquals(r2, obj1);
    }

    public static void main(String[] args) throws Throwable {
        TestCatchException test = new TestCatchException();
        test.testNoThrowPath();
        test.testThrowPath();
        test.testVarargsCollectorNoThrow();
        test.testVarargsCollectorThrow();
        System.out.println("TEST PASSED");
    }
}