"""city service 单元测试（回归 haveChild 尾部逗号导致恒为元组/恒 truthy 的 bug）。"""
import json

from modules.city import service as city_service
from modules.city.models import City
from modules.city.repository import city_repo


def _resp_body(resp):
    return json.loads(resp.body)


def test_get_city_list_have_child_is_bool(fake_request, sqlite_db):
    city_repo.create(
        city_code="100000", area_code="100000", parent_code="0", zip_code="000000",
        level=1, parent_id=0, name="北京市", short_name="北京", full_name="北京市",
        pinyin="beijing", lng="116.4", lat="39.9",
    )
    fake_request.query_params = {"parent_id": 0}

    resp = city_service.get_city_list(fake_request)
    data = _resp_body(resp)["data"]
    assert isinstance(data, list) and len(data) == 1
    assert data[0]["haveChild"] is True


def test_get_city_list_leaf_have_child_false(fake_request, sqlite_db):
    city_repo.create(
        city_code="100000", area_code="100000", parent_code="0", zip_code="000000",
        level=4, parent_id=0, name="某街道", short_name="街道", full_name="某街道",
        pinyin="jiedao", lng="116.4", lat="39.9",
    )
    fake_request.query_params = {"parent_id": 0}

    resp = city_service.get_city_list(fake_request)
    data = _resp_body(resp)["data"]
    assert isinstance(data, list) and len(data) == 1
    assert data[0]["haveChild"] is False