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)