55from freezegun import freeze_time
66from taskiq import ScheduledTask
77
8- from taskiq_redis .array_schedule_source import ArrayRedisScheduleSource
8+ from taskiq_redis .list_schedule_source import ListRedisScheduleSource
99from taskiq_redis .schedule_source import RedisScheduleSource
1010
1111
1414async def test_schedule_cron (redis_url : str ) -> None :
1515 """Test adding a cron schedule."""
1616 prefix = uuid .uuid4 ().hex
17- source = ArrayRedisScheduleSource (redis_url , prefix = prefix )
17+ source = ListRedisScheduleSource (redis_url , prefix = prefix )
1818 schedule = ScheduledTask (
1919 task_name = "test_task" ,
2020 labels = {},
@@ -32,7 +32,7 @@ async def test_schedule_cron(redis_url: str) -> None:
3232async def test_schedule_from_past (redis_url : str ) -> None :
3333 """Test adding a cron schedule."""
3434 prefix = uuid .uuid4 ().hex
35- source = ArrayRedisScheduleSource (redis_url , prefix = prefix )
35+ source = ListRedisScheduleSource (redis_url , prefix = prefix )
3636 schedule = ScheduledTask (
3737 task_name = "test_task" ,
3838 labels = {},
@@ -57,7 +57,7 @@ async def test_schedule_from_past(redis_url: str) -> None:
5757async def test_schedule_removal (redis_url : str ) -> None :
5858 """Test adding a cron schedule."""
5959 prefix = uuid .uuid4 ().hex
60- source = ArrayRedisScheduleSource (redis_url , prefix = prefix )
60+ source = ListRedisScheduleSource (redis_url , prefix = prefix )
6161 schedule = ScheduledTask (
6262 task_name = "test_task" ,
6363 labels = {},
@@ -82,7 +82,7 @@ async def test_schedule_removal(redis_url: str) -> None:
8282async def test_deletion (redis_url : str ) -> None :
8383 """Test adding a cron schedule."""
8484 prefix = uuid .uuid4 ().hex
85- source = ArrayRedisScheduleSource (redis_url , prefix = prefix )
85+ source = ListRedisScheduleSource (redis_url , prefix = prefix )
8686 schedule = ScheduledTask (
8787 task_name = "test_task" ,
8888 labels = {},
@@ -121,7 +121,7 @@ async def test_migration(redis_url: str) -> None:
121121
122122 old_schedules = await old_source .get_schedules ()
123123
124- source = ArrayRedisScheduleSource (
124+ source = ListRedisScheduleSource (
125125 redis_url ,
126126 prefix = new_prefix ,
127127 skip_past_schedules = True ,
0 commit comments