60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
"""Story 变更触发章节物化：调用链与 Celery 派发的行为验证（无真实 DB）。"""
|
||
|
||
import asyncio
|
||
import unittest
|
||
from unittest.mock import AsyncMock, MagicMock, patch
|
||
|
||
from app.features.story.service import StoryService
|
||
|
||
|
||
class ChapterStoryComposeFlowTest(unittest.TestCase):
|
||
def test_append_version_marks_dirty_and_delays_recompose(self):
|
||
async def run():
|
||
db = MagicMock()
|
||
db.flush = AsyncMock()
|
||
db.commit = AsyncMock()
|
||
db.add = MagicMock()
|
||
story = MagicMock()
|
||
story.id = "story-1"
|
||
story.current_version_id = None
|
||
story.title = "T"
|
||
|
||
version = MagicMock()
|
||
version.id = "ver-new"
|
||
|
||
with (
|
||
patch(
|
||
"app.features.story.service.get_story_by_id",
|
||
new_callable=AsyncMock,
|
||
return_value=story,
|
||
),
|
||
patch(
|
||
"app.features.story.service.count_story_versions",
|
||
new_callable=AsyncMock,
|
||
return_value=0,
|
||
),
|
||
patch(
|
||
"app.features.story.service.create_story_version",
|
||
new_callable=AsyncMock,
|
||
return_value=version,
|
||
),
|
||
patch(
|
||
"app.features.story.service._extract_and_store_image_intent",
|
||
new_callable=AsyncMock,
|
||
),
|
||
patch(
|
||
"app.features.memoir.repo.mark_chapters_dirty_for_story",
|
||
new_callable=AsyncMock,
|
||
) as m_mark,
|
||
patch(
|
||
"app.tasks.chapter_compose_tasks.recompose_chapters_for_story"
|
||
) as m_task,
|
||
):
|
||
m_task.delay = MagicMock()
|
||
svc = StoryService(db=db)
|
||
await svc.append_version("story-1", "# 新正文")
|
||
m_mark.assert_awaited_once_with(db, "story-1")
|
||
m_task.delay.assert_called_once_with("story-1")
|
||
|
||
asyncio.run(run())
|