"""
城市 full_name/pinyin 自动补全测试
=============================================================

验证新增/编辑城市时自动补全派生字段：
    - full_name：沿 pid 链从首层城市拼接（如 台湾省,云林,元长,华鹿）
    - pinyin：由城市简称生成无音调拼音，音节首字母大写（如 台湾 -> TaiWan、华鹿 -> HuaLu）

覆盖新增、编辑改名、编辑变更父级三种场景。
"""

from src.core.database import db
from src.modules.city.repository import CityRepository
from src.modules.city.service import CityService


def _payload(name, short_name, pid, level):
    """构造合法的城市表单数据（不含派生字段）"""
    return {
        "cityCode": "012",
        "areaCode": "640300",
        "parentCode": "0",
        "zipCode": "750000",
        "level": level,
        "name": name,
        "shortName": short_name,
        "pid": pid,
        "lng": "106.2",
        "lat": "23.7",
    }


def _add_city(app, name, short_name, pid, level):
    """经 CityService.add 新增城市并返回其ID"""
    with app.app_context():
        CityService.add(_payload(name, short_name, pid, level))
        db.session.commit()
        return CityRepository.first_by({"name": name}).id


class TestCityFullName:
    """full_name 沿 pid 链从首层拼接"""

    def test_full_name_chain(self, app):
        province_id = _add_city(app, "台湾省", "台湾", 0, 1)
        city_id = _add_city(app, "云林", "云林", province_id, 2)
        district_id = _add_city(app, "元长", "元长", city_id, 3)
        street_id = _add_city(app, "华鹿", "华鹿", district_id, 4)

        with app.app_context():
            province = CityRepository.get_by_id(province_id)
            city = CityRepository.get_by_id(city_id)
            district = CityRepository.get_by_id(district_id)
            street = CityRepository.get_by_id(street_id)

        assert province.full_name == "台湾省"
        assert city.full_name == "台湾省,云林"
        assert district.full_name == "台湾省,云林,元长"
        assert street.full_name == "台湾省,云林,元长,华鹿"

    def test_update_rename_recomputes(self, app):
        province_id = _add_city(app, "广东省", "广东", 0, 1)
        city_id = _add_city(app, "广州市", "广州", province_id, 2)

        with app.app_context():
            data = _payload("东莞市", "东莞", province_id, 2)
            data["id"] = city_id
            CityService.update(data)
            db.session.commit()
            db.session.expire_all()
            city = CityRepository.get_by_id(city_id)

        assert city.name == "东莞市"
        assert city.full_name == "广东省,东莞市"

    def test_update_reparent_recomputes(self, app):
        prov_a = _add_city(app, "A省", "A省", 0, 1)
        prov_b = _add_city(app, "B省", "B省", 0, 1)
        city_id = _add_city(app, "旧城", "旧城", prov_a, 2)

        with app.app_context():
            data = _payload("新城", "新城", prov_b, 2)
            data["id"] = city_id
            CityService.update(data)
            db.session.commit()
            db.session.expire_all()
            city = CityRepository.get_by_id(city_id)

        assert city.full_name == "B省,新城"


class TestCityPinyin:
    """pinyin 由简称生成（无音调）"""

    def test_pinyin_uses_short_name(self, app):
        province_id = _add_city(app, "台湾省", "台湾", 0, 1)
        city_id = _add_city(app, "云林", "云林", province_id, 2)

        with app.app_context():
            province = CityRepository.get_by_id(province_id)
            city = CityRepository.get_by_id(city_id)

        assert province.pinyin == "TaiWan"
        assert city.pinyin == "YunLin"

    def test_update_rename_recomputes_pinyin(self, app):
        province_id = _add_city(app, "广东省", "广东", 0, 1)
        city_id = _add_city(app, "广州市", "广州", province_id, 2)

        with app.app_context():
            data = _payload("深圳市", "深圳", province_id, 2)
            data["id"] = city_id
            CityService.update(data)
            db.session.commit()
            db.session.expire_all()
            city = CityRepository.get_by_id(city_id)

        assert city.pinyin == "ShenZhen"