#!/usr/bin/env python3

# -*- coding: ascii -*-

import sys

import os

import doctest

import six

import execjs

if sys.version_info < (2, 7):

import unittest2 as unittest

else:

import unittest

class RuntimeTestBase:

def test_context_call(self):

context = self.runtime.compile("id = function(v) { return v; }")

self.assertEqual("bar", context.call("id", "bar"))

def test_nested_context_call(self):

context = self.runtime.compile("a = {}; a.b = {}; a.b.id = function(v) { return v; }")

self.assertEqual("bar", context.call("a.b.id", "bar"))

def test_context_call_missing_function(self):

context = self.runtime.compile("")

with self.assertRaises(execjs.ProgramError):

context.call("missing")

def test_exec(self):

self.assertIsNone(self.runtime.exec_("1"))

self.assertIsNone(self.runtime.exec_("return"))

self.assertIsNone(self.runtime.exec_("return null"))

self.assertIsNone(self.runtime.exec_("return function() {}"))

self.assertIs(0, self.runtime.exec_("return 0"))

self.assertIs(True, self.runtime.exec_("return true"))

self.assertEqual("hello", self.runtime.exec_("return 'hello'"))

self.assertEqual([1, 2], self.runtime.exec_("return [1, 2]"))

self.assertEqual({"a": 1, "b": 2}, self.runtime.exec_("return {a:1,b:2}"))

self.assertEqual("\u3042", self.runtime.exec_('return "\u3042"')) # unicode char

self.assertEqual("\u3042", self.runtime.exec_(r'return "\u3042"')) # unicode char by escape sequence

self.assertEqual("\\", self.runtime.exec_('return "\\\\"'))

def test_eval(self):

self.assertIsNone(self.runtime.eval(""))

self.assertIsNone(self.runtime.eval(" "))

self.assertIsNone(self.runtime.eval("null"))

self.assertIsNone(self.runtime.eval("function(){}"))

self.assertIs(0, self.runtime.eval("0"))

self.assertIs(True, self.runtime.eval("true"))

self.assertEqual([1, 2], self.runtime.eval("[1, 2]"))

self.assertEqual([1, None], self.runtime.eval("[1, function() {}]"))

self.assertEqual("hello", self.runtime.eval("'hello'"))

self.assertEqual(["red", "yellow", "blue"], self.runtime.eval("'red yellow blue'.split(' ')"))

self.assertEqual({"a": 1, "b": 2}, self.runtime.eval("{a:1, b:2}"))

self.assertEqual({"a": True}, self.runtime.eval("{a:true,b:function (){}}"))

self.assertEqual("\u3042", self.runtime.eval('"\u3042"'))

self.assertEqual("\u3042", self.runtime.eval(r'"\u3042"'))

self.assertEqual(r"\\", self.runtime.eval(r'"\\\\"'))

def test_compile(self):

context = self.runtime.compile("foo = function() { return \"bar\"; }")

self.assertEqual("bar", context.exec_("return foo()"))

self.assertEqual("bar", context.eval("foo()"))

self.assertEqual("bar", context.call("foo"))

def test_this_is_global_scope(self):

self.assertIs(True, self.runtime.eval("this === (function() {return this})()"))

self.assertIs(True, self.runtime.exec_("return this === (function() {return this})()"))

def test_compile_large_scripts(self):

body = "var foo = 'bar';\n" * (10 ** 4)

code = "function foo() {\n" + body + "\n};\nreturn true"

self.assertTrue(self.runtime.exec_(code))

def test_syntax_error(self):

with self.assertRaises(execjs.RuntimeError):

self.runtime.exec_(")")

def test_thrown_exception(self):

with self.assertRaises(execjs.ProgramError):

self.runtime.exec_("throw 'hello'")

def test_broken_substitutions(self):

s = '#{source}#{encoded_source}#{json2_source}'

self.assertEqual(s, self.runtime.eval('"' + s + '"'))

class DefaultRuntimeTest(unittest.TestCase, RuntimeTestBase):

def setUp(self):

self.runtime = execjs

for name, runtime in list(execjs.runtimes().items()):

if not runtime.is_available():

continue

class_name = name.capitalize() + "RuntimeTest"

def f(runtime=runtime):

class RuntimeTest(unittest.TestCase, RuntimeTestBase):

def setUp(self):

self.runtime = runtime

RuntimeTest.__name__ = str(class_name) # 2.x compatibility

return RuntimeTest

exec("{class_name} = f()".format(class_name=class_name))

class CommonTest(unittest.TestCase):

def test_empty_path_environ(self):

"""

Some version of passenger-nginx set PATH empty.

"""

orig_path = os.environ['PATH']

try:

del os.environ['PATH']

ctx = execjs.compile("""

function add(x, y) {

return x + y;

}

""")

ctx.call("add", 1, 2)

finally:

os.environ['PATH'] = orig_path

def test_runtime_availability(self):

r = execjs.ExternalRuntime("fail", ["nonexistent"], "")

self.assertFalse(r.is_available())

r = execjs.ExternalRuntime("success", ["python"], "")

self.assertTrue(r.is_available())

def test_attributes_export(self):

for name in execjs.__all__:

self.assertTrue(hasattr(execjs, name), "{} is not defined".format(name))

def load_tests(loader, tests, ignore):

if six.PY3:

tests.addTests(doctest.DocTestSuite(execjs))

return tests

if __name__ == "__main__":

unittest.main()