attila@963: /*
attila@963:  * Copyright (c) 2014, Oracle and/or its affiliates. All rights reserved.
attila@963:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
attila@963:  *
attila@963:  * This code is free software; you can redistribute it and/or modify it
attila@963:  * under the terms of the GNU General Public License version 2 only, as
attila@963:  * published by the Free Software Foundation.
attila@963:  *
attila@963:  * This code is distributed in the hope that it will be useful, but WITHOUT
attila@963:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
attila@963:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
attila@963:  * version 2 for more details (a copy is included in the LICENSE file that
attila@963:  * accompanied this code).
attila@963:  *
attila@963:  * You should have received a copy of the GNU General Public License version
attila@963:  * 2 along with this work; if not, write to the Free Software Foundation,
attila@963:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
attila@963:  *
attila@963:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
attila@963:  * or visit www.oracle.com if you need additional information or have any
attila@963:  * questions.
attila@963:  */
attila@963: 
attila@963: /**
attila@963:  * JDK-8055107: Extension directives to turn on callsite profiling, tracing, AST print and other debug features locally
attila@963:  *
attila@963:  * @test
attila@963:  * @option -Dnashorn.debug=true
attila@963:  * @option -scripting
attila@963:  * @run
attila@963:  * @fork
attila@963:  */
attila@963: 
attila@963: function runScriptEngine(code) {
attila@963:     var imports = new JavaImporter(
attila@963:         java.io, java.lang, java.util, javax.script);
attila@963: 
attila@963:     with(imports) {
attila@963:         var m = new ScriptEngineManager();
attila@963:         // get current System.err
attila@963:         var oldErr = System.err;
attila@963:         var baos = new ByteArrayOutputStream();
attila@963:         var newErr = new PrintStream(baos);
attila@963:         try {
attila@963:             // set new standard err
attila@963:             System.setErr(newErr);
attila@963:             var engine = m.getEngineByName("nashorn");
attila@963:             engine.eval(code);
attila@963:             newErr.flush();
attila@963:             return new java.lang.String(baos.toByteArray());
attila@963:         } finally {
attila@963:             // restore System.err to old value
attila@963:             System.setErr(oldErr);
attila@963:         }
attila@963:     }
attila@963: }
attila@963: 
attila@963: // nashorn callsite trace enterexit
attila@963: var str = runScriptEngine(<<CODE
attila@963: function func() {
attila@963:    "nashorn callsite trace enterexit";
attila@963:    k();
attila@963: }
attila@963: 
attila@963: function k() {
attila@963:     var x = "hello";
attila@963: }
attila@963: 
attila@963: func();
attila@963: CODE);
attila@963: 
attila@963: if (!str.contains(" ENTER ")) {
attila@963:     fail("expected 'ENTER' in trace mode output");
attila@963: }
attila@963: 
attila@963: if (!str.contains(" EXIT ")) {
attila@963:     fail("expected 'EXIT' in trace mode output");
attila@963: }
attila@963: 
attila@963: // nashorn callsite trace objects
attila@963: var str = runScriptEngine(<<CODE
attila@963: "nashorn callsite trace objects";
attila@963: function func(x) {
attila@963: }
attila@963: 
attila@963: func("hello");
attila@963: CODE);
attila@963: 
attila@963: if (!str.contains(" ENTER ")) {
attila@963:     fail("expected 'ENTER' in trace mode output");
attila@963: }
attila@963: 
attila@963: if (!str.contains(" EXIT ")) {
attila@963:     fail("expected 'EXIT' in trace mode output");
attila@963: }
attila@963: 
attila@963: if (!str.contains("hello")) {
attila@963:     fail("expected argument to be traced in trace objects mode");
attila@963: }
attila@963: 
attila@963: // nashorn callsite trace misses
attila@963: str = runScriptEngine(<<CODE
attila@963: function f() {
attila@963:    "nashorn callsite trace misses";
attila@963:    k();
attila@963: }
attila@963: 
attila@963: function k() {}
attila@963: f();
attila@963: CODE);
attila@963: 
attila@963: if (!str.contains(" MISS ")) {
attila@963:     fail("expected callsite MISS trace messages");
attila@963: }
attila@963: 
attila@963: // nashorn print lower ast
attila@963: str = runScriptEngine(<<CODE
attila@963: function foo() {
attila@963:     "nashorn print lower ast";
attila@963:     var x = 'hello';
attila@963: }
attila@963: foo();
attila@963: CODE);
attila@963: 
attila@963: if (!str.contains("Lower AST for: 'foo'") ||
attila@963:     !str.contains("nashorn print lower ast")) {
attila@963:     fail("expected Lower AST to be printed for 'foo'");
attila@963: }
attila@963: 
attila@963: // nashorn print ast
attila@963: str = runScriptEngine(<<CODE
attila@963: function foo() {
attila@963:   "nashorn print ast";
attila@963: }
attila@963: CODE);
attila@963: if (!str.contains("[function ") ||
attila@963:     !str.contains("nashorn print ast")) {
attila@963:     fail("expected AST to be printed");
attila@963: }
attila@963: 
attila@963: // nashorn print symbols
attila@963: str = runScriptEngine(<<CODE
attila@963: function bar(a) {
attila@963:     "nashorn print symbols";
attila@963:     if (a) print(a);
attila@963: }
attila@963: 
attila@963: bar();
attila@963: CODE)
attila@963: 
attila@963: if (!str.contains("[BLOCK in 'Function bar']")) {
attila@963:     fail("expected symbols to be printed for 'bar'");
attila@963: }
attila@963: 
attila@963: // nashorn print parse
attila@963: str = runScriptEngine(<<CODE
attila@963: "nashorn print parse";
attila@963: 
attila@963: function func() {}
attila@963: CODE);
attila@963: 
attila@963: if (!str.contains("function func") ||
attila@963:     !str.contains("nashorn print parse")) {
attila@963:     fail("expected nashorn print parse output");
attila@963: }
attila@963: 
attila@963: // nashorn print lower parse
attila@963: str = runScriptEngine(<<CODE
attila@963: "nashorn print lower parse";
attila@963: 
attila@963: function func() {}
attila@963: 
attila@963: func()
attila@963: CODE);
attila@963: 
attila@963: if (!str.contains("function {U%}func") ||
attila@963:     !str.contains("nashorn print lower parse")) {
attila@963:     fail("expected nashorn print lower parse output");
attila@963: }