more tests

Archmonger · Archmonger · commit 258194b3e40f · 2024-06-21T00:17:07.000-07:00
diff --git a/tests/test_app/pyscript/components/custom_root.py b/tests/test_app/pyscript/components/custom_root.py
@@ -0,0 +1,6 @@
+from reactpy import component, html
+
+
+@component
+def main():
+    return html.div({"id": "custom-root"}, "Component with a custom root name.")
diff --git a/tests/test_app/templates/pyscript.html b/tests/test_app/templates/pyscript.html
@@ -14,7 +14,9 @@
 <body>
     <h1>ReactPy PyScript Test Page</h1>
     <hr>
-    {% pyscript_component "./test_app/pyscript/components/hello_world.py" %}
+    {% pyscript_component "./test_app/pyscript/components/hello_world.py" initial="<div id='hello-world-loading'>Loading...</div>" %}
+    <hr>
+    {% pyscript_component "./test_app/pyscript/components/custom_root.py" root="main" %}
     <hr>
     {% pyscript_component "./test_app/pyscript/components/counter.py" %}
     <hr>
diff --git a/tests/test_app/tests/test_components.py b/tests/test_app/tests/test_components.py
@@ -683,7 +683,9 @@ def test_pyscript_components(self):
         new_page = self.browser.new_page()
         try:
             new_page.goto(f"{self.live_server_url}/pyscript/")
+            new_page.wait_for_selector("#hello-world-loading")
             new_page.wait_for_selector("#hello-world")
+            new_page.wait_for_selector("#custom-root")
 
             new_page.wait_for_selector("#counter")
             new_page.wait_for_selector("#counter pre[data-value='0']")
