async-timer 1.1.3__tar.gz → 1.1.5__tar.gz

{async_timer-1.1.3 → async_timer-1.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: async-timer
-Version: 1.1.3
+Version: 1.1.5
 Summary: The missing Python async timer.
 License: MIT
 Keywords: async,timer

{async_timer-1.1.3 → async_timer-1.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "async-timer"
-version = "v1.1.3"
+version = "v1.1.5"
 description = "The missing Python async timer."
 authors = ["Ilya O. <vrghost@gmail.com>"]
 license = "MIT"

{async_timer-1.1.3 → async_timer-1.1.5}/src/mock_async_timer/timer.py

@@ -1,3 +1,4 @@
+import asyncio
 import unittest.mock
 
 import async_timer
@@ -13,8 +14,23 @@ class MockPacemaker(async_timer.pacemaker.TimerPacemaker):
     async def _try_wait(self, delay: float):
         if self._cancel_evt.is_set():
             raise StopAsyncIteration()
+
+        await self._sleep_until_next_loop_iter()
         await self.sleep(delay)
 
+    async def _sleep_until_next_loop_iter(self):
+        """Awaiting this function will release on the next async loop iteration"""
+        fut = asyncio.Future()
+        asyncio.get_event_loop().call_soon(lambda: fut.set_result(42))
+        await fut
+
+    @classmethod
+    def fromPacemaker(cls, original: async_timer.pacemaker.TimerPacemaker):
+        """Create MockPacemaker from the non-mock original."""
+        out = cls(delay=original.delay)
+        out.stop_on(original._cancel_futs)
+        return out
+
 
 class MockTimer(async_timer.Timer):
     """Test-friendly mock timer class.
@@ -27,4 +43,4 @@ class MockTimer(async_timer.Timer):
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
-        self.pacemaker = MockPacemaker(self.pacemaker.delay)
+        self.pacemaker = MockPacemaker.fromPacemaker(self.pacemaker)