1 /* 2 * Copyright (c) 2016, 2019, 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 */ 23 24 25 package org.graalvm.compiler.replacements.test; 26 27 import jdk.vm.ci.code.InstalledCode; 28 import jdk.vm.ci.meta.ResolvedJavaMethod; 29 import org.graalvm.compiler.nodes.StructuredGraph; 30 import org.graalvm.compiler.nodes.graphbuilderconf.GraphBuilderConfiguration; 31 import org.graalvm.compiler.options.OptionValues; 32 import org.graalvm.compiler.replacements.ConstantBindingParameterPlugin; 33 import org.junit.Test; 34 35 public class StringIndexOfConstantTest extends StringIndexOfTestBase { 36 StringIndexOfConstantTest(String sourceString, String constantString)37 public StringIndexOfConstantTest(String sourceString, String constantString) { 38 super(sourceString, constantString); 39 } 40 41 /* 42 * These test definitions could live in the superclass except that the mx junit individual test 43 * runner can't find tests in superclasses. 44 */ 45 @Override 46 @Test testStringIndexOfConstant()47 public void testStringIndexOfConstant() { 48 super.testStringIndexOfConstant(); 49 } 50 51 @Override 52 @Test testStringIndexOfConstantOffset()53 public void testStringIndexOfConstantOffset() { 54 super.testStringIndexOfConstantOffset(); 55 } 56 57 @Override 58 @Test testStringBuilderIndexOfConstant()59 public void testStringBuilderIndexOfConstant() { 60 super.testStringBuilderIndexOfConstant(); 61 } 62 63 @Override 64 @Test testStringBuilderIndexOfConstantOffset()65 public void testStringBuilderIndexOfConstantOffset() { 66 super.testStringBuilderIndexOfConstantOffset(); 67 } 68 69 Object[] constantArgs; 70 71 @Override editGraphBuilderConfiguration(GraphBuilderConfiguration conf)72 protected GraphBuilderConfiguration editGraphBuilderConfiguration(GraphBuilderConfiguration conf) { 73 if (constantArgs != null) { 74 ConstantBindingParameterPlugin constantBinding = new ConstantBindingParameterPlugin(constantArgs, this.getMetaAccess(), this.getSnippetReflection()); 75 conf.getPlugins().appendParameterPlugin(constantBinding); 76 } 77 return super.editGraphBuilderConfiguration(conf); 78 } 79 80 @Override test(OptionValues options, ResolvedJavaMethod method, Object receiver, Object... args)81 protected Result test(OptionValues options, ResolvedJavaMethod method, Object receiver, Object... args) { 82 constantArgs = new Object[args.length + 1]; 83 for (int i = 0; i < args.length; i++) { 84 if (args[i] == constantString) { 85 constantArgs[i + 1] = constantString; 86 } 87 } 88 return super.test(options, method, receiver, args); 89 } 90 91 @Override getCode(final ResolvedJavaMethod installedCodeOwner, StructuredGraph graph0, boolean ignoreForceCompile, boolean ignoreInstallAsDefault, OptionValues options)92 protected InstalledCode getCode(final ResolvedJavaMethod installedCodeOwner, StructuredGraph graph0, boolean ignoreForceCompile, boolean ignoreInstallAsDefault, OptionValues options) { 93 // Force recompile if constant binding should be done 94 return super.getCode(installedCodeOwner, graph0, 95 /* forceCompile */ true, /* installAsDefault */ false, options); 96 } 97 } 98