40 lines
1.6 KiB
Python
40 lines
1.6 KiB
Python
from sqlalchemy.orm import Session
|
|
|
|
from app.models import Comment, Task
|
|
from app.platforms.base import CommentData, ContentItemData, HotspotData
|
|
from app.schemas import CreateTaskRequest
|
|
from app.services.task_service import create_task, run_task
|
|
from tests.helpers import make_test_client
|
|
|
|
|
|
class FakeDouyinPlatform:
|
|
def fetch_hotspots(self, *, limit):
|
|
return [HotspotData(source_hot_id="d1", title="抖音热点", rank=1, heat_value="200", raw_data={})]
|
|
|
|
def search_items_by_hotspot(self, keyword, *, limit):
|
|
return [ContentItemData(source_item_id="v1", item_type="video", title="视频", raw_data={})]
|
|
|
|
def fetch_comments(self, source_item_id, *, limit):
|
|
return [CommentData(source_comment_id="dc1", content="抖音评论", raw_data={})]
|
|
|
|
|
|
def test_douyin_task_flow_persists_comments(monkeypatch):
|
|
with make_test_client() as (_client, engine):
|
|
monkeypatch.setattr("app.services.task_service.build_platform", lambda _platform: FakeDouyinPlatform())
|
|
|
|
with Session(engine) as session:
|
|
task = create_task(
|
|
session,
|
|
CreateTaskRequest(platform="douyin", hotspot_limit=1, item_limit_per_hotspot=1, comment_limit_per_item=10),
|
|
submit_background=False,
|
|
)
|
|
task_id = task.id
|
|
|
|
with Session(engine) as session:
|
|
run_task(task_id, session_factory=lambda: session)
|
|
task = session.get(Task, task_id)
|
|
comment = session.query(Comment).filter_by(task_id=task_id).one()
|
|
|
|
assert task.status == "success"
|
|
assert comment.content == "抖音评论"
|