o2o-site-AEO/solution/backend/tests/test_place_search.py

"""상호명 공개 검색 — 로그인 없이 열려 있고, 업종을 LLM 없이 외부 분류로 추정한다.

★ 인증이 없는 유일한 place 경로다. 우리 DB 값이 하나도 나가지 않는지 여기서 고정한다.
"""
from common.enums import ErrorType, PlaceCategory
from common.utils import rate_limit
from services.external import kakao as kakao_client
from services.external import naver as naver_client
from services.place_category import guess_category


def _kp(name, category_name, group_code, kakao_id="1"):
    return kakao_client.KakaoPlace(
        kakao_place_id=kakao_id, name=name, road_address="강원 양양군 A로 1", address="양양읍 1-1",
        phone="033-000-0000", latitude=38.0, longitude=128.6,
        category_name=category_name, category_group_code=group_code, place_url="http://x",
    )


def _patch_kakao(monkeypatch, rows):
    monkeypatch.setattr(kakao_client.KakaoLocalClient, "enabled", property(lambda self: True))

    async def _search(self, name, x=None, y=None, size=15):
        return rows

    async def _close(self):
        return None

    monkeypatch.setattr(kakao_client.KakaoLocalClient, "search_keyword", _search)
    monkeypatch.setattr(kakao_client.KakaoLocalClient, "aclose", _close)


# ── 업종 추정 ────────────────────────────────────────────────────────────
def test_group_code_maps_to_category():
    assert guess_category("가정,생활 > 숙박 > 펜션", "AD5") is PlaceCategory.LODGING
    assert guess_category("음식점 > 카페", "CE7") is PlaceCategory.CAFE
    assert guess_category("음식점 > 한식 > 육류,고기", "FD6") is PlaceCategory.RESTAURANT


def test_hospital_needs_the_right_department():
    """HP8 은 병원이라는 것까지만 말한다 — 우리가 여는 건 피부과·성형외과뿐이다."""
    assert guess_category("의료,건강 > 병원 > 피부과", "HP8") is PlaceCategory.CLINIC
    assert guess_category("의료,건강 > 병원 > 치과", "HP8") is None


def test_skin_care_shop_is_not_a_clinic():
    """'피부' 로 보면 피부관리실이 병원으로 걸린다 — 의료 광고 규제가 붙는 업종이라 제일 비싸다."""
    assert guess_category("생활,서비스 > 미용 > 피부관리") is None


def test_naver_has_no_group_code():
    assert guess_category("숙박>게스트하우스") is PlaceCategory.LODGING
    assert guess_category("음식점>카페,디저트") is PlaceCategory.CAFE


def test_unknown_category_is_none():
    """모르면 None. 억지로 고르지 않는다 — 화면이 사장님에게 직접 묻는다."""
    assert guess_category(None) is None
    assert guess_category("스포츠,레저 > 골프장") is None


# ── 엔드포인트 ───────────────────────────────────────────────────────────
async def test_search_needs_no_token(client, monkeypatch):
    rate_limit.reset()
    _patch_kakao(monkeypatch, [_kp("하조대펜션", "가정,생활 > 숙박 > 펜션", "AD5")])

    res = await client.get("/v1/place/search", params={"q": "하조대펜션"})

    assert res.status_code == 200
    body = res.json()
    assert body["result"]["success"] is True
    assert body["items"][0]["name"] == "하조대펜션"
    assert body["items"][0]["category"] == PlaceCategory.LODGING.value


async def test_search_leaks_nothing_of_ours(client, monkeypatch):
    """공개 응답이라 우리 DB 값이 새면 그대로 밖이다. 좌표·전화도 뺐다."""
    rate_limit.reset()
    _patch_kakao(monkeypatch, [_kp("하조대펜션", "가정,생활 > 숙박 > 펜션", "AD5")])

    item = (await client.get("/v1/place/search", params={"q": "하조대펜션"})).json()["items"][0]

    for leaked in ("place_id", "owner_user_id", "phone", "latitude", "longitude",
                   "external_place_id", "address"):
        assert leaked not in item, f"{leaked} 가 공개 응답에 나갔다"


async def test_search_does_not_create_a_place(client, monkeypatch, auth_headers):
    """검색만으로 사업장이 생기면 안 된다 — 랜딩 트래픽이 그대로 DB 가 된다."""
    rate_limit.reset()
    _patch_kakao(monkeypatch, [_kp("하조대펜션", "가정,생활 > 숙박 > 펜션", "AD5")])
    h = await auth_headers("u1")

    before = (await client.get("/v1/place/list", headers=h)).json()["total"]
    await client.get("/v1/place/search", params={"q": "하조대펜션"})
    after = (await client.get("/v1/place/list", headers=h)).json()["total"]

    assert before == after


async def test_search_rejects_short_query(client):
    rate_limit.reset()
    res = await client.get("/v1/place/search", params={"q": "가"})
    assert res.status_code == 422  # min_length=2


async def test_rate_limit_closes_the_tap(client, monkeypatch):
    """인증이 없는데 유료 외부 API 를 부른다 — 새로고침만으로 요금이 나가면 안 된다.

    ★ 검색어를 매번 바꾼다. 같은 검색어는 ttl_cache 가 레이트리밋보다 **앞에서** 받아
      외부 API 를 아예 안 부르므로(요금이 안 나가므로 제한할 이유도 없다), 같은 말을
      반복하면 이 가드가 아니라 캐시를 시험하게 된다."""
    rate_limit.reset()
    _patch_kakao(monkeypatch, [_kp("하조대펜션", "가정,생활 > 숙박 > 펜션", "AD5")])

    from services import place_service

    monkeypatch.setattr(place_service, "_PUBLIC_SEARCH_PER_MIN", 2)
    for i in range(2):
        r = (await client.get("/v1/place/search", params={"q": f"하조대펜션{i}"})).json()
        assert r["result"]["success"] is True

    blocked = (await client.get("/v1/place/search", params={"q": "하조대펜션9"})).json()
    assert blocked["result"]["success"] is False
    assert blocked["result"]["code"] == ErrorType.HTTP_TO_MANY_REQUEST.value


async def test_cached_query_does_not_spend_the_rate_limit(client, monkeypatch):
    """검증: 같은 검색어를 제한 횟수보다 많이 친다.
    기대결과: 통과한다 — 캐시가 받으면 외부 API 를 안 부르고, 안 부르면 요금도 안 난다.

    ★ 레이트리밋을 캐시보다 앞으로 옮기면 사장님이 상호를 고쳐 가며 치는 평범한 사용이
      막힌다. 이 순서가 의도라는 것을 여기서 못 박는다."""
    rate_limit.reset()
    _patch_kakao(monkeypatch, [_kp("하조대펜션", "가정,생활 > 숙박 > 펜션", "AD5")])

    from services import place_service

    monkeypatch.setattr(place_service, "_PUBLIC_SEARCH_PER_MIN", 2)
    for _ in range(5):
        r = (await client.get("/v1/place/search", params={"q": "같은검색어"})).json()
        assert r["result"]["success"] is True