longpeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update test_state_manager_lock_warning_threshold_contend expectations

Browse Source 
Now that all substates are persisted together, there is only 1 warning message
expected, not 7
This commit is contained in:
Masen Furer 2024-12-13 10:56:48 -08:00
parent f8bfc78f8f
commit c347dc8e4b
No known key found for this signature in database
GPG Key ID: B0008AD22B3B3A95
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/units/test_state.py
View File

@ -31,6 +31,7 @@ from pydantic.v1 import BaseModel as BaseModelV1
import reflex as rx
import reflex.config
import reflex.utils.console
from reflex import constants
from reflex.app import App
from reflex.base import Base
@ -1857,7 +1858,7 @@ async def test_state_manager_lock_warning_threshold_contend(
substate_token_redis: A token + substate name for looking up in state manager.
mocker: Pytest mocker object.
"""
console_warn = mocker.patch("reflex.utils.console.warn")
console_warn = mocker.spy(reflex.utils.console, "warn")
state_manager_redis.lock_expiration = LOCK_EXPIRATION
state_manager_redis.lock_warning_threshold = LOCK_WARNING_THRESHOLD
@ -1875,7 +1876,7 @@ async def test_state_manager_lock_warning_threshold_contend(
await tasks[0]
console_warn.assert_called()
assert console_warn.call_count == 7
assert console_warn.call_count == 1
class CopyingAsyncMock(AsyncMock):