mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-04 02:26:11 +08:00
68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
import respx
|
|
import pytest
|
|
from flaky import flaky
|
|
from nonebug.app import App
|
|
from httpx import Response, AsyncClient
|
|
|
|
from .utils import get_file, get_json
|
|
|
|
|
|
@pytest.fixture()
|
|
def mcbbsnews(app: App):
|
|
from nonebot_bison.utils import ProcessContext
|
|
from nonebot_bison.platform import platform_manager
|
|
|
|
return platform_manager["mcbbsnews"](ProcessContext(), AsyncClient())
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def raw_post_list():
|
|
return get_json("mcbbsnews/mcbbsnews_raw_post_list_update.json")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.render
|
|
@respx.mock
|
|
@flaky(max_runs=3, min_passes=1)
|
|
async def test_fetch_new(mcbbsnews, dummy_user_subinfo, raw_post_list):
|
|
from nonebot_bison.post import Post
|
|
|
|
news_router = respx.get("https://www.mcbbs.net/forum-news-1.html")
|
|
news_router.mock(return_value=Response(200, text=get_file("mcbbsnews/mock/mcbbsnews_post_list_html-0.html")))
|
|
new_post = respx.get("https://www.mcbbs.net/thread-1340927-1-1.html")
|
|
new_post.mock(return_value=Response(200, text=get_file("mcbbsnews/mock/mcbbsnews_new_post_html.html")))
|
|
|
|
target = ""
|
|
res = await mcbbsnews.fetch_new_post(target, [dummy_user_subinfo])
|
|
assert news_router.called
|
|
assert len(res) == 0
|
|
|
|
news_router.mock(return_value=Response(200, text=get_file("mcbbsnews/mock/mcbbsnews_post_list_html-1.html")))
|
|
res1 = await mcbbsnews.fetch_new_post(target, [dummy_user_subinfo])
|
|
assert news_router.called
|
|
post: Post = res1[0][1][0]
|
|
raw_post = raw_post_list[0]
|
|
assert post.platform.name == "MCBBS幻翼块讯"
|
|
assert post.content == "{}\n│\n└由 {} 发表".format(raw_post["title"], raw_post["author"])
|
|
assert post.url == "https://www.mcbbs.net/{}".format(raw_post["url"])
|
|
assert post.nickname == raw_post["category"]
|
|
assert post.images
|
|
assert len(post.images) == 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.render
|
|
@respx.mock
|
|
@flaky(max_runs=3, min_passes=1)
|
|
async def test_news_render(mcbbsnews, dummy_user_subinfo):
|
|
new_post = respx.get("https://www.mcbbs.net/thread-1340927-1-1.html")
|
|
new_post.mock(return_value=Response(200, text=get_file("mcbbsnews/mock/mcbbsnews_new_post_html.html")))
|
|
pics = await mcbbsnews._news_render("https://www.mcbbs.net/thread-1340927-1-1.html", "#post_25849603")
|
|
assert len(pics) == 1
|
|
|
|
pics_err_on_assert = await mcbbsnews._news_render("", "##post_25849603")
|
|
assert len(pics_err_on_assert) == 2
|
|
|
|
pics_err_on_other = await mcbbsnews._news_render("https://www.mcbbs.net/thread-1340927-1-1.html", "#post_err")
|
|
assert len(pics_err_on_other) == 2
|