/*

 * Copyright (c) 2014, 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.

 */

/**

 * JDK-8055107: Extension directives to turn on callsite profiling, tracing, AST print and other debug features locally

 *

 * @test

 * @option -Dnashorn.debug=true

 * @option -scripting

 * @run

 * @fork

 */

function runScriptEngine(code) {

    var imports = new JavaImporter(

        java.io, java.lang, java.util, javax.script);

    with(imports) {

        var m = new ScriptEngineManager();

        // get current System.err

        var oldErr = System.err;

        var baos = new ByteArrayOutputStream();

        var newErr = new PrintStream(baos);

        try {

            // set new standard err

            System.setErr(newErr);

            var engine = m.getEngineByName("nashorn");

            engine.eval(code);

            newErr.flush();

            return new java.lang.String(baos.toByteArray());

        } finally {

            // restore System.err to old value

            System.setErr(oldErr);

        }

    }

}

// nashorn callsite trace enterexit

var str = runScriptEngine(<<CODE

function func() {

   "nashorn callsite trace enterexit";

   k();

}

function k() {

    var x = "hello";

}

func();

CODE);

if (!str.contains(" ENTER ")) {

    fail("expected 'ENTER' in trace mode output");

}

if (!str.contains(" EXIT ")) {

    fail("expected 'EXIT' in trace mode output");

}

// nashorn callsite trace objects

var str = runScriptEngine(<<CODE

"nashorn callsite trace objects";

function func(x) {

}

func("hello");

CODE);

if (!str.contains(" ENTER ")) {

    fail("expected 'ENTER' in trace mode output");

}

if (!str.contains(" EXIT ")) {

    fail("expected 'EXIT' in trace mode output");

}

if (!str.contains("hello")) {

    fail("expected argument to be traced in trace objects mode");

}

// nashorn callsite trace misses

str = runScriptEngine(<<CODE

function f() {

   "nashorn callsite trace misses";

   k();

}

function k() {}

f();

CODE);

if (!str.contains(" MISS ")) {

    fail("expected callsite MISS trace messages");

}

// nashorn print lower ast

str = runScriptEngine(<<CODE

function foo() {

    "nashorn print lower ast";

    var x = 'hello';

}

foo();

CODE);

if (!str.contains("Lower AST for: 'foo'") ||

    !str.contains("nashorn print lower ast")) {

    fail("expected Lower AST to be printed for 'foo'");

}

// nashorn print ast

str = runScriptEngine(<<CODE

function foo() {

  "nashorn print ast";

}

CODE);

if (!str.contains("[function ") ||

    !str.contains("nashorn print ast")) {

    fail("expected AST to be printed");

}

// nashorn print symbols

str = runScriptEngine(<<CODE

function bar(a) {

    "nashorn print symbols";

    if (a) print(a);

}

bar();

CODE)

if (!str.contains("[BLOCK in 'Function bar']")) {

    fail("expected symbols to be printed for 'bar'");

}

// nashorn print parse

str = runScriptEngine(<<CODE

"nashorn print parse";

function func() {}

CODE);

if (!str.contains("function func") ||

    !str.contains("nashorn print parse")) {

    fail("expected nashorn print parse output");

}

// nashorn print lower parse

str = runScriptEngine(<<CODE

"nashorn print lower parse";

function func() {}

func()

CODE);

if (!str.contains("function {U%}func") ||

    !str.contains("nashorn print lower parse")) {

    fail("expected nashorn print lower parse output");

}