"""宿主导入模板表头 / lookup 展示值契约（replaces:import）。"""

from __future__ import annotations

import csv
import tempfile
import unittest
from pathlib import Path


class TableFileImportTests(unittest.TestCase):
    def test_normalize_header_host_leading_and_legacy_trailing_star(self) -> None:
        from service.table_file_import import map_headers, normalize_header

        self.assertEqual(normalize_header("*长尾词"), "长尾词")
        self.assertEqual(normalize_header("* 长尾词"), "长尾词")
        self.assertEqual(normalize_header("长尾词*"), "长尾词")
        self.assertEqual(normalize_header("＊关键词＊"), "关键词")
        self.assertEqual(normalize_header("*关键词（必填）"), "关键词")
        self.assertEqual(normalize_header("名称"), "名称")

        aliases = {
            "phrase": ("长尾词", "phrase", "问题"),
            "keyword_id": ("所属关键词", "keyword_id", "关键词编号"),
        }
        mapping = map_headers(["*长尾词", "所属关键词"], aliases)
        self.assertEqual(mapping.get("phrase"), 0)
        self.assertEqual(mapping.get("keyword_id"), 1)

        mapping_legacy = map_headers(["长尾词*", "所属关键词"], aliases)
        self.assertEqual(mapping_legacy.get("phrase"), 0)

    def test_optional_int_accepts_lookup_display_label(self) -> None:
        from service.table_file_import import optional_int

        self.assertEqual(optional_int("157"), 157)
        self.assertEqual(optional_int("157 · 波兰小包专线"), 157)
        self.assertEqual(optional_int("157·波兰小包专线"), 157)
        self.assertEqual(optional_int("2 · 通用私信-留资"), 2)
        self.assertIsNone(optional_int(""))
        with self.assertRaises(ValueError):
            optional_int("波兰小包专线")

    def test_load_table_rows_skips_example_and_maps_leading_star(self) -> None:
        from service.table_file_import import load_table_rows

        aliases = {
            "phrase": ("长尾词",),
            "keyword_id": ("所属关键词",),
        }
        with tempfile.TemporaryDirectory() as tmp:
            path = Path(tmp) / "rows.csv"
            with path.open("w", encoding="utf-8-sig", newline="") as fh:
                writer = csv.writer(fh)
                writer.writerow(["*长尾词", "所属关键词"])
                writer.writerow(["【示例】示例长尾词", "152 · 示例"])
                writer.writerow(["波兰小包专线怎么选", "157 · 波兰小包专线"])

            rows = load_table_rows(str(path), aliases=aliases)
            self.assertEqual(len(rows), 1)
            self.assertEqual(rows[0]["phrase"], "波兰小包专线怎么选")
            self.assertEqual(rows[0]["keyword_id"], "157 · 波兰小包专线")
            self.assertEqual(rows[0]["_row"], "3")


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