42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
||
"""extract-chapter-knowledge：章后抽取 lesson 离线钉死。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import pathlib
|
||
import sys
|
||
import unittest
|
||
from unittest.mock import patch
|
||
|
||
ROOT = pathlib.Path(__file__).resolve().parents[3]
|
||
SCRIPT = ROOT / "muse" / "content" / "entity" / "skills" / "extract" / "extract-chapter-knowledge" / "scripts"
|
||
EVIDENCE = ROOT / "muse" / "authority" / "evidence" / "skills" / "record-run-evidence" / "scripts"
|
||
for path in (SCRIPT, EVIDENCE):
|
||
if str(path) not in sys.path:
|
||
sys.path.insert(0, str(path))
|
||
|
||
from extract_knowledge import _propose_chapter_extract_lesson # noqa: E402
|
||
|
||
|
||
class ChapterExtractLessonOfflineTest(unittest.TestCase):
|
||
def test_registers_win(self):
|
||
calls: list[dict] = []
|
||
|
||
def fake(**kwargs):
|
||
calls.append(kwargs)
|
||
return {"status": "proposed", "lesson_id": 3, "deduped": False}
|
||
|
||
with patch("extract_knowledge.propose_lesson_dedup", side_effect=fake):
|
||
out = _propose_chapter_extract_lesson(
|
||
run_id="run-ch-9", work_id=1, chapter_order=9,
|
||
draft_count=4, state_draft_id=99,
|
||
)
|
||
self.assertEqual(out["lesson_id"], 3)
|
||
self.assertEqual(calls[0]["kind"], "win")
|
||
self.assertEqual(calls[0]["title"], "章后知识抽取落库成功")
|
||
self.assertEqual(calls[0]["detail"]["chapter_order"], 9)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|