test/hotspot/jtreg/vmTestbase/vm/mlvm/meth/share/SimpleUnitTest.java

/*
 * Copyright (c) 2011, 2018, 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.
 */

package vm.mlvm.meth.share;

import java.lang.invoke.MethodHandles;
import java.lang.invoke.MethodType;

import vm.mlvm.share.Env;
import vm.mlvm.share.MlvmTest;
import vm.share.options.Option;

public class SimpleUnitTest extends MlvmTest {

    @Option(name = "failOnce", default_value = "false", description = "exit after the first failure")
    private boolean failOnce = true;

    public SimpleUnitTest() {}

    public SimpleUnitTest test1(int a, float b) {
        Env.traceNormal("test1(%d, %f) called", a, b);
        return this;
    }

    private static SimpleUnitTest sut = new SimpleUnitTest();

    public static SimpleUnitTest test2(int a, float b) {
        Env.traceNormal("test2(%d, %f) called", a, b);
        return sut;
    }

    public static int test3(int a) {
        Env.traceNormal("test3(%d) called", a);
        return a;
    }

    public void test4() {
        Env.traceNormal("test4() called");
    }

    public SimpleUnitTest test5() {
        Env.traceNormal("test5() called");
        return this;
    }

    public static void main(String[] args) { MlvmTest.launch(args); }

    @Override
    public boolean run() throws Throwable {
        try {
            Argument retArg;
            retArg = new Argument(SimpleUnitTest.class, sut);
            retArg.setPreserved(true);

            Argument intArg = new Argument(int.class, new Integer(1));

            for ( ;; ) {
                try {
                    switch ( Env.getRNG().nextInt(5) ) {
                    case 0:
                        MHTransformationGen.createAndCallSequence(
                            retArg,
                            sut,
                            MethodHandles.lookup().findVirtual(
                                    SimpleUnitTest.class,
                                    "test1",
                                    MethodType.methodType(SimpleUnitTest.class, int.class, float.class)
                            ),
                            new Argument[] { new Argument(int.class, new Integer(1)), new Argument(float.class, new Float(1.0)) },
                            true);
                        break;

                    case 1:
                        MHTransformationGen.createAndCallSequence(
                                retArg,
                                null,
                                MethodHandles.lookup().findStatic(
                                        SimpleUnitTest.class,
                                        "test2",
                                        MethodType.methodType(SimpleUnitTest.class, int.class, float.class)
                                ),
                                new Argument[] { new Argument(int.class, new Integer(1)), new Argument(float.class, new Float(1.0)) },
                                true);
                            break;

                    case 2:
                        MHTransformationGen.createAndCallSequence(
                                intArg,
                                null,
                                MethodHandles.lookup().findStatic(
                                        SimpleUnitTest.class,
                                        "test3",
                                        MethodType.methodType(int.class, int.class)
                                ),
                                new Argument[] { intArg },
                                true);
                            break;

                    case 3:
                        MHTransformationGen.createAndCallSequence(
                                new Argument(void.class, null),
                                sut,
                                MethodHandles.lookup().findVirtual(
                                        SimpleUnitTest.class,
                                        "test4",
                                        MethodType.methodType(void.class)
                                ),
                                new Argument[0],
                                false);
                            break;

                    default:
                        MHTransformationGen.createAndCallSequence(
                                retArg,
                                sut,
                                MethodHandles.lookup().findVirtual(
                                        SimpleUnitTest.class,
                                        "test5",
                                        MethodType.methodType(SimpleUnitTest.class)
                                ),
                                new Argument[0],
                                true);
                            break;

                    }
                } catch ( Throwable e ) {
                    Env.getLog().complain("Caught exception", e);
                    if ( failOnce )
                        return false;
                }
            }
        } catch ( Throwable t ) {
            t.printStackTrace();
            return false;
        }
    }

}