drain _process in tests

tests/units/test_app.py

@@ -471,15 +471,15 @@ async def test_dynamic_var_event(test_state: Type[ATestState], token: str):
     """
     state = test_state()  # pyright: ignore [reportCallIssue]
     state.add_var("int_val", int, 0)
-    result = await state._process(
+    async for result in state._process(
         Event(
             token=token,
             name=f"{test_state.get_name()}.set_int_val",
             router_data={"pathname": "/", "query": {}},
             payload={"value": 50},
         )
-    ).__anext__()
+    ):
-    assert result.delta == {test_state.get_name(): {"int_val": 50}}
+        assert result.delta == {test_state.get_name(): {"int_val": 50}}
 
 
 @pytest.mark.asyncio
@@ -583,18 +583,17 @@ async def test_list_mutation_detection__plain_list(
         token: a Token.
     """
     for event_name, expected_delta in event_tuples:
-        result = await list_mutation_state._process(
+        async for result in list_mutation_state._process(
             Event(
                 token=token,
                 name=f"{list_mutation_state.get_name()}.{event_name}",
                 router_data={"pathname": "/", "query": {}},
                 payload={},
             )
-        ).__anext__()
+        ):
-
+            # prefix keys in expected_delta with the state name
-        # prefix keys in expected_delta with the state name
+            expected_delta = {list_mutation_state.get_name(): expected_delta}
-        expected_delta = {list_mutation_state.get_name(): expected_delta}
+            assert result.delta == expected_delta
-        assert result.delta == expected_delta
 
 
 @pytest.mark.asyncio
@@ -709,19 +708,18 @@ async def test_dict_mutation_detection__plain_list(
         token: a Token.
     """
     for event_name, expected_delta in event_tuples:
-        result = await dict_mutation_state._process(
+        async for result in dict_mutation_state._process(
             Event(
                 token=token,
                 name=f"{dict_mutation_state.get_name()}.{event_name}",
                 router_data={"pathname": "/", "query": {}},
                 payload={},
             )
-        ).__anext__()
+        ):
+            # prefix keys in expected_delta with the state name
+            expected_delta = {dict_mutation_state.get_name(): expected_delta}
 
-        # prefix keys in expected_delta with the state name
+            assert result.delta == expected_delta
-        expected_delta = {dict_mutation_state.get_name(): expected_delta}
-
-        assert result.delta == expected_delta
 
 
 @pytest.mark.asyncio

tests/units/test_state.py

@@ -789,17 +789,16 @@ async def test_process_event_simple(test_state):
     assert test_state.num1 == 0
 
     event = Event(token="t", name="set_num1", payload={"value": 69})
-    update = await test_state._process(event).__anext__()
+    async for update in test_state._process(event):
+        # The event should update the value.
+        assert test_state.num1 == 69
 
-    # The event should update the value.
+        # The delta should contain the changes, including computed vars.
-    assert test_state.num1 == 69
+        assert update.delta == {
-
+            TestState.get_full_name(): {"num1": 69, "sum": 72.14},
-    # The delta should contain the changes, including computed vars.
+            GrandchildState3.get_full_name(): {"computed": ""},
-    assert update.delta == {
+        }
-        TestState.get_full_name(): {"num1": 69, "sum": 72.14},
+        assert update.events == []
-        GrandchildState3.get_full_name(): {"computed": ""},
-    }
-    assert update.events == []
 
 
 @pytest.mark.asyncio
@@ -819,15 +818,15 @@ async def test_process_event_substate(test_state, child_state, grandchild_state)
         name=f"{ChildState.get_name()}.change_both",
         payload={"value": "hi", "count": 12},
     )
-    update = await test_state._process(event).__anext__()
+    async for update in test_state._process(event):
-    assert child_state.value == "HI"
+        assert child_state.value == "HI"
-    assert child_state.count == 24
+        assert child_state.count == 24
-    assert update.delta == {
+        assert update.delta == {
-        # TestState.get_full_name(): {"sum": 3.14, "upper": ""},
+            # TestState.get_full_name(): {"sum": 3.14, "upper": ""},
-        ChildState.get_full_name(): {"value": "HI", "count": 24},
+            ChildState.get_full_name(): {"value": "HI", "count": 24},
-        GrandchildState3.get_full_name(): {"computed": ""},
+            GrandchildState3.get_full_name(): {"computed": ""},
-    }
+        }
-    test_state._clean()
+        test_state._clean()
 
     # Test with the granchild state.
     assert grandchild_state.value2 == ""
@@ -836,13 +835,13 @@ async def test_process_event_substate(test_state, child_state, grandchild_state)
         name=f"{GrandchildState.get_full_name()}.set_value2",
         payload={"value": "new"},
     )
-    update = await test_state._process(event).__anext__()
+    async for update in test_state._process(event):
-    assert grandchild_state.value2 == "new"
+        assert grandchild_state.value2 == "new"
-    assert update.delta == {
+        assert update.delta == {
-        # TestState.get_full_name(): {"sum": 3.14, "upper": ""},
+            # TestState.get_full_name(): {"sum": 3.14, "upper": ""},
-        GrandchildState.get_full_name(): {"value2": "new"},
+            GrandchildState.get_full_name(): {"value2": "new"},
-        GrandchildState3.get_full_name(): {"computed": ""},
+            GrandchildState3.get_full_name(): {"computed": ""},
-    }
+        }
 
 
 @pytest.mark.asyncio
@@ -2909,10 +2908,10 @@ async def test_preprocess(app_module_mock, token, test_state, expected, mocker):
 
     events = updates[0].events
     assert len(events) == 2
-    assert (await state._process(events[0]).__anext__()).delta == {
+    async for update in state._process(events[0]):
-        test_state.get_full_name(): {"num": 1}
+        assert update.delta == {test_state.get_full_name(): {"num": 1}}
-    }
+    async for update in state._process(events[1]):
-    assert (await state._process(events[1]).__anext__()).delta == exp_is_hydrated(state)
+        assert update.delta == exp_is_hydrated(state)
 
     if isinstance(app.state_manager, StateManagerRedis):
         await app.state_manager.close()
@@ -2957,13 +2956,12 @@ async def test_preprocess_multiple_load_events(app_module_mock, token, mocker):
 
     events = updates[0].events
     assert len(events) == 3
-    assert (await state._process(events[0]).__anext__()).delta == {
+    async for update in state._process(events[0]):
-        OnLoadState.get_full_name(): {"num": 1}
+        assert update.delta == {OnLoadState.get_full_name(): {"num": 1}}
-    }
+    async for update in state._process(events[1]):
-    assert (await state._process(events[1]).__anext__()).delta == {
+        assert update.delta == {OnLoadState.get_full_name(): {"num": 2}}
-        OnLoadState.get_full_name(): {"num": 2}
+    async for update in state._process(events[2]):
-    }
+        assert update.delta == exp_is_hydrated(state)
-    assert (await state._process(events[2]).__anext__()).delta == exp_is_hydrated(state)
 
     if isinstance(app.state_manager, StateManagerRedis):
         await app.state_manager.close()