Files
social-app/backend/tests/integration/test_schedule_items_routes.py
T
qzl 50b38de488 feat: add schedule items CRUD API
- Add ScheduleItem Pydantic schemas with metadata support
- Add repository layer with CRUD operations
- Add service layer with authorization
- Add FastAPI router with all endpoints
- Add unit and integration tests
- Update API documentation
2026-02-28 11:34:49 +08:00

206 lines
6.6 KiB
Python

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 core.auth.models import CurrentUser
from v1.schedule_items.dependencies import get_schedule_item_service
from v1.schedule_items.schemas import (
ScheduleItemCreateRequest,
ScheduleItemListRequest,
ScheduleItemResponse,
ScheduleItemSourceType,
ScheduleItemStatus,
ScheduleItemUpdateRequest,
)
from v1.schedule_items.service import ScheduleItemService
class FakeScheduleItemService:
def __init__(self, item: ScheduleItemResponse | None) -> None:
self._item = item
async def create(self, request: ScheduleItemCreateRequest) -> ScheduleItemResponse:
if not self._item:
raise HTTPException(status_code=503, detail="Store unavailable")
return self._item
async def get_by_id(self, item_id: UUID) -> ScheduleItemResponse:
if not self._item or str(self._item.id) != str(item_id):
raise HTTPException(status_code=404, detail="Schedule item not found")
return self._item
async def update(
self, item_id: UUID, request: ScheduleItemUpdateRequest
) -> ScheduleItemResponse:
if not self._item or str(self._item.id) != str(item_id):
raise HTTPException(status_code=404, detail="Schedule item not found")
return self._item
async def delete(self, item_id: UUID) -> None:
if not self._item or str(self._item.id) != str(item_id):
raise HTTPException(status_code=404, detail="Schedule item not found")
async def list_by_date_range(
self, request: ScheduleItemListRequest
) -> list[ScheduleItemResponse]:
return [self._item] if self._item else []
def _override_schedule_item_service(
service: FakeScheduleItemService,
) -> Callable[[], ScheduleItemService]:
def _get_service() -> ScheduleItemService:
return service # type: ignore[return-value]
return _get_service
def _override_current_user(user_id: UUID) -> Callable[[], CurrentUser]:
def _get_user() -> CurrentUser:
return CurrentUser(id=user_id)
return _get_user
def test_create_schedule_item_returns_201() -> None:
item = ScheduleItemResponse(
id=uuid4(),
title="Test Event",
start_at=datetime(2026, 2, 28, 16, 0, 0, tzinfo=timezone.utc),
timezone="UTC",
status=ScheduleItemStatus.ACTIVE,
source_type=ScheduleItemSourceType.MANUAL,
created_at=datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc),
updated_at=datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc),
)
app.dependency_overrides[get_schedule_item_service] = (
_override_schedule_item_service(FakeScheduleItemService(item))
)
client = TestClient(app)
try:
response = client.post(
"/api/v1/schedule-items",
json={
"title": "Test Event",
"start_at": "2026-02-28T16:00:00Z",
},
)
assert response.status_code == 201
finally:
app.dependency_overrides = {}
def test_list_schedule_items_returns_200() -> None:
item = ScheduleItemResponse(
id=uuid4(),
title="Test Event",
start_at=datetime(2026, 2, 28, 16, 0, 0, tzinfo=timezone.utc),
timezone="UTC",
status=ScheduleItemStatus.ACTIVE,
source_type=ScheduleItemSourceType.MANUAL,
created_at=datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc),
updated_at=datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc),
)
app.dependency_overrides[get_schedule_item_service] = (
_override_schedule_item_service(FakeScheduleItemService(item))
)
client = TestClient(app)
try:
response = client.get(
"/api/v1/schedule-items",
params={
"start_at": "2026-02-01T00:00:00Z",
"end_at": "2026-02-28T23:59:59Z",
},
)
assert response.status_code == 200
assert isinstance(response.json(), list)
finally:
app.dependency_overrides = {}
def test_get_schedule_item_returns_200() -> None:
item_id = uuid4()
item = ScheduleItemResponse(
id=item_id,
title="Test Event",
start_at=datetime(2026, 2, 28, 16, 0, 0, tzinfo=timezone.utc),
timezone="UTC",
status=ScheduleItemStatus.ACTIVE,
source_type=ScheduleItemSourceType.MANUAL,
created_at=datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc),
updated_at=datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc),
)
app.dependency_overrides[get_schedule_item_service] = (
_override_schedule_item_service(FakeScheduleItemService(item))
)
client = TestClient(app)
try:
response = client.get(f"/api/v1/schedule-items/{item_id}")
assert response.status_code == 200
finally:
app.dependency_overrides = {}
def test_update_schedule_item_returns_200() -> None:
item_id = uuid4()
item = ScheduleItemResponse(
id=item_id,
title="Updated Event",
start_at=datetime(2026, 2, 28, 16, 0, 0, tzinfo=timezone.utc),
timezone="UTC",
status=ScheduleItemStatus.ACTIVE,
source_type=ScheduleItemSourceType.MANUAL,
created_at=datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc),
updated_at=datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc),
)
app.dependency_overrides[get_schedule_item_service] = (
_override_schedule_item_service(FakeScheduleItemService(item))
)
client = TestClient(app)
try:
response = client.patch(
f"/api/v1/schedule-items/{item_id}",
json={"title": "Updated Event"},
)
assert response.status_code == 200
finally:
app.dependency_overrides = {}
def test_delete_schedule_item_returns_204() -> None:
item_id = uuid4()
item = ScheduleItemResponse(
id=item_id,
title="Test Event",
start_at=datetime(2026, 2, 28, 16, 0, 0, tzinfo=timezone.utc),
timezone="UTC",
status=ScheduleItemStatus.ACTIVE,
source_type=ScheduleItemSourceType.MANUAL,
created_at=datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc),
updated_at=datetime(2026, 2, 27, 10, 0, 0, tzinfo=timezone.utc),
)
app.dependency_overrides[get_schedule_item_service] = (
_override_schedule_item_service(FakeScheduleItemService(item))
)
client = TestClient(app)
try:
response = client.delete(f"/api/v1/schedule-items/{item_id}")
assert response.status_code == 204
finally:
app.dependency_overrides = {}