"""ip_service 区码扩展(同步自上游 ai-kefu tests/test_ip_service_region.py)。

仅改 import 路径(app.services → app.engine.services);断言原样——
iptocc 数据面行为两库同源,实测 41.90.64.1→KE / 24.114.0.1→加拿大。
"""
from app.engine.services.ip_service import COUNTRY_NAMES, resolve_country, resolve_region_code


def test_resolve_region_code_direct_passthrough():
    assert resolve_region_code("8.8.8.8") in ("US", None)  # iptocc 数据面,断言不抛
    assert resolve_region_code("127.0.0.1") is None        # 本地
    assert resolve_region_code("") is None


def test_resolve_region_code_matches_resolve_country_space():
    # 同一 IP:resolve_region_code 返回 ISO,resolve_country 走 COUNTRY_NAMES 映射
    code = resolve_region_code("41.90.64.1")  # 肯尼亚段
    assert code == "KE"


def test_ca_country_name_fixed():
    assert COUNTRY_NAMES["CA"] == "加拿大"
    assert COUNTRY_NAMES["AU"] == "澳大利亚"
    assert resolve_country("24.114.0.1") != "CA"  # 加拿大段不再透出字面量