Add selenium test for wasm

palaviv · palaviv · commit 3d95b70e0f28 · 2019-03-15T09:13:55.000+02:00
diff --git a/tests/wasm/test_demo.py b/tests/wasm/test_demo.py
@@ -0,0 +1,40 @@
+import time
+
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+import pytest
+
+RUN_CODE_TEMPLATE = """
+var output = "";
+save_output = function(text) {{
+	output += text
+}};
+var vm = window.rp.vmStore.init('test_vm');
+vm.setStdout(save_output);
+vm.exec('{}');
+vm.destroy();
+return output;
+"""
+
+@pytest.fixture(scope="module")
+def driver(request):
+	options = Options()
+	options.headless = True
+	options.add_argument('--disable-gpu')
+	driver = webdriver.Chrome(options=options)
+	driver.get("http://localhost:8080")
+	assert "RustPython" in driver.title
+	time.sleep(5)
+	yield driver
+	driver.close()	
+    
+
+@pytest.mark.parametrize("script, output",
+	[
+		("print(5)", "5"),
+		("a=5;b=4;print(a+b)", "9")
+	]
+)
+def test_demo(driver, script, output):
+	script = RUN_CODE_TEMPLATE.format(script)
+	assert driver.execute_script(script).strip() == output
