✅ 初步实现添加cookie的单元测试

nonebot_bison/sub_manager/__init__.py

@@ -47,7 +47,7 @@ do_del_sub(del_sub_matcher)
 add_cookie_matcher = on_command(
     "添加cookie",
     aliases={"添加Cookie"},
-    rule=configurable_to_me,
+    rule=to_me(),
     permission=SUPERUSER,
     priority=5,
     block=True,
@@ -57,7 +57,7 @@ do_add_cookie(add_cookie_matcher)
 add_cookie_target_matcher = on_command(
     "关联cookie",
     aliases={"关联Cookie"},
-    rule=configurable_to_me,
+    rule=to_me(),
     permission=SUPERUSER,
     priority=5,
     block=True,
@@ -67,7 +67,7 @@ do_add_cookie_target(add_cookie_target_matcher)
 del_cookie_target_matcher = on_command(
     "取消关联cookie",
     aliases={"取消关联Cookie"},
-    rule=configurable_to_me,
+    rule=to_me(),
     permission=SUPERUSER,
     priority=5,
     block=True,
@@ -77,7 +77,7 @@ do_del_cookie_target(del_cookie_target_matcher)
 del_cookie_matcher = on_command(
     "删除cookie",
     aliases={"删除Cookie"},
-    rule=configurable_to_me,
+    rule=to_me(),
     permission=SUPERUSER,
     priority=5,
     block=True,
@@ -167,4 +167,8 @@ __all__ = [
     "del_sub_matcher",
     "group_manage_matcher",
     "no_permission_matcher",
+    "add_cookie_matcher",
+    "add_cookie_target_matcher",
+    "del_cookie_target_matcher",
+    "del_cookie_matcher",
 ]

nonebot_bison/sub_manager/add_cookie.py

@@ -57,7 +57,7 @@ def do_add_cookie(add_cookie: type[Matcher]):
         )
         cookie_text = cookie.extract_plain_text()
         if not await client_mgr.validate_cookie(cookie_text):
-            await add_cookie.reject("无效的 Cookie，请检查后重新输入，详情见<待添加的文档>")
+            await add_cookie.reject(state["site"].cookie_format_prompt)
         state["cookie"] = cookie_text
 
     @add_cookie.handle()

nonebot_bison/utils/site.py

@@ -156,6 +156,7 @@ class Site(metaclass=RegistryMeta, base=True):
     client_mgr: type[ClientManager] = DefaultClientManager
     require_browser: bool = False
     registry: list[type["Site"]]
+    cookie_format_prompt = "无效的 Cookie，请检查后重新输入，详情见<待添加的文档>"
 
     def __str__(self):
         return f"[{self.name}]-{self.name}-{self.schedule_setting}"