From 45f6b4ab4a656894fb1fc98cba97b191390c3eef Mon Sep 17 00:00:00 2001 From: Artur Shiriev Date: Mon, 22 Dec 2025 13:34:23 +0300 Subject: [PATCH] refactor fetch_ready_timers to be able to use it in tests --- redis_timers/timers.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/redis_timers/timers.py b/redis_timers/timers.py index 1d8a34f..05abd21 100644 --- a/redis_timers/timers.py +++ b/redis_timers/timers.py @@ -53,10 +53,8 @@ def include_routers(self, *routers: Router) -> None: def _convert_datetime_to_redis_format(datetime_point: datetime.datetime) -> str: return str(int(datetime_point.timestamp())) - async def _fetch_ready_timers(self) -> list[str]: - current_timestamp: typing.Final = self._convert_datetime_to_redis_format( - datetime.datetime.now(tz=TIMEZONE), - ) + async def fetch_ready_timers(self, timestamp: datetime.datetime) -> list[str]: + current_timestamp: typing.Final = self._convert_datetime_to_redis_format(timestamp) return await self.redis_client.zrangebyscore( settings.TIMERS_TIMELINE_KEY, "-inf", @@ -85,7 +83,7 @@ async def _handle_one_timer(self, timer_key: str) -> None: await handler.handler(payload, self.context) async def handle_ready_timers(self) -> None: - ready_timers = await self._fetch_ready_timers() + ready_timers = await self.fetch_ready_timers(datetime.datetime.now(tz=TIMEZONE)) for timer_key in ready_timers: lock = consume_lock( redis_client=self.redis_client,