feat: 提交热榜评论分析工具 MVP 基线

2026-07-03 11:12:12 +08:00
parent 0231be6338
commit 9bad7b6e63
60 changed files with 4058 additions and 0 deletions
@@ -0,0 +1,118 @@
+from fastapi.testclient import TestClient
+from sqlalchemy import create_engine
+from sqlalchemy.pool import StaticPool
+
+from app.db import Base, get_db_session
+from app.main import app
+from app.models import Task
+
+
+def make_test_client():
+    engine = create_engine(
+        "sqlite:///:memory:",
+        connect_args={"check_same_thread": False},
+        poolclass=StaticPool,
+    )
+    Base.metadata.create_all(engine)
+
+    def override_db_session():
+        from sqlalchemy.orm import Session
+
+        with Session(engine) as session:
+            yield session
+
+    app.dependency_overrides[get_db_session] = override_db_session
+    return TestClient(app), engine
+
+
+def test_create_task_returns_task_id_and_running_status():
+    client, engine = make_test_client()
+    try:
+        response = client.post(
+            "/api/tasks",
+            json={
+                "platform": "xiaohongshu",
+                "hotspot_limit": 5,
+                "item_limit_per_hotspot": 5,
+                "comment_limit_per_item": 50,
+            },
+        )
+
+        assert response.status_code == 201
+        data = response.json()
+        assert data["task_id"]
+        assert data["status"] == "running"
+    finally:
+        engine.dispose()
+        app.dependency_overrides.clear()
+
+
+def test_create_task_rejects_invalid_platform_and_limits():
+    client, engine = make_test_client()
+    try:
+        response = client.post(
+            "/api/tasks",
+            json={
+                "platform": "weibo",
+                "hotspot_limit": 11,
+                "item_limit_per_hotspot": 0,
+                "comment_limit_per_item": 101,
+            },
+        )
+
+        assert response.status_code == 422
+    finally:
+        engine.dispose()
+        app.dependency_overrides.clear()
+
+
+def test_create_task_returns_400_when_running_task_exists():
+    client, engine = make_test_client()
+    try:
+        from sqlalchemy.orm import Session
+
+        with Session(engine) as session:
+            session.add(Task(platform="douyin", status="running"))
+            session.commit()
+
+        response = client.post(
+            "/api/tasks",
+            json={
+                "platform": "douyin",
+                "hotspot_limit": 5,
+                "item_limit_per_hotspot": 5,
+                "comment_limit_per_item": 50,
+            },
+        )
+
+        assert response.status_code == 400
+        assert response.json() == {"detail": "当前有正在运行的任务，请稍后再试"}
+    finally:
+        engine.dispose()
+        app.dependency_overrides.clear()
+
+
+def test_task_list_and_detail_return_created_tasks():
+    client, engine = make_test_client()
+    try:
+        created = client.post(
+            "/api/tasks",
+            json={
+                "platform": "douyin",
+                "hotspot_limit": 3,
+                "item_limit_per_hotspot": 4,
+                "comment_limit_per_item": 30,
+            },
+        ).json()
+
+        list_response = client.get("/api/tasks")
+        detail_response = client.get(f"/api/tasks/{created['task_id']}")
+
+        assert list_response.status_code == 200
+        assert list_response.json()[0]["task_id"] == created["task_id"]
+        assert detail_response.status_code == 200
+        assert detail_response.json()["platform"] == "douyin"
+        assert detail_response.json()["hotspot_limit"] == 3
+    finally:
+        engine.dispose()
+        app.dependency_overrides.clear()