nonebot-bison/tests/platforms/test_bilibili.py

import typing
from datetime import datetime

import respx
import pytest
from nonebug.app import App
from httpx import Response, AsyncClient

from .utils import 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.utils import ProcessContext
    from nonebot_bison.platform import platform_manager

    return platform_manager["bilibili"](ProcessContext(), AsyncClient())


@pytest.fixture()
def without_dynamic(app: App):
    return {
        "code": 0,
        "msg": "",
        "message": "",
        "data": {
            "has_more": 0,
            "next_offset": 0,
            "_gt_": 0,
        },
    }


@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提供者：赫默\n=================\n《可露希尔的秘密档案》11话：来宿舍休息一下吧"
    )


@pytest.mark.asyncio
async def test_video_forward_without_dynamic(bilibili, bing_dy_list):
    # 视频简介和动态文本其中一方为空的情况
    post = await bilibili.parse(bing_dy_list[2])
    assert (
        post.text
        == "阿消的罗德岛闲谈直播#01:《女人最喜欢的女人，就是在战场上熠熠生辉的女人》"
        + "\n\n"
        + "本系列视频为饼组成员的有趣直播录播，主要内容为方舟相关，未来可能系列其他视频会包含部分饼组团建日常等。"
        "仅为娱乐性视频，内容与常规饼学预测无关。视频仅为当期主播主观观点，不代表饼组观点。仅供娱乐。"
        "\n\n直播主播:@寒蝉慕夏 \n后期剪辑:@Melodiesviel \n\n本群视频为9.11组员慕夏直播录播，"
        "包含慕夏对新PV的个人解读，风笛厨力疯狂放出，CP言论输出，9.16轮换池预测视频分析和理智规划杂谈内容。"
        "\n注意:内含大量个人性质对风笛的厨力观点，与多CP混乱发言，不适者请及时点击退出或跳到下一片段。"
    )


@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_without_dynamic(bilibili, dummy_user_subinfo, without_dynamic):
    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=without_dynamic))
    bilibili_main_page_router = respx.get("https://www.bilibili.com/")
    bilibili_main_page_router.mock(return_value=Response(200))
    target = "161775300"
    res = await bilibili.fetch_new_post(target, [dummy_user_subinfo])
    assert post_router.called
    assert len(res) == 0


@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")))
    bilibili_main_page_router = respx.get("https://www.bilibili.com/")
    bilibili_main_page_router.mock(return_value=Response(200))
    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")


@pytest.fixture(scope="module")
def post_list():
    return get_json("bilibili_fake_dy_list.json")["data"]["cards"]


# 测试新tag机制的平台推送情况
@pytest.mark.asyncio
async def test_filter_user_custom(bilibili, post_list):
    only_banned_tags = ["~可露希尔的秘密档案"]
    res0 = await bilibili.filter_user_custom(post_list, [], only_banned_tags)
    assert len(res0) == 8

    only_subscribed_tags = ["可露希尔的秘密档案"]
    res1 = await bilibili.filter_user_custom(post_list, [], only_subscribed_tags)
    assert len(res1) == 4

    multi_subs_tags_1 = ["可露希尔的秘密档案", "罗德岛相簿"]
    res2 = await bilibili.filter_user_custom(post_list, [], multi_subs_tags_1)
    assert len(res2) == 4

    multi_subs_tags_2 = ["罗德岛相簿", "风暴瞭望"]
    res3 = await bilibili.filter_user_custom(post_list, [], multi_subs_tags_2)
    assert len(res3) == 4

    multi_subs_tags_3 = ["明日方舟", "~饼学大厦"]
    res4 = await bilibili.filter_user_custom(post_list, [], multi_subs_tags_3)
    assert len(res4) == 2