31 lines
628 B
Python
31 lines
628 B
Python
|
from schemas.chat_message_response import ChatMessageResponseDTO
|
||
|
|
|
||
|
|
|
||
|
|
EXPECTED_KEYS = [
|
||
|
|
"id",
|
||
|
|
"event",
|
||
|
|
"task_id",
|
||
|
|
"message_id",
|
||
|
|
"conversation_id",
|
||
|
|
"answer",
|
||
|
|
"created_at",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def test_chat_message_response_matches_java_dto_shape():
|
||
|
|
dto = ChatMessageResponseDTO(
|
||
|
|
id="id-1",
|
||
|
|
task_id="task-1",
|
||
|
|
message_id="msg-1",
|
||
|
|
conversation_id="cid-1",
|
||
|
|
answer="hello",
|
||
|
|
created_at=1705395332,
|
||
|
|
)
|
||
|
|
|
||
|
|
dumped = dto.model_dump()
|
||
|
|
|
||
|
|
assert list(dumped.keys()) == EXPECTED_KEYS
|
||
|
|
assert dumped["event"] == "message"
|
||
|
|
assert isinstance(dumped["created_at"], int)
|
||
|
|
|