mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-05 19:36:43 +08:00
101 lines
5.1 KiB
Python
101 lines
5.1 KiB
Python
import typing
|
||
from datetime import datetime
|
||
|
||
import pytest
|
||
import respx
|
||
from httpx import Response
|
||
from nonebug.app import App
|
||
from pytz import timezone
|
||
|
||
from .utils import get_file, get_json
|
||
|
||
|
||
@pytest.fixture(scope="module")
|
||
def bing_dy_list():
|
||
return get_json("bilibili_bing_list.json")["data"]["cards"]
|
||
|
||
|
||
if typing.TYPE_CHECKING:
|
||
from nonebot_bison.platform.bilibili import Bilibili
|
||
|
||
|
||
@pytest.fixture
|
||
def bilibili(app: App):
|
||
from nonebot_bison.platform import platform_manager
|
||
|
||
return platform_manager["bilibili"]
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_video_forward(bilibili, bing_dy_list):
|
||
post = await bilibili.parse(bing_dy_list[1])
|
||
assert (
|
||
post.text
|
||
== "答案揭晓:宿舍!来看看投票结果\nhttps://t.bilibili.com/568093580488553786\n--------------\n#可露希尔的秘密档案# \n11:来宿舍休息一下吧 \n档案来源:lambda:\\罗德岛内务\\秘密档案 \n发布时间:9/12 1:00 P.M. \n档案类型:可见 \n档案描述:今天请了病假在宿舍休息。很舒适。 \n提供者:赫默"
|
||
)
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_article_forward(bilibili, bing_dy_list):
|
||
post = await bilibili.parse(bing_dy_list[4])
|
||
assert (
|
||
post.text
|
||
== "#明日方舟##饼学大厦#\n9.11专栏更新完毕,这还塌了实属没跟新运营对上\n后边除了周日发饼和PV没提及的中文语音,稳了\n别忘了来参加#可露希尔的秘密档案#的主题投票\nhttps://t.bilibili.com/568093580488553786?tab=2"
|
||
+ "\n--------------\n"
|
||
+ "【明日方舟】饼学大厦#12~14(风暴瞭望&玛莉娅·临光&红松林&感谢庆典)9.11更新 更新记录09.11更新:覆盖09.10更新;以及排期更新,猜测周一周五开活动09.10更新:以周五开活动为底,PV/公告调整位置,整体结构更新09.08更新:饼学大厦#12更新,新增一件六星商店服饰(周日发饼)09.06更新:饼学大厦整栋整栋翻新,改为9.16开主线(四日无饼!)09.05凌晨更新:10.13后的排期(两日无饼,鹰角背刺,心狠手辣)前言感谢楪筱祈ぺ的动态-哔哩哔哩 (bilibili.com) 对饼学的贡献!后续排期:9.17【风暴瞭望】、10.01【玛莉娅·临光】复刻、10.1"
|
||
)
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_dynamic_forward(bilibili, bing_dy_list):
|
||
post = await bilibili.parse(bing_dy_list[5])
|
||
assert (
|
||
post.text
|
||
== "饼组主线饼学预测——9.11版\n①今日结果\n9.11 殿堂上的游禽-星极(x,新运营实锤了)\n②后续预测\n9.12 #罗德岛相簿#+#可露希尔的秘密档案#11话\n9.13 六星先锋(执旗手)干员-琴柳\n9.14 宣传策略-空弦+家具\n9.15 轮换池(+中文语音前瞻)\n9.16 停机\n9.17 #罗德岛闲逛部#+新六星EP+EP09·风暴瞭望开启\n9.19 #罗德岛相簿#"
|
||
+ "\n--------------\n"
|
||
+ "#明日方舟#\n【新增服饰】\n//殿堂上的游禽 - 星极\n塞壬唱片偶像企划《闪耀阶梯》特供服饰/殿堂上的游禽。星极自费参加了这项企划,尝试着用大众能接受的方式演绎天空之上的故事。\n\n_____________\n谦逊留给观众,骄傲发自歌喉,此夜,唯我璀璨。 "
|
||
)
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
@respx.mock
|
||
async def test_fetch_new(bilibili, dummy_user_subinfo):
|
||
post_router = respx.get(
|
||
"https://api.vc.bilibili.com/dynamic_svr/v1/dynamic_svr/space_history?host_uid=161775300&offset=0&need_top=0"
|
||
)
|
||
post_router.mock(
|
||
return_value=Response(200, json=get_json("bilibili_strange_post-0.json"))
|
||
)
|
||
target = "161775300"
|
||
res = await bilibili.fetch_new_post(target, [dummy_user_subinfo])
|
||
assert post_router.called
|
||
assert len(res) == 0
|
||
|
||
mock_data = get_json("bilibili_strange_post.json")
|
||
mock_data["data"]["cards"][0]["desc"]["timestamp"] = int(datetime.now().timestamp())
|
||
post_router.mock(return_value=Response(200, json=mock_data))
|
||
res2 = await bilibili.fetch_new_post(target, [dummy_user_subinfo])
|
||
assert len(res2[0][1]) == 1
|
||
post = res2[0][1][0]
|
||
assert (
|
||
post.text
|
||
== "#罗德厨房——回甘##明日方舟#\r\n明日方舟官方美食漫画,正式开餐。\r\n往事如烟,安然即好。\r\nMenu 01:高脚羽兽烤串与罗德岛的领袖\r\n\r\n哔哩哔哩漫画阅读:https://manga.bilibili.com/detail/mc31998?from=manga_search\r\n\r\n关注并转发本动态,我们将会在5月27日抽取10位博士赠送【兔兔奇境】周边礼盒一份。 互动抽奖"
|
||
)
|
||
|
||
|
||
async def test_parse_target(bilibili: "Bilibili"):
|
||
from nonebot_bison.platform.platform import Platform
|
||
|
||
res = await bilibili.parse_target(
|
||
"https://space.bilibili.com/161775300?from=search&seid=130517740606234234234&spm_id_from=333.337.0.0"
|
||
)
|
||
assert res == "161775300"
|
||
res2 = await bilibili.parse_target(
|
||
"space.bilibili.com/161775300?from=search&seid=130517740606234234234&spm_id_from=333.337.0.0"
|
||
)
|
||
assert res2 == "161775300"
|
||
with pytest.raises(Platform.ParseTargetException):
|
||
await bilibili.parse_target(
|
||
"https://www.bilibili.com/video/BV1qP4y1g738?spm_id_from=333.999.0.0"
|
||
)
|