suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2026-06-23 14:16:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 优化对 weibo get_cookie_name 的 mock

Browse Source
This commit is contained in:
suyiiyii
2024-10-28 22:47:31 +08:00
parent 641cc44a12
commit 3a0f95b712
3 changed files with 59 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/config/test_cookie.py
+5 -9
View File
@@ -1,12 +1,12 @@
import json
from typing import cast
from datetime import datetime
from unittest.mock import patch
import pytest
from nonebug import App
@pytest.mark.usefixtures("_patch_weibo_get_cookie_name")
async def test_cookie(app: App, init_scheduler):
from nonebot_plugin_saa import TargetQQGroup
@@ -34,11 +34,9 @@ async def test_cookie(app: App, init_scheduler):
cookies = await config.get_cookie(site_name=site.name)
assert len(cookies) == 1
with patch("nonebot_bison.platform.weibo.WeiboSite._get_current_user_name") as mock:
mock.return_value = "test_name"
# 添加用户cookie
await client_mgr.add_user_cookie(json.dumps({"test_cookie": "1"}))
await client_mgr.add_user_cookie(json.dumps({"test_cookie": "2"}))
# 添加用户cookie
await client_mgr.add_user_cookie(json.dumps({"test_cookie": "1"}))
await client_mgr.add_user_cookie(json.dumps({"test_cookie": "2"}))
cookies = await config.get_cookie(site_name=site.name)
assert len(cookies) == 3
@@ -70,9 +68,7 @@ async def test_cookie(app: App, init_scheduler):
tags=[],
)
with patch("nonebot_bison.platform.weibo.WeiboSite._get_current_user_name") as mock:
mock.return_value = "test_name"
await client_mgr.add_user_cookie(json.dumps({"test_cookie": "3"}))
await client_mgr.add_user_cookie(json.dumps({"test_cookie": "3"}))
cookies = await config.get_cookie(site_name=site.name, is_anonymous=False)
# 多个target，多个cookie