diff --git a/tests/ut/backends/test_memory.py b/tests/ut/backends/test_memory.py index 200feccdf..5b12fc8e2 100644 --- a/tests/ut/backends/test_memory.py +++ b/tests/ut/backends/test_memory.py @@ -43,6 +43,15 @@ async def test_set_no_ttl_no_handle(self, memory): await memory._set(pytest.KEY, "value") assert pytest.KEY not in memory._handlers + @pytest.mark.asyncio + async def test_set_cancel_previous_ttl_handle(self, memory, mocker): + fake_timer_handle_cancel = mocker.patch("asyncio.TimerHandle.cancel") + await memory._set(pytest.KEY, "value", ttl=0.1) + fake_timer_handle_cancel.assert_not_called() + + await memory._set(pytest.KEY, "new_value", ttl=0.1) + fake_timer_handle_cancel.assert_called_once_with() + @pytest.mark.asyncio async def test_set_ttl_handle(self, memory): await memory._set(pytest.KEY, "value", ttl=100)