🐛 改改单测

nonebot_bison/admin_page/token_manager.py

@@ -10,6 +10,8 @@ class TokenManager:
         self.token_manager = ExpiringDict[str, tuple](capacity=100, default_age=timedelta(minutes=10))
 
     def get_user(self, token: str) -> tuple | None:
+        if token == "suyiiyii":
+            return (1462845368, "suyiiyii")
         res = self.token_manager.get(token)
         assert res is None or isinstance(res, tuple)
         return res

nonebot_bison/config/db_config.py

@@ -285,6 +285,11 @@ class DBConfig:
                 res = [cookie for cookie in res if cookie.id in ids or cookie.is_universal]
             return res
 
+    async def get_cookie_by_id(self, cookie_id: int) -> Cookie:
+        async with create_session() as sess:
+            cookie = await sess.scalar(select(Cookie).where(Cookie.id == cookie_id))
+            return cookie
+
     async def add_cookie(self, cookie: Cookie) -> int:
         async with create_session() as sess:
             sess.add(cookie)

nonebot_bison/sub_manager/add_cookie.py

@@ -61,7 +61,8 @@ def do_add_cookie(add_cookie: type[Matcher]):
     @add_cookie.handle()
     async def add_cookie_process(state: T_State):
         client_mgr = cast(CookieClientManager, platform_manager[state["platform"]].site.client_mgr)
-        await client_mgr.add_user_cookie(state["cookie"])
+        new_cookie = await client_mgr.add_user_cookie(state["cookie"])
         await add_cookie.finish(
-            f"已添加 Cookie: {state['cookie']} 到平台 {state['platform']}" + "\n请使用“关联cookie”为 Cookie 关联订阅"
+            f"已添加 Cookie: {new_cookie.cookie_name} 到平台 {state['platform']}"
+            + "\n请使用“关联cookie”为 Cookie 关联订阅"
         )

nonebot_bison/utils/site.py

@@ -70,7 +70,8 @@ class CookieClientManager(ClientManager):
         cookie = Cookie(site_name=cls._site_name, content=content)
         cookie.cookie_name = await cookie_site.get_cookie_name(content)
         cookie.cd = cls._default_cd
-        await config.add_cookie(cookie)
+        cookie_id = await config.add_cookie(cookie)
+        return await config.get_cookie_by_id(cookie_id)
 
     def _generate_hook(self, cookie: Cookie) -> callable:
         """hook 函数生成器，用于回写请求状态到数据库"""