🐛 改改单测

This commit is contained in:
2024-10-13 21:51:47 +08:00
parent b6ba904a68
commit 2093622672
5 changed files with 63 additions and 53 deletions
+5
View File
@@ -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)