Fix typing on tests

stevej2608 · Nov 30, 2024 · ba2a7e2 · ba2a7e2
1 parent 341ad50
commit ba2a7e2
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 11 deletions.
diff --git a/.github/workflows/workflow-simple-test.yml b/.github/workflows/workflow-simple-test.yml
diff --git a/tests/test_custom_dynamic_context.py b/tests/test_custom_dynamic_context.py
@@ -5,6 +5,8 @@
 from reactpy_utils import CustomDynamicContextModel, create_dynamic_context
 from reactpy_utils.types import EventArgs
 
+from .tooling import page_stable
+
 
 class CurrentUserState(CustomDynamicContextModel):
     def __init__(self, user_name: str, password: str, dark_mode: bool = True):
@@ -45,6 +47,7 @@ def TestApp():
         return AppContext(Child(), value=(state, set_state))
 
     await display.show(TestApp)
+    await page_stable(display.page)
 
     assert test_app_render_count == 1
     assert child_render_count == 1

diff --git a/tests/test_use_query.py b/tests/test_use_query.py
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, cast
+from typing import TYPE_CHECKING, cast, Union
 
 import pytest
 from pydantic import BaseModel
@@ -59,8 +59,8 @@ def sample():
 @pytest.mark.anyio
 async def test_search_params(display: DisplayFixture):
     class MyParams(BaseModel):
-        hello: str | None = None
-        thing: list[int] | None = None
+        hello: Union[str, None] = None
+        thing: Union[list[int], None] = None
 
     expected_query: MyParams = MyParams()