24 lines
572 B
Python
24 lines
572 B
Python
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"
|
|
|