update dataclass encoder tests

This commit is contained in:
yanyongyu
2022-02-17 15:35:39 +08:00
parent 9c0e05c615
commit 1d6a333b49
2 changed files with 9 additions and 15 deletions

View File

@ -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"