make common test hooks module

Author: rmorshea

tests/test_core/test_layout.py

@@ -20,6 +20,7 @@
     capture_idom_logs,
 )
 from idom.utils import Ref
+from tests.tooling.hooks import use_toggle
 
 
 @pytest.fixture(autouse=True)
@@ -494,11 +495,6 @@ def SomeComponent():
     )
 
 
-def use_toggle(init=False):
-    state, set_state = idom.hooks.use_state(init)
-    return state, lambda: set_state(lambda old: not old)
-
-
 async def test_model_key_preserves_callback_identity_for_common_elements(caplog):
     called_good_trigger = idom.Ref(False)
     good_handler = StaticEventHandler()
@@ -813,18 +809,13 @@ def Child(state):
             await layout.render()
 
 
-def use_toggle():
-    state, set_state = idom.hooks.use_state(True)
-    return state, lambda: set_state(not state)
-
-
 async def test_elements_and_components_with_the_same_key_can_be_interchanged():
     set_toggle = idom.Ref()
     effects = []
 
     @idom.component
     def Root():
-        toggle, set_toggle.current = use_toggle()
+        toggle, set_toggle.current = use_toggle(True)
         if toggle:
             return SomeComponent("x")
         else:

tests/test_html.py

@@ -3,11 +3,7 @@
 from idom import component, config, html, use_state
 from idom.testing import DisplayFixture, poll
 from idom.utils import Ref
-
-
-def use_toggle(initial=True):
-    state, set_state = use_state(initial)
-    return state, lambda: set_state(not state)
+from tests.tooling.hooks import use_toggle
 
 
 def use_counter(initial_value):
@@ -20,7 +16,7 @@ async def test_script_mount_unmount(display: DisplayFixture):
 
     @component
     def Root():
-        is_mounted, toggle_is_mounted.current = use_toggle()
+        is_mounted, toggle_is_mounted.current = use_toggle(True)
         return html.div(
             html.div({"id": "mount-state", "data-value": False}),
             HasScript() if is_mounted else html.div(),

tests/tooling/hooks.py

@@ -0,0 +1,6 @@
+from idom import use_state
+
+
+def use_toggle(init=False):
+    state, set_state = use_state(init)
+    return state, lambda: set_state(lambda old: not old)