/*

 * Copyright (c) 2010, 2013, 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.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * 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.

 */

package jdk.nashorn.api.scripting;

import static org.testng.Assert.assertEquals;

import static org.testng.Assert.assertFalse;

import static org.testng.Assert.assertNotNull;

import static org.testng.Assert.assertTrue;

import static org.testng.Assert.fail;

import java.io.StringReader;

import java.io.StringWriter;

import java.lang.reflect.Method;

import java.util.HashMap;

import java.util.Map;

import java.util.Objects;

import java.util.concurrent.Callable;

import javax.script.Bindings;

import javax.script.Compilable;

import javax.script.CompiledScript;

import javax.script.Invocable;

import javax.script.ScriptContext;

import javax.script.ScriptEngine;

import javax.script.ScriptEngineFactory;

import javax.script.ScriptEngineManager;

import javax.script.ScriptException;

import javax.script.SimpleScriptContext;

import org.testng.Assert;

import org.testng.annotations.Test;

/**

 * Tests for JSR-223 script engine for Nashorn.

 *

 * @test

 * @build jdk.nashorn.api.scripting.Window jdk.nashorn.api.scripting.WindowEventHandler jdk.nashorn.api.scripting.VariableArityTestInterface jdk.nashorn.api.scripting.ScriptEngineTest

 * @run testng jdk.nashorn.api.scripting.ScriptEngineTest

 */

public class ScriptEngineTest {

    private void log(String msg) {

        org.testng.Reporter.log(msg, true);

    }

    @Test

    public void argumentsTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        String[] args = new String[] { "hello", "world" };

        try {

            e.put("arguments", args);

            Object arg0 = e.eval("arguments[0]");

            Object arg1 = e.eval("arguments[1]");

            assertEquals(args[0], arg0);

            assertEquals(args[1], arg1);

        } catch (final Exception exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

    }

    @Test

    public void argumentsWithTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        String[] args = new String[] { "hello", "world" };

        try {

            e.put("arguments", args);

            Object arg0 = e.eval("var imports = new JavaImporter(java.io); " +

                    " with(imports) { arguments[0] }");

            Object arg1 = e.eval("var imports = new JavaImporter(java.util, java.io); " +

                    " with(imports) { arguments[1] }");

            assertEquals(args[0], arg0);

            assertEquals(args[1], arg1);

        } catch (final Exception exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

    }

    @Test

    public void argumentsEmptyTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        try {

            assertEquals(e.eval("arguments instanceof Array"), true);

            assertEquals(e.eval("arguments.length == 0"), true);

        } catch (final Exception exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

    }

    @Test

    public void factoryTests() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        assertNotNull(e);

        final ScriptEngineFactory fac = e.getFactory();

        assertEquals(fac.getLanguageName(), "ECMAScript");

        assertEquals(fac.getParameter(ScriptEngine.NAME), "javascript");

        assertEquals(fac.getLanguageVersion(), "ECMA - 262 Edition 5.1");

        assertEquals(fac.getEngineName(), "Oracle Nashorn");

        assertEquals(fac.getOutputStatement("context"), "print(context)");

        assertEquals(fac.getProgram("print('hello')", "print('world')"), "print('hello');print('world');");

        assertEquals(fac.getParameter(ScriptEngine.NAME), "javascript");

        boolean seenJS = false;

        for (String ext : fac.getExtensions()) {

            if (ext.equals("js")) {

                seenJS = true;

            }

        }

        assertEquals(seenJS, true);

        String str = fac.getMethodCallSyntax("obj", "foo", "x");

        assertEquals(str, "obj.foo(x)");

        boolean seenNashorn = false, seenJavaScript = false, seenECMAScript = false;

        for (String name : fac.getNames()) {

            switch (name) {

                case "nashorn": seenNashorn = true; break;

                case "javascript": seenJavaScript = true; break;

                case "ECMAScript": seenECMAScript = true; break;

            }

        }

        assertTrue(seenNashorn);

        assertTrue(seenJavaScript);

        assertTrue(seenECMAScript);

        boolean seenAppJS = false, seenAppECMA = false, seenTextJS = false, seenTextECMA = false;

        for (String mime : fac.getMimeTypes()) {

            switch (mime) {

                case "application/javascript": seenAppJS = true; break;

                case "application/ecmascript": seenAppECMA = true; break;

                case "text/javascript": seenTextJS = true; break;

                case "text/ecmascript": seenTextECMA = true; break;

            }

        }

        assertTrue(seenAppJS);

        assertTrue(seenAppECMA);

        assertTrue(seenTextJS);

        assertTrue(seenTextECMA);

    }

    @Test

    public void evalTests() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        e.put(ScriptEngine.FILENAME, "myfile.js");

        try {

            e.eval("print('hello')");

        } catch (final ScriptException se) {

            fail(se.getMessage());

        }

        try {

            e.eval("print('hello)");

            fail("script exception expected");

        } catch (final ScriptException se) {

            assertEquals(se.getLineNumber(), 1);

            assertEquals(se.getColumnNumber(), 13);

            assertEquals(se.getFileName(), "myfile.js");

            // se.printStackTrace();

        }

        try {

            Object obj = e.eval("34 + 41");

            assertTrue(34.0 + 41.0 == ((Number)obj).doubleValue());

            obj = e.eval("x = 5");

            assertTrue(5.0 == ((Number)obj).doubleValue());

        } catch (final ScriptException se) {

            se.printStackTrace();

            fail(se.getMessage());

        }

    }

    @Test

    public void compileTests() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        CompiledScript script = null;

        try {

            script = ((Compilable)e).compile("print('hello')");

        } catch (final ScriptException se) {

            fail(se.getMessage());

        }

        try {

            script.eval();

        } catch (final ScriptException | NullPointerException se) {

            se.printStackTrace();

            fail(se.getMessage());

        }

        // try to compile from a Reader

        try {

            script = ((Compilable)e).compile(new StringReader("print('world')"));

        } catch (final ScriptException se) {

            fail(se.getMessage());

        }

        try {

            script.eval();

        } catch (final ScriptException | NullPointerException se) {

            se.printStackTrace();

            fail(se.getMessage());

        }

    }

    @Test

    public void createBindingsTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        Bindings b = e.createBindings();

        b.put("foo", 42.0);

        Object res = null;

        try {

            res = e.eval("foo == 42.0", b);

        } catch (final ScriptException | NullPointerException se) {

            se.printStackTrace();

            fail(se.getMessage());

        }

        assertEquals(res, Boolean.TRUE);

    }

    @Test

    public void getInterfaceTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        final Invocable inv = (Invocable)e;

        // try to get interface from global functions

        try {

            e.eval("function run() { print('run'); };");

            final Runnable runnable = inv.getInterface(Runnable.class);

            runnable.run();

        } catch (final Exception exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

        // try interface on specific script object

        try {

            e.eval("var obj = { run: function() { print('run from obj'); } };");

            Object obj = e.get("obj");

            final Runnable runnable = inv.getInterface(obj, Runnable.class);

            runnable.run();

        } catch (final Exception exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

    }

    public interface Foo {

        public void bar();

    }

    public interface Foo2 extends Foo {

        public void bar2();

    }

    @Test

    public void getInterfaceMissingTest() {

        final ScriptEngineManager manager = new ScriptEngineManager();

        final ScriptEngine engine = manager.getEngineByName("nashorn");

        // don't define any function.

        try {

            engine.eval("");

        } catch (final Exception exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

        Runnable runnable = ((Invocable)engine).getInterface(Runnable.class);

        if (runnable != null) {

            fail("runnable is not null!");

        }

        // now define "run"

        try {

            engine.eval("function run() { print('this is run function'); }");

        } catch (final Exception exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

        runnable = ((Invocable)engine).getInterface(Runnable.class);

        // should not return null now!

        runnable.run();

        // define only one method of "Foo2"

        try {

            engine.eval("function bar() { print('bar function'); }");

        } catch (final Exception exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

        Foo2 foo2 = ((Invocable)engine).getInterface(Foo2.class);

        if (foo2 != null) {

            throw new RuntimeException("foo2 is not null!");

        }

        // now define other method of "Foo2"

        try {

            engine.eval("function bar2() { print('bar2 function'); }");

        } catch (final Exception exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

        foo2 = ((Invocable)engine).getInterface(Foo2.class);

        foo2.bar();

        foo2.bar2();

    }

    @Test

    /**

     * Try passing non-interface Class object for interface implementation.

     */

    public void getNonInterfaceGetInterfaceTest() {

        final ScriptEngineManager manager = new ScriptEngineManager();

        final ScriptEngine engine = manager.getEngineByName("nashorn");

        try {

            log(Objects.toString(((Invocable)engine).getInterface(Object.class)));

            fail("Should have thrown IllegalArgumentException");

        } catch (final Exception exp) {

            if (! (exp instanceof IllegalArgumentException)) {

                fail("IllegalArgumentException expected, got " + exp);

            }

        }

    }

    @Test

    /**

     * Check that we can get interface out of a script object even after

     * switching to use different ScriptContext.

     */

    public void getInterfaceDifferentContext() {

       ScriptEngineManager m = new ScriptEngineManager();

       ScriptEngine e = m.getEngineByName("nashorn");

       try {

           Object obj = e.eval("({ run: function() { } })");

           // change script context

           ScriptContext ctxt = new SimpleScriptContext();

           ctxt.setBindings(e.createBindings(), ScriptContext.ENGINE_SCOPE);

           e.setContext(ctxt);

           Runnable r = ((Invocable)e).getInterface(obj, Runnable.class);

           r.run();

       }catch (final Exception exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

       }

    }

    @Test

    /**

     * Check that getInterface on non-script object 'thiz' results in IllegalArgumentException.

     */

    public void getInterfaceNonScriptObjectThizTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        try {

            ((Invocable)e).getInterface(new Object(), Runnable.class);

            fail("should have thrown IllegalArgumentException");

        } catch (final Exception exp) {

            if (! (exp instanceof IllegalArgumentException)) {

                exp.printStackTrace();

                fail(exp.getMessage());

            }

        }

    }

    @Test

    /**

     * Check that getInterface on null 'thiz' results in IllegalArgumentException.

     */

    public void getInterfaceNullThizTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        try {

            ((Invocable)e).getInterface(null, Runnable.class);

            fail("should have thrown IllegalArgumentException");

        } catch (final Exception exp) {

            if (! (exp instanceof IllegalArgumentException)) {

                exp.printStackTrace();

                fail(exp.getMessage());

            }

        }

    }

    @Test

    /**

     * Check that calling getInterface on mirror created by another engine results in IllegalArgumentException.

     */

    public void getInterfaceMixEnginesTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine engine1 = m.getEngineByName("nashorn");

        final ScriptEngine engine2 = m.getEngineByName("nashorn");

        try {

            Object obj = engine1.eval("({ run: function() {} })");

            // pass object from engine1 to engine2 as 'thiz' for getInterface

            ((Invocable)engine2).getInterface(obj, Runnable.class);

            fail("should have thrown IllegalArgumentException");

        } catch (final Exception exp) {

            if (! (exp instanceof IllegalArgumentException)) {

                exp.printStackTrace();

                fail(exp.getMessage());

            }

        }

    }

    @Test

    public void accessGlobalTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        try {

            e.eval("var x = 'hello'");

            assertEquals(e.get("x"), "hello");

        } catch (final ScriptException exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

    }

    @Test

    public void exposeGlobalTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        try {

            e.put("y", "foo");

            e.eval("print(y)");

        } catch (final ScriptException exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

    }

    @Test

    public void putGlobalFunctionTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        e.put("callable", new Callable<String>() {

            @Override

            public String call() throws Exception {

                return "callable was called";

            }

        });

        try {

            e.eval("print(callable.call())");

        } catch (final ScriptException exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

    }

    @Test

    public void windowAlertTest() {

        final ScriptEngineManager m = new ScriptEngineManager();

        final ScriptEngine e = m.getEngineByName("nashorn");

        final Window window = new Window();

        try {

            e.put("window", window);

            e.eval("print(window.alert)");

            e.eval("window.alert('calling window.alert...')");

        } catch (final Exception exp) {

            exp.printStackTrace();

            fail(exp.getMessage());

        }

    }