/* * 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 void identityMatch(Stream got, T expected) { Iterator 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))); } }