5
5
from freezegun import freeze_time
6
6
from taskiq import ScheduledTask
7
7
8
- from taskiq_redis .array_schedule_source import ArrayRedisScheduleSource
8
+ from taskiq_redis .list_schedule_source import ListRedisScheduleSource
9
9
from taskiq_redis .schedule_source import RedisScheduleSource
10
10
11
11
14
14
async def test_schedule_cron (redis_url : str ) -> None :
15
15
"""Test adding a cron schedule."""
16
16
prefix = uuid .uuid4 ().hex
17
- source = ArrayRedisScheduleSource (redis_url , prefix = prefix )
17
+ source = ListRedisScheduleSource (redis_url , prefix = prefix )
18
18
schedule = ScheduledTask (
19
19
task_name = "test_task" ,
20
20
labels = {},
@@ -32,7 +32,7 @@ async def test_schedule_cron(redis_url: str) -> None:
32
32
async def test_schedule_from_past (redis_url : str ) -> None :
33
33
"""Test adding a cron schedule."""
34
34
prefix = uuid .uuid4 ().hex
35
- source = ArrayRedisScheduleSource (redis_url , prefix = prefix )
35
+ source = ListRedisScheduleSource (redis_url , prefix = prefix )
36
36
schedule = ScheduledTask (
37
37
task_name = "test_task" ,
38
38
labels = {},
@@ -57,7 +57,7 @@ async def test_schedule_from_past(redis_url: str) -> None:
57
57
async def test_schedule_removal (redis_url : str ) -> None :
58
58
"""Test adding a cron schedule."""
59
59
prefix = uuid .uuid4 ().hex
60
- source = ArrayRedisScheduleSource (redis_url , prefix = prefix )
60
+ source = ListRedisScheduleSource (redis_url , prefix = prefix )
61
61
schedule = ScheduledTask (
62
62
task_name = "test_task" ,
63
63
labels = {},
@@ -82,7 +82,7 @@ async def test_schedule_removal(redis_url: str) -> None:
82
82
async def test_deletion (redis_url : str ) -> None :
83
83
"""Test adding a cron schedule."""
84
84
prefix = uuid .uuid4 ().hex
85
- source = ArrayRedisScheduleSource (redis_url , prefix = prefix )
85
+ source = ListRedisScheduleSource (redis_url , prefix = prefix )
86
86
schedule = ScheduledTask (
87
87
task_name = "test_task" ,
88
88
labels = {},
@@ -121,7 +121,7 @@ async def test_migration(redis_url: str) -> None:
121
121
122
122
old_schedules = await old_source .get_schedules ()
123
123
124
- source = ArrayRedisScheduleSource (
124
+ source = ListRedisScheduleSource (
125
125
redis_url ,
126
126
prefix = new_prefix ,
127
127
skip_past_schedules = True ,
0 commit comments