Kris
6c95dc006a
完成渠道模块的单元测试目录搭建，新增多个领域模型、端口、中间件、事件、服务以及基础设施层的单元测试文件，同时补充了conftest.py的环境变量配置，完善测试基础环境。
108 lines
3.4 KiB
Python
108 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from yuxi.channel.domain.model.message.unified_message import UnifiedMessage
|
|
from yuxi.channel.domain.model.message.peer import Peer
|
|
from yuxi.channel.domain.model.message.attachment import Attachment
|
|
|
|
|
|
class TestUnifiedMessage:
|
|
def test_basic_creation(self):
|
|
msg = UnifiedMessage(
|
|
message_id="msg123",
|
|
channel_type="web",
|
|
sender=Peer(id="user1", name="User"),
|
|
content="Hello",
|
|
)
|
|
assert msg.message_id == "msg123"
|
|
assert msg.channel_type == "web"
|
|
assert msg.content == "Hello"
|
|
|
|
def test_to_dict(self):
|
|
msg = UnifiedMessage(
|
|
message_id="msg123",
|
|
channel_type="web",
|
|
sender=Peer(id="user1", name="User"),
|
|
content="Hello",
|
|
metadata={"key": "value"},
|
|
)
|
|
d = msg.to_dict()
|
|
assert d["message_id"] == "msg123"
|
|
assert d["content"] == "Hello"
|
|
assert d["metadata"] == {"key": "value"}
|
|
|
|
def test_from_dict(self):
|
|
d = {
|
|
"message_id": "msg123",
|
|
"channel_type": "web",
|
|
"sender": {"id": "user1", "name": "User"},
|
|
"content": "Hello",
|
|
"metadata": {"key": "value"},
|
|
}
|
|
msg = UnifiedMessage.from_dict(d)
|
|
assert msg.message_id == "msg123"
|
|
assert msg.sender.id == "user1"
|
|
assert msg.metadata == {"key": "value"}
|
|
|
|
def test_with_attachments(self):
|
|
msg = UnifiedMessage(
|
|
message_id="msg123",
|
|
channel_type="web",
|
|
sender=Peer(id="user1", name="User"),
|
|
content="Hello",
|
|
attachments=[
|
|
Attachment(url="http://example.com/image.png", media_type="image", filename="image.png"),
|
|
],
|
|
)
|
|
assert len(msg.attachments) == 1
|
|
assert msg.attachments[0].media_type == "image"
|
|
|
|
def test_empty_message(self):
|
|
msg = UnifiedMessage(
|
|
message_id="",
|
|
channel_type="",
|
|
sender=Peer(id="", name=""),
|
|
content="",
|
|
)
|
|
assert msg.message_id == ""
|
|
assert msg.content == ""
|
|
|
|
|
|
class TestPeer:
|
|
def test_creation(self):
|
|
peer = Peer(id="user1", name="User")
|
|
assert peer.id == "user1"
|
|
assert peer.name == "User"
|
|
|
|
def test_to_dict(self):
|
|
peer = Peer(id="user1", name="User")
|
|
d = peer.to_dict()
|
|
assert d == {"id": "user1", "name": "User"}
|
|
|
|
def test_from_dict(self):
|
|
d = {"id": "user1", "name": "User"}
|
|
peer = Peer.from_dict(d)
|
|
assert peer.id == "user1"
|
|
assert peer.name == "User"
|
|
|
|
|
|
class TestAttachment:
|
|
def test_creation(self):
|
|
att = Attachment(url="http://example.com/file.pdf", media_type="document", filename="file.pdf")
|
|
assert att.url == "http://example.com/file.pdf"
|
|
assert att.media_type == "document"
|
|
assert att.filename == "file.pdf"
|
|
|
|
def test_to_dict(self):
|
|
att = Attachment(url="http://example.com/file.pdf", media_type="document", filename="file.pdf")
|
|
d = att.to_dict()
|
|
assert d["url"] == "http://example.com/file.pdf"
|
|
assert d["media_type"] == "document"
|
|
|
|
def test_from_dict(self):
|
|
d = {"url": "http://example.com/file.pdf", "media_type": "document", "filename": "file.pdf"}
|
|
att = Attachment.from_dict(d)
|
|
assert att.url == "http://example.com/file.pdf"
|
|
assert att.filename == "file.pdf"
|