#!/usr/bin/env python3
"""文风取数端回归：load_confirmed_style / derive_style_constraints（桩连接，不连真库）。

read-context 的文风取数端，与细纲/范式取数端并排成一等公民。优先独立 style 规划行，
回退设定行的 style 字段（书12 现状为一句话 blob）。结构化画像抽取仍待建。
"""

import pathlib
import sys
import unittest

sys.path.insert(0, str(pathlib.Path(__file__).resolve().parent))

from retrieve_writer_sources import derive_style_constraints, load_confirmed_style  # noqa: E402


class FakeCursor:
    def __init__(self, row):
        self._row = row

    def fetchone(self):
        return self._row


class FakeConn:
    """按 SQL 区分两次查询：schema_type='style' 的独立文风行 / section_type='setting' 的设定行。"""

    def __init__(self, style_row=None, setting_row=None):
        self.style_row = style_row
        self.setting_row = setting_row

    def execute(self, sql, params=None):
        if "schema_type='style'" in sql:
            return FakeCursor(self.style_row)
        if "section_type='setting'" in sql:
            return FakeCursor(self.setting_row)
        return FakeCursor(None)


class DeriveStyleTest(unittest.TestCase):
    def test_blob_becomes_single_constraint(self):
        self.assertEqual(derive_style_constraints("冷峻、克制、微恐。"), ["冷峻、克制、微恐。"])

    def test_structured_dict_projects_per_aspect(self):
        self.assertEqual(
            derive_style_constraints({"叙事人称": "第三人称限制视角", "AI味黑名单": "禁 仿佛/似乎", "达标样张": ""}),
            ["叙事人称：第三人称限制视角", "AI味黑名单：禁 仿佛/似乎"],
        )

    def test_blank_and_illegal_return_empty(self):
        self.assertEqual(derive_style_constraints("   "), [])
        self.assertEqual(derive_style_constraints(None), [])
        self.assertEqual(derive_style_constraints(42), [])


class LoadConfirmedStyleTest(unittest.TestCase):
    def test_prefers_dedicated_style_section(self):
        conn = FakeConn(style_row=('{"叙事人称": "第三人称"}',),
                        setting_row=('{"style": "不该用到我"}',))
        self.assertEqual(load_confirmed_style(conn, work_id=12), ["叙事人称：第三人称"])

    def test_falls_back_to_setting_style_field(self):
        conn = FakeConn(style_row=None, setting_row=('{"title": "深渊机神", "style": "冷峻克制"}',))
        self.assertEqual(load_confirmed_style(conn, work_id=12), ["冷峻克制"])

    def test_no_confirmed_style_returns_empty(self):
        conn = FakeConn(style_row=None, setting_row=None)
        self.assertEqual(load_confirmed_style(conn, work_id=12), [])


if __name__ == "__main__":
    unittest.main()