from schemas.messages import MessagesDTO


def test_messages_schema_matches_required_fields():
    dto = MessagesDTO(
        message_id="mid-1",
        conversation_id="cid-1",
        user="tester",
        query="hello",
        answer="world",
        feedback=None,
        feedback_content=None,
        created_at=1,
        updated_at=1,
        log={"trace_id": "t-1"},
    )

    assert dto.message_id == "mid-1"
    assert dto.conversation_id == "cid-1"
    assert dto.query == "hello"
    assert dto.answer == "world"
    assert dto.log["trace_id"] == "t-1"