social-app/backend/tests/unit/v1/inbox_messages/test_schemas.py

from datetime import datetime, timezone
from uuid import uuid4

from v1.inbox_messages.schemas import (
    InboxMessageResponse,
    InboxMessageStatus,
    InboxMessageType,
)


def test_inbox_message_response_schema() -> None:
    msg_id = uuid4()
    response = InboxMessageResponse(
        id=msg_id,
        recipient_id=uuid4(),
        sender_id=uuid4(),
        message_type=InboxMessageType.CALENDAR,
        schedule_item_id=uuid4(),
        content={"type": "invite", "permission": 1, "action": "pending"},
        is_read=False,
        status=InboxMessageStatus.PENDING,
        created_at=datetime.now(timezone.utc),
    )

    assert response.message_type.value == "calendar"
    assert response.status.value == "pending"