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 as rx
import reflex.config import reflex.config
import reflex.utils.console
from reflex import constants from reflex import constants
from reflex.app import App from reflex.app import App
from reflex.base import Base 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. substate_token_redis: A token + substate name for looking up in state manager.
mocker: Pytest mocker object. 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_expiration = LOCK_EXPIRATION
state_manager_redis.lock_warning_threshold = LOCK_WARNING_THRESHOLD 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] await tasks[0]
console_warn.assert_called() console_warn.assert_called()
assert console_warn.call_count == 7 assert console_warn.call_count == 1
class CopyingAsyncMock(AsyncMock): class CopyingAsyncMock(AsyncMock):