test/jdk/java/lang/invoke/condy/CondyStaticArgumentsTest.java
changeset 48826 c4d9d1b08e2e
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/test/jdk/java/lang/invoke/condy/CondyStaticArgumentsTest.java	Fri Sep 08 10:46:46 2017 -0700
@@ -0,0 +1,201 @@
+/*
+ * Copyright (c) 2017, 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 8186046
+ * @summary Test bootstrap arguments for condy
+ * @library /lib/testlibrary/bytecode /java/lang/invoke/common
+ * @build jdk.experimental.bytecode.BasicClassBuilder test.java.lang.invoke.lib.InstructionHelper
+ * @run testng CondyStaticArgumentsTest
+ * @run testng/othervm -XX:+UnlockDiagnosticVMOptions -XX:UseBootstrapCallInfo=3 CondyStaticArgumentsTest
+ */
+
+import jdk.experimental.bytecode.PoolHelper;
+import org.testng.Assert;
+import org.testng.annotations.Test;
+import test.java.lang.invoke.lib.InstructionHelper;
+
+import java.lang.invoke.ConstantCallSite;
+import java.lang.invoke.MethodHandle;
+import java.lang.invoke.MethodHandleInfo;
+import java.lang.invoke.MethodHandles;
+import java.lang.invoke.MethodType;
+import java.lang.reflect.Method;
+import java.math.BigDecimal;
+import java.math.MathContext;
+import java.util.StringJoiner;
+import java.util.stream.Stream;
+
+import static java.lang.invoke.MethodType.methodType;
+
+public class CondyStaticArgumentsTest {
+    static final MethodHandles.Lookup L = MethodHandles.lookup();
+
+    static class BSMInfo {
+        final String methodName;
+        final MethodHandle handle;
+        final String descriptor;
+
+        BSMInfo(String name) {
+            methodName = name;
+
+            Method m = Stream.of(CondyStaticArgumentsTest.class.getDeclaredMethods())
+                    .filter(x -> x.getName().equals(methodName)).findFirst()
+                    .get();
+            try {
+                handle = MethodHandles.lookup().unreflect(m);
+            }
+            catch (Exception e) {
+                throw new Error(e);
+            }
+            descriptor = handle.type().toMethodDescriptorString();
+        }
+
+        static BSMInfo of(String name) {
+            return new BSMInfo(name);
+        }
+    }
+
+    static String basicArgs(MethodHandles.Lookup l, String name, Class<?> type,
+                            int i, long j, float f, double d,
+                            Class<?> c, String s,
+                            MethodType mt, MethodHandle mh) {
+        return new StringJoiner("-")
+                .add(name)
+                .add(type.getSimpleName())
+                .add(Integer.toString(i))
+                .add(Long.toString(j))
+                .add(Float.toString(f))
+                .add(Double.toString(d))
+                .add(c.getSimpleName())
+                .add(s)
+                .add(mt.toString())
+                .add(Integer.toString(mh.type().parameterCount()))
+                .toString();
+    }
+
+    @Test
+    public void testBasicArgs() throws Throwable {
+        BSMInfo bi = BSMInfo.of("basicArgs");
+        MethodHandleInfo mhi = MethodHandles.lookup().revealDirect(bi.handle);
+
+        MethodHandle mh = InstructionHelper.ldcDynamicConstant(
+                L, "constant-name", String.class,
+                bi.methodName, bi.handle.type(),
+                S -> S.add(1).add(2L).add(3.0f).add(4.0d)
+                        .add("java/lang/Number", PoolHelper::putClass)
+                        .add("something", PoolHelper::putString)
+                        .add("(IJFD)V", PoolHelper::putMethodType)
+                        .add(mhi, (P, Z) -> {
+                            return P.putHandle(mhi.getReferenceKind(), "CondyStaticArgumentsTest", mhi.getName(), bi.descriptor);
+                        }));
+
+        Assert.assertEquals(mh.invoke(), "constant-name-String-1-2-3.0-4.0-Number-something-(int,long,float,double)void-11");
+    }
+
+
+    static MathContext mathContext(MethodHandles.Lookup l, String value, Class<?> type) {
+        switch (value) {
+            case "UNLIMITED":
+                return MathContext.UNLIMITED;
+            case "DECIMAL32":
+                return MathContext.DECIMAL32;
+            case "DECIMAL64":
+                return MathContext.DECIMAL64;
+            case "DECIMAL128":
+                return MathContext.DECIMAL128;
+            default:
+                throw new UnsupportedOperationException();
+        }
+    }
+
+    static BigDecimal bigDecimal(MethodHandles.Lookup l, String name, Class<?> type,
+                                 String value, MathContext mc) {
+        return new BigDecimal(value, mc);
+    }
+
+    static String condyWithCondy(MethodHandles.Lookup l, String name, Class<?> type,
+                                 BigDecimal d) {
+        return new StringJoiner("-")
+                .add(name)
+                .add(type.getSimpleName())
+                .add(d.toString())
+                .add(Integer.toString(d.precision()))
+                .toString();
+    }
+
+    static <E> int bigDecimalPoolHelper(String value, String mc, PoolHelper<String, String, E> P) {
+        BSMInfo bi = BSMInfo.of("bigDecimal");
+        return P.putDynamicConstant("big-decimal", "Ljava/math/BigDecimal;", InstructionHelper.csym(L.lookupClass()), bi.methodName, bi.descriptor,
+                                    S -> S.add(value, PoolHelper::putString)
+                                            .add(mc, (P2, s) -> {
+                                                return mathContextPoolHelper(s, P2);
+                                            }));
+    }
+
+    static <E> int mathContextPoolHelper(String mc, PoolHelper<String, String, E> P) {
+        BSMInfo bi = BSMInfo.of("mathContext");
+        return P.putDynamicConstant(mc, "Ljava/math/MathContext;", InstructionHelper.csym(L.lookupClass()), bi.methodName, bi.descriptor,
+                                    S -> {
+                                    });
+    }
+
+    @Test
+    public void testCondyWithCondy() throws Throwable {
+        BSMInfo bi = BSMInfo.of("condyWithCondy");
+
+        MethodHandle mh = InstructionHelper.ldcDynamicConstant(
+                L, "big-decimal-math-context", String.class,
+                bi.methodName, bi.handle.type(),
+                S -> S.add(null, (P, v) -> {
+                    return bigDecimalPoolHelper("3.14159265358979323846", "DECIMAL32", P);
+                }));
+        Assert.assertEquals(mh.invoke(), "big-decimal-math-context-String-3.141593-7");
+    }
+
+
+    static ConstantCallSite indyWithCondy(MethodHandles.Lookup l, String name, MethodType type,
+                                          BigDecimal d) {
+        String s = new StringJoiner("-")
+                .add(name)
+                .add(type.toMethodDescriptorString())
+                .add(d.toString())
+                .add(Integer.toString(d.precision()))
+                .toString();
+        return new ConstantCallSite(MethodHandles.constant(String.class, s));
+    }
+
+    @Test
+    public void testIndyWithCondy() throws Throwable {
+        BSMInfo bi = BSMInfo.of("indyWithCondy");
+
+        MethodHandle mh = InstructionHelper.invokedynamic(
+                L, "big-decimal-math-context", methodType(String.class),
+                bi.methodName, bi.handle.type(),
+                S -> S.add(null, (P, v) -> {
+                    return bigDecimalPoolHelper("3.14159265358979323846", "DECIMAL32", P);
+                }));
+        Assert.assertEquals(mh.invoke(), "big-decimal-math-context-()Ljava/lang/String;-3.141593-7");
+    }
+}