Add test_inject_module

coolreader18 · coolreader18 · commit 9925d5b3bc0b · 2020-04-03T12:34:00.000-05:00
diff --git a/wasm/tests/test_demo.py b/wasm/tests/test_demo.py
@@ -1,8 +1,6 @@
 import time
 import sys
 
-from selenium import webdriver
-from selenium.webdriver.firefox.options import Options
 import pytest
 
 RUN_CODE_TEMPLATE = """
diff --git a/wasm/tests/test_inject_module.py b/wasm/tests/test_inject_module.py
@@ -0,0 +1,19 @@
+def test_inject_private(wdriver):
+    assert wdriver.execute_script(
+        """
+const vm = rp.vmStore.init("vm")
+vm.injectModule(
+"mod",
+`
+__all__ = ['get_thing']
+def get_thing(): return __thing()
+`,
+{ __thing: () => 1 },
+true
+)
+return vm.execSingle(
+    `import mod; mod.get_thing() == 1 and "__thing" not in dir(mod)`
+);
+        """
+    )
+
