test/langtools/jdk/jshell/ReplaceTest.java
changeset 47216 71c04702a3d5
parent 43366 9e917e04f53c
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/test/langtools/jdk/jshell/ReplaceTest.java	Tue Sep 12 19:03:39 2017 +0200
@@ -0,0 +1,272 @@
+/*
+ * Copyright (c) 2015, 2016, 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.
+ */
+
+/*
+ * @test 8080069 8152925
+ * @summary Test of Snippet redefinition and replacement.
+ * @build KullaTesting TestingInputStream
+ * @run testng ReplaceTest
+ */
+
+import java.util.Iterator;
+import java.util.stream.Stream;
+import jdk.jshell.Snippet;
+import jdk.jshell.MethodSnippet;
+import jdk.jshell.TypeDeclSnippet;
+import jdk.jshell.VarSnippet;
+import org.testng.annotations.Test;
+
+import static org.testng.Assert.assertFalse;
+import static jdk.jshell.Snippet.Status.*;
+import static jdk.jshell.Snippet.SubKind.*;
+import static org.testng.Assert.assertTrue;
+
+@Test
+public class ReplaceTest extends KullaTesting {
+
+    public void testRedefine() {
+        Snippet vx = varKey(assertEval("int x;"));
+        Snippet mu = methodKey(assertEval("int mu() { return x * 4; }"));
+        Snippet c = classKey(assertEval("class C { String v() { return \"#\" + mu(); } }"));
+        assertEval("C c0  = new C();");
+        assertEval("c0.v();", "\"#0\"");
+        assertEval("int x = 10;", "10",
+                ste(MAIN_SNIPPET, VALID, VALID, false, null),
+                ste(vx, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        assertEval("c0.v();", "\"#40\"");
+        assertEval("C c = new C();");
+        assertEval("c.v();", "\"#40\"");
+        assertEval("int mu() { return x * 3; }",
+                ste(MAIN_SNIPPET, VALID, VALID, false, null),
+                ste(mu, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        assertEval("c.v();", "\"#30\"");
+        assertEval("class C { String v() { return \"@\" + mu(); } }",
+                ste(MAIN_SNIPPET, VALID, VALID, false, null),
+                ste(c, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        assertEval("c0.v();", "\"@30\"");
+        assertEval("c = new C();");
+        assertEval("c.v();", "\"@30\"");
+        assertActiveKeys();
+    }
+
+    public void testReplaceClassToVar() {
+        Snippet oldA = classKey(assertEval("class A { public String toString() { return \"old\"; } }"));
+        Snippet v = varKey(assertEval("A a = new A();", "old"));
+        assertEval("a;", "old");
+        Snippet midA = classKey(assertEval("class A { public String toString() { return \"middle\"; } }",
+                ste(MAIN_SNIPPET, VALID, VALID, false, null),
+                ste(oldA, VALID, OVERWRITTEN, false, MAIN_SNIPPET)));
+        assertEval("a;", "middle");
+        assertEval("class A { int x; public String toString() { return \"new\"; } }",
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(midA, VALID, OVERWRITTEN, false, MAIN_SNIPPET),
+                ste(v, VALID, VALID, true, MAIN_SNIPPET));
+        assertEval("a;", "null");
+        assertActiveKeys();
+    }
+
+    private <T extends Snippet> void identityMatch(Stream<T> got, T expected) {
+        Iterator<T> it = got.iterator();
+        assertTrue(it.hasNext(), "expected exactly one");
+        assertTrue(expected == it.next(), "Identity must not change");
+        assertFalse(it.hasNext(), "expected exactly one");
+    }
+
+    public void testReplaceVarToMethod() {
+        Snippet x = varKey(assertEval("int x;"));
+        MethodSnippet musn = methodKey(assertEval("double mu() { return x * 4; }"));
+        assertEval("x == 0;", "true");
+        assertEval("mu() == 0.0;", "true");
+        assertEval("double x = 2.5;",
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(x, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        identityMatch(getState().methods(), musn);
+        assertEval("x == 2.5;", "true");
+        assertEval("mu() == 10.0;", "true");  // Auto redefine
+        assertActiveKeys();
+    }
+
+    public void testReplaceMethodToMethod() {
+        Snippet a = methodKey(assertEval("double a() { return 2; }"));
+        Snippet b = methodKey(assertEval("double b() { return a() * 10; }"));
+        assertEval("double c() { return b() * 3; }");
+        assertEval("double d() { return c() + 1000; }");
+        assertEval("d();", "1060.0");
+        assertEval("int a() { return 5; }",
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(a, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        assertEval("d();", "1150.0");
+        assertActiveKeys();
+    }
+
+    public void testReplaceClassToMethod() {
+        Snippet c = classKey(assertEval("class C { int f() { return 7; } }"));
+        Snippet m = methodKey(assertEval("int m() { return new C().f(); }"));
+        assertEval("m();", "7");
+        assertEval("class C { int x = 99; int f() { return x; } }",
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(c, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        assertEval("m();", "99");
+        assertActiveKeys();
+    }
+
+    public void testReplaceVarToClass() {
+        Snippet x = varKey(assertEval("int x;"));
+        TypeDeclSnippet c = classKey(assertEval("class A { double a = 4 * x; }"));
+        assertEval("x == 0;", "true");
+        assertEval("new A().a == 0.0;", "true");
+        assertEval("double x = 2.5;",
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(x, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        identityMatch(getState().types(), c);
+        assertEval("x == 2.5;", "true");
+        assertEval("new A().a == 10.0;", "true");
+        assertActiveKeys();
+    }
+
+    public void testReplaceMethodToClass() {
+        Snippet x = methodKey(assertEval("int x() { return 0; }"));
+        TypeDeclSnippet c = classKey(assertEval("class A { double a = 4 * x(); }"));
+        assertEval("x() == 0;", "true");
+        assertEval("new A().a == 0.0;", "true");
+        assertEval("double x() { return 2.5; }",
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(x, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        assertEval("x();", "2.5");
+        identityMatch(getState().types(), c);
+        assertEval("x() == 2.5;", "true");
+        assertEval("new A().a == 10.0;", "true");
+        assertActiveKeys();
+    }
+
+    public void testReplaceClassToClass() {
+        TypeDeclSnippet a = classKey(assertEval("class A {}"));
+        assertTypeDeclSnippet(a, "A", VALID, CLASS_SUBKIND, 0, 0);
+        TypeDeclSnippet b = classKey(assertEval("class B extends A {}"));
+        TypeDeclSnippet c = classKey(assertEval("class C extends B {}"));
+        TypeDeclSnippet d = classKey(assertEval("class D extends C {}"));
+        assertEval("class A { int x; public String toString() { return \"NEW\"; } }",
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(a, VALID, OVERWRITTEN, false, MAIN_SNIPPET),
+                ste(b, VALID, VALID, true, MAIN_SNIPPET),
+                ste(c, VALID, VALID, true, b),
+                ste(d, VALID, VALID, true, c));
+        assertTypeDeclSnippet(b, "B", VALID, CLASS_SUBKIND, 0, 0);
+        assertTypeDeclSnippet(c, "C", VALID, CLASS_SUBKIND, 0, 0);
+        assertTypeDeclSnippet(d, "D", VALID, CLASS_SUBKIND, 0, 0);
+        assertEval("new D();", "NEW");
+        assertActiveKeys();
+    }
+
+    public void testOverwriteReplaceMethod() {
+        MethodSnippet k1 = methodKey(assertEval("String m(Integer i) { return i.toString(); }"));
+        MethodSnippet k2 = methodKey(assertEval("String m(java.lang.Integer i) { return \"java.lang.\" + i.toString(); }",
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(k1, VALID, OVERWRITTEN, false, MAIN_SNIPPET)));
+        assertMethodDeclSnippet(k1, "m", "(Integer)String", OVERWRITTEN, 0, 0);
+        assertEval("m(6);", "\"java.lang.6\"");
+        assertEval("String m(Integer i) { return i.toString(); }",
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(k2, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        assertMethodDeclSnippet(k2, "m", "(java.lang.Integer)String", OVERWRITTEN, 0, 0);
+        assertEval("m(6);", "\"6\"");
+        assertActiveKeys();
+    }
+
+    public void testImportDeclare() {
+        Snippet singleImport = importKey(assertEval("import java.util.List;", added(VALID)));
+        Snippet importOnDemand = importKey(assertEval("import java.util.*;", added(VALID)));
+        Snippet singleStaticImport = importKey(assertEval("import static java.lang.Math.abs;", added(VALID)));
+        Snippet staticImportOnDemand = importKey(assertEval("import static java.lang.Math.*;", added(VALID)));
+        assertEval("import java.util.List; //again",
+                ste(MAIN_SNIPPET, VALID, VALID, false, null),
+                ste(singleImport, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        assertEval("import java.util.*; //again",
+                ste(MAIN_SNIPPET, VALID, VALID, false, null),
+                ste(importOnDemand, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        assertEval("import static java.lang.Math.abs; //again",
+                ste(MAIN_SNIPPET, VALID, VALID, false, null),
+                ste(singleStaticImport, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        assertEval("import static java.lang.Math.*; //again",
+                ste(MAIN_SNIPPET, VALID, VALID, false, null),
+                ste(staticImportOnDemand, VALID, OVERWRITTEN, false, MAIN_SNIPPET));
+        assertActiveKeys();
+    }
+
+    @Test(enabled = false) // TODO 8129420
+    public void testLocalClassEvolve() {
+        Snippet j = methodKey(assertEval("Object j() { return null; }", added(VALID)));
+        assertEval("Object j() { class B {}; return null; }",
+                ste(MAIN_SNIPPET, VALID, VALID, false, null));
+        assertEval("Object j() { class B {}; return new B(); }",
+                ste(MAIN_SNIPPET, VALID, VALID, false, null));
+        assertEval("j().getClass().getSimpleName();", "\"B\"");
+        assertEval("Object j() { class B { int p; public String toString() { return \"Yep\";} }; return new B(); }",
+                ste(MAIN_SNIPPET, VALID, VALID, false, null));
+        assertEval("j().getClass().getSimpleName();", "\"B\"");
+        assertEval("j();", "Yep");
+    }
+
+    public void testReplaceCausesMethodReferenceError() {
+        Snippet l = classKey(assertEval("interface Logger { public void log(String message); }", added(VALID)));
+        Snippet v = varKey(assertEval("Logger l = System.out::println;", added(VALID)));
+        assertEval("interface Logger { public boolean accept(String message);  }",
+                DiagCheck.DIAG_OK,
+                DiagCheck.DIAG_ERROR,
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(l, VALID, OVERWRITTEN, false, MAIN_SNIPPET),
+                ste(v, VALID, RECOVERABLE_NOT_DEFINED, true, MAIN_SNIPPET));
+    }
+
+    public void testReplaceCausesClassCompilationError() {
+        Snippet l = classKey(assertEval("interface L { }", added(VALID)));
+        Snippet c = classKey(assertEval("class C implements L { }", added(VALID)));
+        assertEval("interface L { void m(); }",
+                DiagCheck.DIAG_OK,
+                DiagCheck.DIAG_ERROR,
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(l, VALID, OVERWRITTEN, false, MAIN_SNIPPET),
+                ste(c, VALID, RECOVERABLE_NOT_DEFINED, true, MAIN_SNIPPET));
+    }
+
+    public void testOverwriteNoUpdate() {
+        String xsi = "int x = 5;";
+        String xsd = "double x = 3.14159;";
+        VarSnippet xi = varKey(assertEval(xsi, added(VALID)));
+        String ms1 = "double m(Integer i) { return i + x; }";
+        String ms2 = "double m(java.lang.Integer i) { return i + x; }";
+        MethodSnippet k1 = methodKey(assertEval(ms1, added(VALID)));
+        VarSnippet xd = varKey(assertEval(xsd,
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(xi, VALID, OVERWRITTEN, false, MAIN_SNIPPET)));
+        MethodSnippet k2 = methodKey(assertEval(ms2,
+                ste(MAIN_SNIPPET, VALID, VALID, true, null), //TODO: technically, should be false
+                ste(k1, VALID, OVERWRITTEN, false, MAIN_SNIPPET)));
+        VarSnippet xi2 = varKey(assertEval(xsi,
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(xd, VALID, OVERWRITTEN, false, MAIN_SNIPPET)));
+        varKey(assertEval(xsd,
+                ste(MAIN_SNIPPET, VALID, VALID, true, null),
+                ste(xi2, VALID, OVERWRITTEN, false, MAIN_SNIPPET)));
+    }
+}