package org.graalvm.compiler.replacements.test;
import org.graalvm.compiler.core.test.GraalCompilerTest;
import org.graalvm.compiler.nodes.ConstantNode;
import org.graalvm.compiler.nodes.FixedNode;
import org.graalvm.compiler.nodes.ReturnNode;
import org.graalvm.compiler.nodes.StructuredGraph;
import org.graalvm.compiler.nodes.ValueNode;
import jdk.vm.ci.meta.JavaConstant;
import org.junit.Test;
import jdk.vm.ci.meta.ResolvedJavaMethod;
import static org.hamcrest.CoreMatchers.instanceOf;
import org.junit.Assert;
public class StringEqualsConstantTest extends GraalCompilerTest {
private ValueNode asConstant(StructuredGraph graph, String str) {
return ConstantNode.forConstant(getSnippetReflection().forObject(str), getMetaAccess(), graph);
}
private void testStringEquals(String s0, String s1) {
ResolvedJavaMethod method = getResolvedJavaMethod("stringEquals");
StructuredGraph graph = parseForCompile(method);
graph.getParameter(0).replaceAndDelete(asConstant(graph, s0));
graph.getParameter(1).replaceAndDelete(asConstant(graph, s1));
compile(method, graph);
FixedNode firstFixed = graph.start().next();
Assert.assertThat(firstFixed, instanceOf(ReturnNode.class));
ReturnNode ret = (ReturnNode) firstFixed;
JavaConstant result = ret.result().asJavaConstant();
if (result == null) {
Assert.fail("result not constant: " + ret.result());
} else {
int expected = s0.equals(s1) ? 1 : 0;
Assert.assertEquals("result", expected, result.asInt());
}
}
@Test
public void testSameString() {
String s = "equal-string";
testStringEquals(s, s);
}
@Test
public void testEqualString() {
String s = "equal-string";
testStringEquals(s, new String(s.toCharArray()));
}
@Test
public void testDifferentString() {
testStringEquals("some-string", "different-string");
}
@Test
public void testSameLengthString() {
testStringEquals("string-1", "string-2");
}
public static boolean stringEquals(String a, String b) {
return a.equals(b);
}
}