test_Context.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import unittest

import STPyV8


class TestContext(unittest.TestCase):
    def testEval(self):
        with STPyV8.JSContext() as context:
            self.assertEqual(2, context.eval("1+1"))
            self.assertEqual('Hello world', context.eval("'Hello ' + 'world'"))

    def testMultiNamespace(self):
        self.assertTrue(not bool(STPyV8.JSContext.inContext))
        self.assertTrue(not bool(STPyV8.JSContext.entered))

        class Global:
            name = "global"

        g = Global()

        with STPyV8.JSContext(g) as ctxt:
            self.assertTrue(ctxt)
            self.assertTrue(bool(STPyV8.JSContext.inContext))
            self.assertEqual(g.name, str(STPyV8.JSContext.entered.locals.name))

            class Local:
                name = "local"

            l = Local()

            with STPyV8.JSContext(l):
                self.assertTrue(bool(STPyV8.JSContext.inContext))
                self.assertEqual(l.name, str(STPyV8.JSContext.entered.locals.name))

            self.assertTrue(bool(STPyV8.JSContext.inContext))
            self.assertEqual(g.name, str(STPyV8.JSContext.current.locals.name))

        self.assertTrue(not bool(STPyV8.JSContext.entered))
        self.assertTrue(not bool(STPyV8.JSContext.inContext))

    def testMultiContext(self):
        with STPyV8.JSContext() as ctxt0:
            ctxt0.securityToken = "password"

            global0 = ctxt0.locals
            global0.custom = 1234

            self.assertEqual(1234, int(global0.custom))

            with STPyV8.JSContext() as ctxt1:
                ctxt1.securityToken = ctxt0.securityToken

                global1 = ctxt1.locals
                global1.custom = 1234

                with ctxt0:
                    self.assertEqual(1234, int(global0.custom))

                self.assertEqual(1234, int(global1.custom))

    def testPromiseResolved(self):
        with STPyV8.JSContext() as ctxt:
            ctxt.eval("""
                var message;
                let done = true;

                const isItDoneYet = new Promise((resolve, reject) => {
                    if (done) {
                        const workDone = 'Here is the thing I built'
                        resolve(workDone)
                    } else {
                        const why = 'Still working on something else'
                        reject(why)
                    }
                })

                const checkIfItsDone = () => {
                    isItDoneYet.then(ok => {
                        message = ok;
                    }).catch(err => {
                        message = err;
                    })
                }

                checkIfItsDone()
            """)

            self.assertEqual('Here is the thing I built', ctxt.locals.message)

    def testPromiseRejected(self):
        with STPyV8.JSContext() as ctxt:
            ctxt.eval("""
                var message;
                let done = false;

                const isItDoneYet = new Promise((resolve, reject) => {
                    if (done) {
                        const workDone = 'Here is the thing I built'
                        resolve(workDone)
                    } else {
                        const why = 'Still working on something else'
                        reject(why)
                    }
                })

                const checkIfItsDone = () => {
                    isItDoneYet.then(ok => {
                        message = ok;
                    }).catch(err => {
                        message = err;
                    })
                }

                checkIfItsDone()
            """)

            self.assertEqual('Still working on something else', ctxt.locals.message)

    def testSecurityChecks(self):
        with STPyV8.JSContext() as env1:
            env1.securityToken = "foo"

            # Create a function in env1.
            env1.eval("spy = function(){return spy;}")

            spy = env1.locals.spy

            self.assertTrue(isinstance(spy, STPyV8.JSFunction))

            # Create another function accessing global objects.
            env1.eval("spy2 = function(){return 123;}")

            spy2 = env1.locals.spy2

            self.assertTrue(isinstance(spy2, STPyV8.JSFunction))

            # Switch to env2 in the same domain and invoke spy on env2.
            env2 = STPyV8.JSContext()
            env2.securityToken = "foo"

            with env2:
                result = spy.apply(env2.locals)
                self.assertTrue(isinstance(result, STPyV8.JSFunction))

            env2.securityToken = "bar"

            # FIXME
            # Call cross_domain_call, it should throw an exception
            # with env2:
            #    self.assertRaises(STPyV8.JSError, spy2.apply, env2.locals)