import typing from datetime import datetime import respx import pytest import feedparser from pytz import timezone from nonebug.app import App from httpx import Response, AsyncClient from .utils import get_file, get_json if typing.TYPE_CHECKING: from nonebot_bison.platform.weibo import Weibo image_cdn_router = respx.route(host__regex=r"wx\d.sinaimg.cn", path__startswith="/large/") @pytest.fixture() def weibo(app: App): from nonebot_bison.utils import ProcessContext from nonebot_bison.platform import platform_manager return platform_manager["weibo"](ProcessContext(), AsyncClient()) @pytest.fixture(scope="module") def weibo_ak_list_1(): return get_json("weibo_ak_list_1.json") @pytest.mark.asyncio @respx.mock async def test_get_name(weibo): profile_router = respx.get("https://m.weibo.cn/api/container/getIndex?containerid=1005056279793937") profile_router.mock(return_value=Response(200, json=get_json("weibo_ak_profile.json"))) name = await weibo.get_target_name(AsyncClient(), "6279793937") assert name == "明日方舟Arknights" @pytest.mark.asyncio @respx.mock async def test_fetch_new(weibo, dummy_user_subinfo): from nonebot_bison.post import Post from nonebot_bison.types import Target, SubUnit ak_list_router = respx.get("https://m.weibo.cn/api/container/getIndex?containerid=1076036279793937") detail_router = respx.get("https://m.weibo.cn/detail/4649031014551911") ak_list_router.mock(return_value=Response(200, json=get_json("weibo_ak_list_0.json"))) detail_router.mock(return_value=Response(200, text=get_file("weibo_detail_4649031014551911"))) image_cdn_router.mock(Response(200, content=b"")) target = Target("6279793937") res = await weibo.fetch_new_post(SubUnit(target, [dummy_user_subinfo])) assert ak_list_router.called assert len(res) == 0 assert not detail_router.called mock_data = get_json("weibo_ak_list_1.json") ak_list_router.mock(return_value=Response(200, json=mock_data)) res2 = await weibo.fetch_new_post(SubUnit(target, [dummy_user_subinfo])) assert len(res2) == 0 mock_data["data"]["cards"][1]["mblog"]["created_at"] = datetime.now(timezone("Asia/Shanghai")).strftime( "%a %b %d %H:%M:%S %z %Y" ) ak_list_router.mock(return_value=Response(200, json=mock_data)) res3 = await weibo.fetch_new_post(SubUnit(target, [dummy_user_subinfo])) assert len(res3[0][1]) == 1 assert not detail_router.called post: Post = res3[0][1][0] assert post.platform.platform_name == "weibo" assert post.content == "#明日方舟#\nSideStory「沃伦姆德的薄暮」复刻现已开启！ " assert post.url == "https://weibo.com/6279793937/KkBtUx2dv" assert post.nickname == "明日方舟Arknights" assert post.images assert len(post.images) == 1 @pytest.mark.asyncio async def test_classification(weibo): mock_data = get_json("weibo_ak_list_1.json") tuwen = mock_data["data"]["cards"][1] retweet = mock_data["data"]["cards"][3] video = mock_data["data"]["cards"][0] mock_data_ys = get_json("weibo_ys_list_0.json") text = mock_data_ys["data"]["cards"][2] assert weibo.get_category(retweet) == 1 assert weibo.get_category(video) == 2 assert weibo.get_category(tuwen) == 3 assert weibo.get_category(text) == 4 @pytest.mark.asyncio @respx.mock async def test_parse_long(weibo): detail_router = respx.get("https://m.weibo.cn/detail/4645748019299849") detail_router.mock(return_value=Response(200, text=get_file("weibo_detail_4645748019299849"))) raw_post = get_json("weibo_ak_list_1.json")["data"]["cards"][0] post = await weibo.parse(raw_post) assert "全文" not in post.content assert detail_router.called def test_tag(weibo, weibo_ak_list_1): raw_post = weibo_ak_list_1["data"]["cards"][0] assert weibo.get_tags(raw_post) == ["明日方舟", "音律联觉"] @pytest.mark.asyncio @pytest.mark.compare async def test_rsshub_compare(weibo): from nonebot_bison.types import Target target = Target("6279793937") raw_posts = filter(weibo.filter_platform_custom, await weibo.get_sub_list(target)) posts = [] for raw_post in raw_posts: posts.append(await weibo.parse(raw_post)) url_set = {x.url for x in posts} feedres = feedparser.parse("https://rsshub.app/weibo/user/6279793937") for entry in feedres.entries[:5]: # print(entry) assert entry.link in url_set test_post = { "mblog": { "text": ( # noqa '#刚出生的小羊驼长啥样#
小羊驼三三来也[好喜欢]'
小羊驼三三 ' ), "bid": "KnssqeqKK", } } def test_chaohua_tag(weibo): tags = weibo.get_tags(test_post) assert "刚出生的小羊驼长啥样" in tags assert "小羊驼三三超话" in tags async def test_parse_target(weibo: "Weibo"): from nonebot_bison.platform.platform import Platform res = await weibo.parse_target("https://weibo.com/u/6441489862") assert res == "6441489862" res = await weibo.parse_target("weibo.com/u/6441489862") assert res == "6441489862" res = await weibo.parse_target("6441489862") assert res == "6441489862" with pytest.raises(Platform.ParseTargetException): await weibo.parse_target("https://weibo.com/arknights")