init
This commit is contained in:
@@ -0,0 +1,30 @@
|
||||
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)
|
||||
|
||||
Reference in New Issue
Block a user