#!/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()