remove simulated pre-hydrated states

Author: masenf

tests/units/test_app.py

@@ -1771,9 +1771,6 @@ async def test_dynamic_route_var_route_change_completed_on_load(
     assert constants.ROUTER in app._state()._var_dependencies
 
     substate_token = BaseStateToken(ident=token, cls=DynamicState)
-    async with app.state_manager.modify_state(substate_token) as state:
-        state.router_data = {"simulate": "hydrated"}
-        assert state.dynamic == ""  # pyright: ignore[reportAttributeAccessIssue]
     exp_vals = ["foo", "foobar", "baz"]
 
     def _event(name, val, **kwargs):
@@ -1915,10 +1912,6 @@ async def test_process_events(
         payload={"c": 5},
         router_data={},
     )
-    async with mock_root_event_context.state_manager.modify_state(
-        BaseStateToken(ident=token, cls=GenState),
-    ) as state:
-        state.router_data = {"simulate": "hydrated"}
 
     async with mock_base_state_event_processor as processor:
         await processor.enqueue(

tests/units/test_state.py

@@ -2242,10 +2242,6 @@ class BackgroundTaskState(BaseState):
     dict_list: dict[str, list[int]] = {"foo": [1, 2, 3]}
     dc: ModelDC = ModelDC()
 
-    def __init__(self, **kwargs):  # noqa: D107
-        super().__init__(**kwargs)
-        self.router_data = {"simulate": "hydrate"}
-
     @rx.var(cache=False)
     def computed_order(self) -> list[str]:
         """Get the order as a computed var.
@@ -2909,7 +2905,7 @@ class BaseFieldSetterState(BaseState):
     assert "c2" in bfss.dirty_vars
 
 
-def exp_is_hydrated(state: BaseState, is_hydrated: bool = True) -> dict[str, Any]:
+def exp_is_hydrated(state: type[BaseState], is_hydrated: bool = True) -> dict[str, Any]:
     """Expected IS_HYDRATED delta that would be emitted by HydrateMiddleware.
 
     Args:
@@ -3006,11 +3002,6 @@ def index():
     app.add_page(index, on_load=test_state.test_handler)
     app._compile_page("index")
 
-    async with mock_root_event_context.state_manager.modify_state(
-        BaseStateToken(ident=token, cls=State)
-    ) as state:
-        state.router_data = {"simulate": "hydrate"}
-
     on_load_internal_name = format.format_event_handler(
         OnLoadInternalState.on_load_internal  # pyright: ignore[reportArgumentType]
     )
@@ -3035,7 +3026,7 @@ def index():
     assert len(emitted_deltas) >= 2
     first_delta = emitted_deltas[0][1]
     assert first_delta[State.get_full_name()].pop("router" + FIELD_MARKER) is not None
-    assert first_delta == exp_is_hydrated(state, False)
+    assert first_delta == exp_is_hydrated(State, False)
 
     # Find the delta containing the test handler's state change
     handler_deltas = [
@@ -3085,10 +3076,6 @@ def index():
 
     app.add_page(index, on_load=[OnLoadState.test_handler, OnLoadState.test_handler])
     app._compile_page("index")
-    async with mock_root_event_context.state_manager.modify_state(
-        BaseStateToken(ident=token, cls=State)
-    ) as state:
-        state.router_data = {"simulate": "hydrate"}
 
     on_load_internal_name = format.format_event_handler(
         OnLoadInternalState.on_load_internal  # pyright: ignore[reportArgumentType]
@@ -3112,7 +3099,7 @@ def index():
     assert len(emitted_deltas) >= 2
     first_delta = emitted_deltas[0][1]
     assert first_delta[State.get_full_name()].pop("router" + FIELD_MARKER) is not None
-    assert first_delta == exp_is_hydrated(state, False)
+    assert first_delta == exp_is_hydrated(State, False)
 
     # Find deltas containing the test handler's state change (num incremented twice)
     handler_deltas = [