77 lines
3.0 KiB
Python
77 lines
3.0 KiB
Python
from datetime import UTC, datetime
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.db import Base
|
|
from app.models import Comment, ContentItem, Hotspot, Task
|
|
from app.services.report_service import build_comment_metrics, generate_item_report
|
|
|
|
|
|
def test_build_comment_metrics_counts_sentiments_and_top_labels():
|
|
comments = [
|
|
Comment(content="好", sentiment="positive", labels='["质量好", "价格好"]', like_count=5),
|
|
Comment(content="差", sentiment="negative", labels='["质量好"]', like_count=2),
|
|
Comment(content="一般", sentiment="neutral", labels="[]", like_count=1),
|
|
Comment(content="未知", sentiment="unknown", labels="[]", like_count=0),
|
|
]
|
|
|
|
metrics = build_comment_metrics(comments)
|
|
|
|
assert metrics["sample_count"] == 4
|
|
assert metrics["sentiment"]["positive"]["count"] == 1
|
|
assert metrics["sentiment"]["negative"]["count"] == 1
|
|
assert metrics["sentiment"]["neutral"]["count"] == 1
|
|
assert metrics["sentiment"]["unknown"]["count"] == 1
|
|
assert metrics["top_labels"][0] == {"name": "质量好", "count": 2}
|
|
|
|
|
|
def test_generate_item_report_persists_markdown_and_default_summary():
|
|
engine = create_engine("sqlite:///:memory:", connect_args={"check_same_thread": False})
|
|
Base.metadata.create_all(engine)
|
|
try:
|
|
with Session(engine) as session:
|
|
task = Task(platform="douyin", status="success")
|
|
session.add(task)
|
|
session.flush()
|
|
hotspot = Hotspot(task_id=task.id, platform="douyin", title="热点", raw_data="{}")
|
|
session.add(hotspot)
|
|
session.flush()
|
|
item = ContentItem(
|
|
task_id=task.id,
|
|
hotspot_id=hotspot.id,
|
|
platform="douyin",
|
|
source_item_id="v1",
|
|
item_type="video",
|
|
title="视频",
|
|
status="success",
|
|
raw_data="{}",
|
|
)
|
|
session.add(item)
|
|
session.flush()
|
|
session.add(
|
|
Comment(
|
|
task_id=task.id,
|
|
hotspot_id=hotspot.id,
|
|
content_item_id=item.id,
|
|
platform="douyin",
|
|
source_comment_id="c1",
|
|
content="好评",
|
|
sentiment="positive",
|
|
labels='["认可"]',
|
|
like_count=10,
|
|
comment_time=datetime(2026, 1, 1, tzinfo=UTC),
|
|
raw_data="{}",
|
|
)
|
|
)
|
|
session.commit()
|
|
|
|
report = generate_item_report(session, item.id, summary_provider=lambda *_args, **_kwargs: (_ for _ in ()).throw(RuntimeError("ai failed")))
|
|
|
|
assert report.report_type == "item"
|
|
assert report.content_item_id == item.id
|
|
assert "总结生成失败，请查看上方统计数据。" in report.markdown_content
|
|
assert "样本评论数量" in report.markdown_content
|
|
finally:
|
|
engine.dispose()
|