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

tests/unit/test_models.py

@@ -0,0 +1,88 @@
+import pytest
+from sqlalchemy import create_engine
+from sqlalchemy.exc import IntegrityError
+from sqlalchemy.orm import Session
+
+from app.db import Base, create_sqlite_engine
+from app.models import Comment, ContentItem, Hotspot, Report, Task, create_report_record
+
+
+def test_all_t03_tables_can_be_created_in_memory_sqlite():
+    engine = create_engine("sqlite:///:memory:", connect_args={"check_same_thread": False})
+    try:
+        Base.metadata.create_all(engine)
+
+        assert set(Base.metadata.tables) >= {
+            "tasks",
+            "hotspots",
+            "content_items",
+            "comments",
+            "reports",
+        }
+    finally:
+        engine.dispose()
+
+
+def test_sqlite_engine_uses_required_connection_options():
+    engine = create_sqlite_engine("sqlite:///data/app.db")
+    try:
+        assert engine.url.database == "data/app.db"
+        assert engine.dialect.connect_args["check_same_thread"] is False
+        assert engine.dialect.connect_args["timeout"] == 10
+
+    finally:
+        engine.dispose()
+
+
+def test_task_status_rejects_partial_status_values():
+    engine = create_engine("sqlite:///:memory:", connect_args={"check_same_thread": False})
+    try:
+        Base.metadata.create_all(engine)
+
+        with Session(engine) as session:
+            session.add(Task(platform="douyin", status="partial_success"))
+
+            with pytest.raises(IntegrityError):
+                session.commit()
+    finally:
+        engine.dispose()
+
+
+def test_analysis_status_insufficient_does_not_change_task_status():
+    task = Task(
+        platform="xiaohongshu",
+        status="success",
+        analysis_status="insufficient",
+        analysis_success_rate=0.5,
+    )
+
+    assert task.status == "success"
+    assert task.analysis_status == "insufficient"
+
+
+def test_models_define_relationships_and_indexes():
+    assert any(index.name == "ix_comments_task_content_item" for index in Comment.__table__.indexes)
+    assert any(index.name == "ix_content_items_task_hotspot" for index in ContentItem.__table__.indexes)
+    assert any(index.name == "ix_reports_task_report_type" for index in Report.__table__.indexes)
+
+    assert Hotspot.task.property.mapper.class_ is Task
+    assert ContentItem.hotspot.property.mapper.class_ is Hotspot
+    assert Comment.content_item.property.mapper.class_ is ContentItem
+    assert Report.task.property.mapper.class_ is Task
+
+
+def test_report_record_application_constraint_requires_matching_owner_id():
+    with pytest.raises(ValueError, match="hotspot_id"):
+        create_report_record(task_id="task-1", report_type="hotspot", title="热点报告")
+
+    with pytest.raises(ValueError, match="content_item_id"):
+        create_report_record(task_id="task-1", report_type="content_item", title="内容报告")
+
+    report = create_report_record(
+        task_id="task-1",
+        report_type="hotspot",
+        title="热点报告",
+        hotspot_id="hot-1",
+    )
+
+    assert report.hotspot_id == "hot-1"