61 lines
2.7 KiB
Python
61 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
"""知识确认合同的离线测试，不连接 PostgreSQL。"""
|
|
import pathlib
|
|
import sys
|
|
import unittest
|
|
|
|
PROJECT_ROOT = pathlib.Path(__file__).resolve().parents[3]
|
|
SCRIPT_DIR = PROJECT_ROOT / "muse" / "content" / "entity" / "skills" / "sovereignty" / "确认知识草稿" / "scripts"
|
|
sys.path.insert(0, str(SCRIPT_DIR))
|
|
|
|
import confirm_knowledge as confirm # noqa: E402
|
|
|
|
|
|
class ConfirmKnowledgeContractTest(unittest.TestCase):
|
|
def test英文实体卡投影到正式实体字段(self):
|
|
card = confirm.normalize_entity_payload({
|
|
"type": "character", "name": "林深", "brief": "机师",
|
|
"fields": {"身份": "驾驶员"},
|
|
"source": {"workId": 12, "chapter": 1, "chapterId": 17473},
|
|
"evidence": "林深坐在二号观测位。",
|
|
})
|
|
self.assertEqual(card["entity_type"], "character")
|
|
self.assertEqual(card["normalized_name"], "林深")
|
|
self.assertEqual(card["attributes"], {"身份": "驾驶员"})
|
|
self.assertEqual(card["source_ref"]["chapterId"], 17473)
|
|
|
|
def test关系端点从对象名称解析(self):
|
|
relation = confirm.normalize_relation_payload({
|
|
"type": "驾驶", "source": {"name": "林深"},
|
|
"target": {"name": "茧"}, "description": "首次实战驾驶",
|
|
})
|
|
self.assertEqual(relation["source_name"], "林深")
|
|
self.assertEqual(relation["target_name"], "茧")
|
|
self.assertEqual(relation["relation_type"], "驾驶")
|
|
|
|
def test缺失关系端点失败关闭(self):
|
|
with self.assertRaisesRegex(confirm.KnowledgeConfirmationError, "relation.target"):
|
|
confirm.normalize_relation_payload({
|
|
"type": "驾驶", "source": "林深", "description": "x",
|
|
})
|
|
|
|
def test来源回放链只追加去重结构化引用(self):
|
|
old = {"sourceRefs": [{"chapter": 1}], "lastDraftId": 10}
|
|
same = confirm._lineage(old, {"chapter": 1}, 11, "chapter_extract")
|
|
self.assertEqual(same["sourceRefs"], [{"chapter": 1}])
|
|
newer = confirm._lineage(same, {"chapter": 2}, 12, "chapter_extract")
|
|
self.assertEqual(newer["sourceRefs"], [{"chapter": 1}, {"chapter": 2}])
|
|
self.assertEqual(newer["lastDraftId"], 12)
|
|
|
|
def test已确认关系点名重试返回幂等结果(self):
|
|
row = (7, 12, "relation", {}, "confirmed", "chapter_extract", 17473, 0.9,
|
|
"active", "allowed", 2, None, 91)
|
|
result = confirm._confirm_relation_locked(None, row, kb_id=3, decided_by="1")
|
|
self.assertEqual(result, {
|
|
"draft_id": 7, "status": "already_confirmed", "relation_id": 91,
|
|
})
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|