1/*
2 * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23package org.graalvm.compiler.replacements.test;
24
25import org.graalvm.compiler.nodes.StructuredGraph;
26import org.graalvm.compiler.nodes.graphbuilderconf.GraphBuilderConfiguration;
27import org.graalvm.compiler.options.OptionValues;
28import org.graalvm.compiler.replacements.ConstantBindingParameterPlugin;
29import org.junit.Test;
30
31import jdk.vm.ci.code.InstalledCode;
32import jdk.vm.ci.meta.ResolvedJavaMethod;
33
34public class StringIndexOfConstantTest extends StringIndexOfTestBase {
35
36    /*
37     * These test definitions could live in the superclass except that the mx junit individual test
38     * runner can't find tests in superclasses.
39     */
40    @Override
41    @Test
42    public void testStringIndexOfConstant() {
43        super.testStringIndexOfConstant();
44    }
45
46    @Override
47    @Test
48    public void testStringIndexOfConstantOffset() {
49        super.testStringIndexOfConstantOffset();
50    }
51
52    @Override
53    @Test
54    public void testStringBuilderIndexOfConstant() {
55        super.testStringBuilderIndexOfConstant();
56    }
57
58    @Override
59    @Test
60    public void testStringBuilderIndexOfConstantOffset() {
61        super.testStringBuilderIndexOfConstantOffset();
62    }
63
64    Object[] constantArgs;
65
66    @Override
67    protected GraphBuilderConfiguration editGraphBuilderConfiguration(GraphBuilderConfiguration conf) {
68        if (constantArgs != null) {
69            ConstantBindingParameterPlugin constantBinding = new ConstantBindingParameterPlugin(constantArgs, this.getMetaAccess(), this.getSnippetReflection());
70            conf.getPlugins().appendParameterPlugin(constantBinding);
71        }
72        return super.editGraphBuilderConfiguration(conf);
73    }
74
75    @Override
76    protected Result test(OptionValues options, ResolvedJavaMethod method, Object receiver, Object... args) {
77        constantArgs = new Object[args.length + 1];
78        for (int i = 0; i < args.length; i++) {
79            if (args[i] == constantString) {
80                constantArgs[i + 1] = constantString;
81            }
82        }
83        return super.test(options, method, receiver, args);
84    }
85
86    @Override
87    protected InstalledCode getCode(final ResolvedJavaMethod installedCodeOwner, StructuredGraph graph0, boolean ignoreForceCompile, boolean ignoreInstallAsDefault, OptionValues options) {
88        // Force recompile if constant binding should be done
89        return super.getCode(installedCodeOwner, graph0, /* forceCompile */true, /* installAsDefault */false, options);
90    }
91}
92