suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-06-02 09:26:12 +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
Signed by: suyiiyii
GPG Key ID: 044704CB29B8AD85
3 changed files with 59 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/config/test_cookie.py
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,8 +34,6 @@ 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"}))
@ -70,8 +68,6 @@ 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"}))
cookies = await config.get_cookie(site_name=site.name, is_anonymous=False)

6
tests/conftest.py
View File

@ -29,7 +29,7 @@ def load_adapters(nonebug_init: None):
return driver
def _patch_refresh_bilibili_anonymous_cookie(mocker: MockerFixture):
def patch_refresh_bilibili_anonymous_cookie(mocker: MockerFixture):
# patch 掉bilibili的匿名cookie生成函数，避免真实请求
from nonebot_bison.platform.bilibili.scheduler import BilibiliClientManager
@ -63,7 +63,7 @@ async def app(tmp_path: Path, request: pytest.FixtureRequest, mocker: MockerFixt
# 如果在 app 前调用会报错“无法找到调用者”
# 而在后面调用又来不及mock，所以只能在中间mock
_patch_refresh_bilibili_anonymous_cookie(mocker)
patch_refresh_bilibili_anonymous_cookie(mocker)
if not param.get("no_init_db"):
await init_db()
@ -153,6 +153,6 @@ async def _clear_db(app: App):
def _patch_weibo_get_cookie_name(app: App, mocker: MockerFixture):
from nonebot_bison.platform import weibo
mocker.patch.object(weibo.WeiboSite, "get_cookie_name", return_value="weibo_cookie_name")
mocker.patch.object(weibo.WeiboSite, "_get_current_user_name", return_value="test_name")
yield
mocker.stopall()

4
tests/sub_manager/test_add_cookie.py
View File

@ -1,5 +1,4 @@
import json
from unittest.mock import patch
import pytest
from nonebug.app import App
@ -84,6 +83,7 @@ async def test_add_cookie_target_no_cookie(app: App):
@pytest.mark.usefixtures("_clear_db")
@pytest.mark.usefixtures("_patch_weibo_get_cookie_name")
async def test_add_cookie(app: App):
from nonebot.adapters.onebot.v11.bot import Bot
from nonebot.adapters.onebot.v11.message import Message
@ -91,8 +91,6 @@ async def test_add_cookie(app: App):
from nonebot_bison.platform import platform_manager
from nonebot_bison.sub_manager import common_platform, add_cookie_matcher, add_cookie_target_matcher
with patch("nonebot_bison.platform.weibo.WeiboSite._get_current_user_name") as mock:
mock.return_value = "test_name"
async with app.test_matcher(add_cookie_matcher) as ctx:
bot = ctx.create_bot(base=Bot)
event_1 = fake_private_message_event(