diff --git a/tests/test_utils.py b/tests/test_utils.py index 66f8b1ee..282b2ab7 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -6,14 +6,14 @@ from utils import make_fake_message def test_dataclass_encoder(): from nonebot.utils import DataclassEncoder - MessageSegment = make_fake_message().get_segment_class() - ms = MessageSegment.node_custom( - "1234", "testtest", "test" + MessageSegment.image("url") - ) + simple = json.dumps("123", cls=DataclassEncoder) + assert simple == '"123"' + + Message = make_fake_message() + MessageSegment = Message.get_segment_class() + ms = MessageSegment.nested(Message(MessageSegment.text("text"))) s = json.dumps(ms, cls=DataclassEncoder) assert ( s - == '{"type": "node", "data": {"user_id": "1234", "nickname": "testtest", "content": [{"type": "text", "data": {"text": "test"}}, {"type": "image", "data": {"url": "url"}}]}}' + == '{"type": "node", "data": {"content": [{"type": "text", "data": {"text": "text"}}]}}' ) - s1 = json.dumps("123", cls=DataclassEncoder) - assert s1 == '"123"' diff --git a/tests/utils.py b/tests/utils.py index e33ce5be..31939be7 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -33,14 +33,8 @@ def make_fake_message(): return FakeMessageSegment("image", {"url": url}) @staticmethod - def node_custom( - user_id: str, - nickname: str, - content: Union["FakeMessage", "FakeMessageSegment", str], - ): - return FakeMessageSegment( - "node", {"user_id": user_id, "nickname": nickname, "content": content} - ) + def nested(content: "FakeMessage"): + return FakeMessageSegment("node", {"content": content}) def is_text(self) -> bool: return self.type == "text"