diff --git a/src/idom/core/hooks.py b/src/idom/core/hooks.py index d0717162e..c19b9e547 100644 --- a/src/idom/core/hooks.py +++ b/src/idom/core/hooks.py @@ -690,6 +690,7 @@ def affect_layout_did_render(self) -> None: if self._schedule_render_later: self._schedule_render() + self._schedule_render_later = False def affect_component_will_unmount(self) -> None: """The component is about to be removed from the layout""" diff --git a/src/idom/testing.py b/src/idom/testing.py index f9581b45f..ad7a9bac1 100644 --- a/src/idom/testing.py +++ b/src/idom/testing.py @@ -25,7 +25,9 @@ from uuid import uuid4 from weakref import ref -from selenium.webdriver import Chrome +from selenium.webdriver.chrome.options import Options as ChromeOptions +from selenium.webdriver.chrome.webdriver import WebDriver as Chrome +from selenium.webdriver.common.options import BaseOptions from selenium.webdriver.remote.webdriver import WebDriver from idom.config import IDOM_WEB_MODULES_DIR @@ -47,7 +49,7 @@ def create_simple_selenium_web_driver( driver_type: Type[WebDriver] = Chrome, - driver_options: Optional[Any] = None, + driver_options: BaseOptions = ChromeOptions(), implicit_wait_timeout: float = 10.0, page_load_timeout: float = 5.0, window_size: Tuple[int, int] = (1080, 800),