used anyOf for scheduler

shvatka/infrastructure/scheduler/context.py

@@ -7,7 +7,6 @@ class ScheduledContextHolder:
     ATTENTION!
     GLOBAL VARIABLE!
     """
-
     dishka: AsyncContainer
 
 

shvatka/infrastructure/scheduler/factory.py

@@ -1,8 +1,8 @@
 from typing import AsyncIterable
 
-from dishka import Provider, Scope, provide, AsyncContainer
+from dishka import Provider, Scope, provide, AsyncContainer, AnyOf
 
-from shvatka.core.interfaces.scheduler import Scheduler
+from shvatka.core.interfaces.scheduler import Scheduler, LevelTestScheduler
 from shvatka.infrastructure.db.config.models.db import RedisConfig
 from shvatka.infrastructure.scheduler import ApScheduler
 
@@ -13,6 +13,6 @@ class SchedulerProvider(Provider):
     @provide
     async def create_scheduler(
         self, dishka: AsyncContainer, redis_config: RedisConfig
-    ) -> AsyncIterable[Scheduler]:
+    ) ->  AsyncIterable[AnyOf[Scheduler, LevelTestScheduler]]:
         async with ApScheduler(dishka=dishka, redis_config=redis_config) as scheduler:
             yield scheduler

shvatka/infrastructure/scheduler/wrappers.py

@@ -86,7 +86,7 @@ async def send_hint_for_testing_wrapper(
     dao: FromDishka[HolderDao],
     bot: FromDishka[Bot],
     file_storage: FromDishka[FileStorage],
-    scheduler: FromDishka[Scheduler],
+    scheduler: FromDishka[LevelTestScheduler],
 ):
     level = await dao.level.get_by_id(level_id)
     game = await dao.game.get_by_id(game_id)
@@ -97,8 +97,6 @@ async def send_hint_for_testing_wrapper(
         suite=dto.LevelTestSuite(level=level, tester=org),
         hint_number=hint_number,
         view=create_level_test_view(bot, dao, file_storage),
-        scheduler=typing.cast(
-            LevelTestScheduler, scheduler
-        ),  # TODO typing.cast replace with better hint
+        scheduler=scheduler,
         dao=dao.level_testing_complex,
     )