1 /*
   2  * Copyright (c) 2016, 2018, 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 
  37     /*
  38      * These test definitions could live in the superclass except that the mx junit individual test
  39      * runner can't find tests in superclasses.
  40      */
  41     @Override
  42     @Test
  43     public void testStringIndexOfConstant() {
  44         super.testStringIndexOfConstant();
  45     }
  46 
  47     @Override
  48     @Test
  49     public void testStringIndexOfConstantOffset() {
  50         super.testStringIndexOfConstantOffset();
  51     }
  52 
  53     @Override
  54     @Test
  55     public void testStringBuilderIndexOfConstant() {
  56         super.testStringBuilderIndexOfConstant();
  57     }
  58 
  59     @Override
  60     @Test
  61     public void testStringBuilderIndexOfConstantOffset() {
  62         super.testStringBuilderIndexOfConstantOffset();
  63     }
  64 
  65     Object[] constantArgs;
  66 
  67     @Override
  68     protected GraphBuilderConfiguration editGraphBuilderConfiguration(GraphBuilderConfiguration conf) {
  69         if (constantArgs != null) {
  70             ConstantBindingParameterPlugin constantBinding = new ConstantBindingParameterPlugin(constantArgs, this.getMetaAccess(), this.getSnippetReflection());
  71             conf.getPlugins().appendParameterPlugin(constantBinding);
  72         }
  73         return super.editGraphBuilderConfiguration(conf);
  74     }
  75 
  76     @Override
  77     protected Result test(OptionValues options, ResolvedJavaMethod method, Object receiver, Object... args) {
  78         constantArgs = new Object[args.length + 1];
  79         for (int i = 0; i < args.length; i++) {
  80             if (args[i] == constantString) {
  81                 constantArgs[i + 1] = constantString;
  82             }
  83         }
  84         return super.test(options, method, receiver, args);
  85     }
  86 
  87     @Override
  88     protected InstalledCode getCode(final ResolvedJavaMethod installedCodeOwner, StructuredGraph graph0, boolean ignoreForceCompile, boolean ignoreInstallAsDefault, OptionValues options) {
  89         // Force recompile if constant binding should be done
  90         return super.getCode(installedCodeOwner, graph0, /* forceCompile */true, /* installAsDefault */false, options);
  91     }
  92 }