social-app/backend/tests/integration/test_inbox_messages_routes.py

from __future__ import annotations

from datetime import datetime, timezone
from typing import Callable
from uuid import UUID, uuid4

from fastapi import HTTPException
from fastapi.testclient import TestClient

from app import app
from v1.inbox_messages.dependencies import get_inbox_message_service
from v1.inbox_messages.schemas import (
    InboxMessageAcceptRequest,
    InboxMessageListRequest,
    InboxMessageResponse,
    InboxMessageStatus,
    InboxMessageType,
)
from v1.inbox_messages.service import InboxMessageService


class FakeInboxMessageService:
    def __init__(
        self,
        messages: list[InboxMessageResponse],
        accepted: InboxMessageResponse,
        dismissed: InboxMessageResponse,
    ) -> None:
        self._messages = messages
        self._accepted = accepted
        self._dismissed = dismissed

    async def list_messages(
        self, request: InboxMessageListRequest
    ) -> list[InboxMessageResponse]:
        if request.status is None:
            return self._messages
        return [
            message for message in self._messages if message.status == request.status
        ]

    async def accept_invitation(
        self,
        message_id: UUID,
        request: InboxMessageAcceptRequest,
    ) -> InboxMessageResponse:
        if message_id != self._accepted.id:
            raise HTTPException(status_code=404, detail="Inbox message not found")
        if not request.permission_view:
            raise HTTPException(status_code=400, detail="permission_view is required")
        return self._accepted

    async def dismiss_invitation(self, message_id: UUID) -> InboxMessageResponse:
        if message_id != self._dismissed.id:
            raise HTTPException(status_code=404, detail="Inbox message not found")
        return self._dismissed


def _override_inbox_message_service(
    service: FakeInboxMessageService,
) -> Callable[[], InboxMessageService]:
    def _get_service() -> InboxMessageService:
        return service  # type: ignore[return-value]

    return _get_service


def _build_message(
    message_id: UUID,
    status: InboxMessageStatus,
) -> InboxMessageResponse:
    return InboxMessageResponse(
        id=message_id,
        recipient_id=uuid4(),
        sender_id=uuid4(),
        message_type=InboxMessageType.CALENDAR,
        schedule_item_id=uuid4(),
        content='{"permission": 1}',
        is_read=False,
        status=status,
        created_at=datetime(2026, 2, 28, 9, 0, 0, tzinfo=timezone.utc),
    )


def test_list_inbox_messages_returns_200() -> None:
    pending_message = _build_message(uuid4(), InboxMessageStatus.PENDING)
    accepted_message = _build_message(uuid4(), InboxMessageStatus.ACCEPTED)
    service = FakeInboxMessageService(
        messages=[pending_message, accepted_message],
        accepted=accepted_message,
        dismissed=_build_message(uuid4(), InboxMessageStatus.DISMISSED),
    )
    app.dependency_overrides[get_inbox_message_service] = (
        _override_inbox_message_service(service)
    )

    client = TestClient(app)
    try:
        response = client.get("/api/v1/inbox/messages", params={"status": "pending"})
        assert response.status_code == 200
        body = response.json()
        assert len(body) == 1
        assert body[0]["status"] == "pending"
    finally:
        app.dependency_overrides = {}


def test_accept_inbox_message_returns_200() -> None:
    accepted_message = _build_message(uuid4(), InboxMessageStatus.ACCEPTED)
    service = FakeInboxMessageService(
        messages=[accepted_message],
        accepted=accepted_message,
        dismissed=_build_message(uuid4(), InboxMessageStatus.DISMISSED),
    )
    app.dependency_overrides[get_inbox_message_service] = (
        _override_inbox_message_service(service)
    )

    client = TestClient(app)
    try:
        response = client.post(
            f"/api/v1/inbox/messages/{accepted_message.id}/accept",
            json={
                "permission_view": True,
                "permission_edit": True,
                "permission_invite": False,
            },
        )
        assert response.status_code == 200
        body = response.json()
        assert body["id"] == str(accepted_message.id)
        assert body["status"] == "accepted"
    finally:
        app.dependency_overrides = {}


def test_dismiss_inbox_message_returns_200() -> None:
    dismissed_message = _build_message(uuid4(), InboxMessageStatus.DISMISSED)
    service = FakeInboxMessageService(
        messages=[dismissed_message],
        accepted=_build_message(uuid4(), InboxMessageStatus.ACCEPTED),
        dismissed=dismissed_message,
    )
    app.dependency_overrides[get_inbox_message_service] = (
        _override_inbox_message_service(service)
    )

    client = TestClient(app)
    try:
        response = client.post(f"/api/v1/inbox/messages/{dismissed_message.id}/dismiss")
        assert response.status_code == 200
        body = response.json()
        assert body["id"] == str(dismissed_message.id)
        assert body["status"] == "dismissed"
    finally:
        app.dependency_overrides = {}