mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-07-10 16:43:00 +08:00
112 lines
3.3 KiB
Python
112 lines
3.3 KiB
Python
from time import time
|
|
from typing import Any
|
|
|
|
import pytest
|
|
from nonebug.app import App
|
|
from httpx import AsyncClient
|
|
|
|
now = time()
|
|
passed = now - 3 * 60 * 60
|
|
|
|
raw_post_list_1 = [{"id": 1, "text": "p1", "date": now, "tags": ["tag1"], "category": 1}]
|
|
|
|
raw_post_list_2 = raw_post_list_1 + [
|
|
{"id": 2, "text": "p2", "date": now, "tags": ["tag1"], "category": 1},
|
|
{"id": 3, "text": "p3", "date": now, "tags": ["tag2"], "category": 2},
|
|
{"id": 4, "text": "p4", "date": now, "tags": ["tag2"], "category": 3},
|
|
]
|
|
|
|
|
|
@pytest.fixture()
|
|
def mock_platform(app: App):
|
|
from nonebot_bison.post import Post
|
|
from nonebot_bison.types import Target, RawPost
|
|
from nonebot_bison.platform.platform import NewMessage
|
|
|
|
class MockPlatform(NewMessage):
|
|
platform_name = "mock_platform"
|
|
name = "Mock Platform"
|
|
enabled = True
|
|
is_common = True
|
|
schedule_interval = 10
|
|
enable_tag = False
|
|
categories = {}
|
|
has_target = True
|
|
|
|
sub_index = 0
|
|
|
|
@classmethod
|
|
async def get_target_name(cls, client, _: "Target"):
|
|
return "MockPlatform"
|
|
|
|
def get_id(self, post: "RawPost") -> Any:
|
|
return post["id"]
|
|
|
|
def get_date(self, raw_post: "RawPost") -> float:
|
|
return raw_post["date"]
|
|
|
|
async def parse(self, raw_post: "RawPost") -> "Post":
|
|
return Post(
|
|
self,
|
|
raw_post["text"],
|
|
url="http://t.tt/" + str(self.get_id(raw_post)),
|
|
nickname="Mock",
|
|
)
|
|
|
|
@classmethod
|
|
async def get_sub_list(cls, _: "Target"):
|
|
if cls.sub_index == 0:
|
|
cls.sub_index += 1
|
|
return raw_post_list_1
|
|
else:
|
|
return raw_post_list_2
|
|
|
|
return MockPlatform
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_msg(mock_platform):
|
|
from nonebot_plugin_saa import Text, Image
|
|
|
|
from nonebot_bison.post import Post
|
|
from nonebot_bison.utils import ProcessContext
|
|
from nonebot_bison.plugin_config import plugin_config
|
|
|
|
post: Post = await mock_platform(ProcessContext(), AsyncClient()).parse(raw_post_list_1[0])
|
|
assert post.platform.default_theme == "basic"
|
|
res = await post.generate()
|
|
assert len(res) == 1
|
|
assert isinstance(res[0], Text)
|
|
assert str(res[0]) == "p1\n来源: Mock Platform Mock\n详情: http://t.tt/1"
|
|
|
|
post.platform.default_theme = "ht2i"
|
|
assert post.get_config_theme() is None
|
|
assert set(post.get_priority_themes()) == {"basic", "ht2i"}
|
|
assert post.get_priority_themes()[0] == "ht2i"
|
|
res1 = await post.generate()
|
|
assert isinstance(res1[0], Image)
|
|
|
|
plugin_config.bison_theme_use_browser = False
|
|
|
|
res3 = await post.generate()
|
|
assert res3[0]
|
|
assert isinstance(res3[0], Text)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.render
|
|
async def test_msg_segments_convert(mock_platform):
|
|
from nonebot_plugin_saa import Image
|
|
|
|
from nonebot_bison.post import Post
|
|
from nonebot_bison.utils import ProcessContext
|
|
from nonebot_bison.plugin_config import plugin_config
|
|
|
|
plugin_config.bison_use_pic = True
|
|
|
|
post: Post = await mock_platform(ProcessContext(), AsyncClient()).parse(raw_post_list_1[0])
|
|
assert post.platform.default_theme == "basic"
|
|
res = await post.generate_messages()
|
|
assert len(res) == 1
|
|
assert isinstance(res[0][0], Image)
|