From c059feba9ccb218d05a26968d77e476fe86d2b51 Mon Sep 17 00:00:00 2001
From: felinae98 <731499577@qq.com>
Date: Fri, 7 Oct 2022 17:10:56 +0800
Subject: [PATCH] fix test

---
 tests/scheduler/test_scheduler.py | 28 +++++++++++++---------------
 1 file changed, 13 insertions(+), 15 deletions(-)

diff --git a/tests/scheduler/test_scheduler.py b/tests/scheduler/test_scheduler.py
index fa4294f..53d0f66 100644
--- a/tests/scheduler/test_scheduler.py
+++ b/tests/scheduler/test_scheduler.py
@@ -1,12 +1,7 @@
-import typing
 from datetime import time
 
-import pytest
 from nonebug import App
 
-if typing.TYPE_CHECKING:
-    from nonebot_bison.scheduler.scheduler import Schedulable
-
 
 async def get_schedule_times(scheduler_class: str, time: int) -> dict[str, int]:
     from nonebot_bison.scheduler import scheduler_dict
@@ -24,7 +19,7 @@ async def get_schedule_times(scheduler_class: str, time: int) -> dict[str, int]:
 async def test_scheduler_without_time(init_scheduler):
     from nonebot_bison.config import config
     from nonebot_bison.config.db_config import WeightConfig
-    from nonebot_bison.scheduler.manager import init_scheduler, scheduler_dict
+    from nonebot_bison.scheduler.manager import init_scheduler
     from nonebot_bison.types import Target as T_Target
 
     await config.add_subscribe(
@@ -38,10 +33,10 @@ async def test_scheduler_without_time(init_scheduler):
     )
 
     await config.update_time_weight_config(
-        T_Target("t2"), "bilibili", WeightConfig(20, [])
+        T_Target("t2"), "bilibili", WeightConfig(default=20, time_config=[])
     )
     await config.update_time_weight_config(
-        T_Target("t2"), "bilibili-live", WeightConfig(30, [])
+        T_Target("t2"), "bilibili-live", WeightConfig(default=30, time_config=[])
     )
 
     await init_scheduler()
@@ -60,7 +55,7 @@ async def test_scheduler_without_time(init_scheduler):
 async def test_scheduler_with_time(app: App, init_scheduler):
     from nonebot_bison.config import config, db_config
     from nonebot_bison.config.db_config import TimeWeightConfig, WeightConfig
-    from nonebot_bison.scheduler.manager import init_scheduler, scheduler_dict
+    from nonebot_bison.scheduler.manager import init_scheduler
     from nonebot_bison.types import Target as T_Target
 
     await config.add_subscribe(
@@ -76,10 +71,15 @@ async def test_scheduler_with_time(app: App, init_scheduler):
     await config.update_time_weight_config(
         T_Target("t2"),
         "bilibili",
-        WeightConfig(20, [TimeWeightConfig(time(10), time(11), 1000)]),
+        WeightConfig(
+            default=20,
+            time_config=[
+                TimeWeightConfig(start_time=time(10), end_time=time(11), weight=1000)
+            ],
+        ),
     )
     await config.update_time_weight_config(
-        T_Target("t2"), "bilibili-live", WeightConfig(30, [])
+        T_Target("t2"), "bilibili-live", WeightConfig(default=30, time_config=[])
     )
 
     await init_scheduler()
@@ -103,8 +103,7 @@ async def test_scheduler_with_time(app: App, init_scheduler):
 
 async def test_scheduler_add_new(init_scheduler):
     from nonebot_bison.config import config
-    from nonebot_bison.config.db_config import WeightConfig
-    from nonebot_bison.scheduler.manager import init_scheduler, scheduler_dict
+    from nonebot_bison.scheduler.manager import init_scheduler
     from nonebot_bison.types import Target as T_Target
 
     await config.add_subscribe(
@@ -125,8 +124,7 @@ async def test_scheduler_add_new(init_scheduler):
 
 async def test_schedule_delete(init_scheduler):
     from nonebot_bison.config import config
-    from nonebot_bison.config.db_config import WeightConfig
-    from nonebot_bison.scheduler.manager import init_scheduler, scheduler_dict
+    from nonebot_bison.scheduler.manager import init_scheduler
     from nonebot_bison.types import Target as T_Target
 
     await config.add_subscribe(