Add a unit test

aio-libs · Nov 11, 2018 · 86cb33e · 86cb33e
1 parent 0d2e566
commit 86cb33e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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)