移除示例项目初始化并修复前后端稳定性问题

This commit is contained in:
Cheng Zhou
2026-03-30 15:35:22 +08:00
parent a67991740e
commit 8c5d03f1cb
24 changed files with 581 additions and 102 deletions
+78
View File
@@ -0,0 +1,78 @@
from contextlib import asynccontextmanager
from unittest.mock import AsyncMock
import pytest
from fastapi import FastAPI
from app import main
class _DummyTask:
def __init__(self):
self.awaited = False
def __await__(self):
async def _wait():
self.awaited = True
return _wait().__await__()
class _DummySession:
async def __aenter__(self):
return self
async def __aexit__(self, exc_type, exc, tb):
return False
class _DummyBegin:
async def __aenter__(self):
return self
async def __aexit__(self, exc_type, exc, tb):
return False
async def run_sync(self, fn):
return None
class _DummyEngine:
def begin(self):
return _DummyBegin()
@pytest.mark.asyncio
async def test_development_startup_does_not_seed_demo_fees(monkeypatch):
ensure_admin_exists = AsyncMock()
seed_demo_fees = AsyncMock()
create_all = AsyncMock()
legacy_pk = AsyncMock()
scheduler_task = _DummyTask()
async def fake_scheduler(stop_event):
return None
def fake_create_task(coroutine):
coroutine.close()
return scheduler_task
@asynccontextmanager
async def fake_session_local():
yield _DummySession()
monkeypatch.setattr(main.settings, "ENV", "development")
monkeypatch.setattr(main, "ensure_admin_exists", ensure_admin_exists)
monkeypatch.setattr(main, "seed_demo_fees", seed_demo_fees, raising=False)
monkeypatch.setattr(main, "run_daily_lost_visit_job", fake_scheduler)
monkeypatch.setattr(main.asyncio, "create_task", fake_create_task)
monkeypatch.setattr(main, "engine", _DummyEngine())
monkeypatch.setattr(main, "SessionLocal", fake_session_local)
monkeypatch.setattr(main, "_ensure_legacy_primary_keys", legacy_pk)
monkeypatch.setattr(main.Base.metadata, "create_all", create_all)
async with main.lifespan(FastAPI()):
pass
ensure_admin_exists.assert_awaited_once()
seed_demo_fees.assert_not_awaited()
+105
View File
@@ -0,0 +1,105 @@
import uuid
import pytest
from fastapi import HTTPException
from starlette.requests import Request
from app.core import deps
class _Study:
def __init__(self, study_id: uuid.UUID, is_locked: bool = False):
self.id = study_id
self.is_locked = is_locked
class _Category:
def __init__(self, study_id: uuid.UUID | None):
self.study_id = study_id
class _FaqItem:
def __init__(self, study_id: uuid.UUID | None):
self.study_id = study_id
def _make_request(*, method: str = "POST", path: str = "/", query_string: bytes = b"", json_body: bytes = b"", path_params=None):
async def receive():
return {"type": "http.request", "body": json_body, "more_body": False}
scope = {
"type": "http",
"http_version": "1.1",
"method": method,
"path": path,
"raw_path": path.encode(),
"query_string": query_string,
"headers": [(b"content-type", b"application/json")],
"path_params": path_params or {},
}
return Request(scope, receive)
@pytest.mark.asyncio
async def test_require_study_not_locked_accepts_study_id_from_request_body(monkeypatch):
study_id = uuid.uuid4()
async def fake_get(_db, requested_study_id):
return _Study(requested_study_id, is_locked=False)
monkeypatch.setattr("app.crud.study.get", fake_get)
dependency = deps.require_study_not_locked()
request = _make_request(json_body=f'{{"study_id":"{study_id}"}}'.encode())
study = await dependency(request=request, db=object())
assert study.id == study_id
@pytest.mark.asyncio
async def test_require_study_not_locked_resolves_study_id_from_category_path(monkeypatch):
study_id = uuid.uuid4()
category_id = uuid.uuid4()
async def fake_get_category(_db, requested_category_id):
assert requested_category_id == category_id
return _Category(study_id)
async def fake_get_study(_db, requested_study_id):
return _Study(requested_study_id, is_locked=False)
monkeypatch.setattr("app.crud.faq_category.get_category", fake_get_category)
monkeypatch.setattr("app.crud.study.get", fake_get_study)
dependency = deps.require_study_not_locked()
request = _make_request(path=f"/api/v1/faqs/categories/{category_id}", path_params={"category_id": str(category_id)})
study = await dependency(request=request, db=object())
assert study.id == study_id
@pytest.mark.asyncio
async def test_require_study_not_locked_rejects_locked_study_resolved_from_item_path(monkeypatch):
study_id = uuid.uuid4()
item_id = uuid.uuid4()
async def fake_get_item(_db, requested_item_id):
assert requested_item_id == item_id
return _FaqItem(study_id)
async def fake_get_study(_db, requested_study_id):
return _Study(requested_study_id, is_locked=True)
monkeypatch.setattr("app.crud.faq_item.get_item", fake_get_item)
monkeypatch.setattr("app.crud.study.get", fake_get_study)
dependency = deps.require_study_not_locked()
request = _make_request(path=f"/api/v1/faqs/{item_id}", path_params={"item_id": str(item_id)})
with pytest.raises(HTTPException) as exc_info:
await dependency(request=request, db=object())
assert exc_info.value.status_code == 403
assert "项目已锁定" in exc_info.value.detail