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

"""템플릿(디자인) 선택 저장.

이 경로가 절대 하면 안 되는 것:
  - 고른 템플릿을 브라우저에만 두는 것 — 발행 잡이 읽을 곳이 없어 업종 기본으로 굽는다.
    사장님이 고른 디자인과 실제 발행본이 갈리는 것이 이 API 가 생긴 이유다.
  - ★ 발행됐다고 템플릿을 잠그는 것 — 주소(slug)와 달리 디자인은 바뀌어도 URL 이 그대로다.
    잠글 이유가 없는 것을 잠그면 사장님이 발행 후에 디자인을 못 바꾼다.
  - 바꿔놓고 재빌드 표시를 안 하는 것 — 관리 화면은 새 디자인, 나가 있는 페이지는 옛 디자인이 된다.
"""
import uuid

from sqlalchemy import text

from common.enums import ErrorType, PlaceCategory, SiteStatus
from services.site_payload import _DEFAULT_THEME, to_site_payload


async def _place(client, headers, name="템플릿펜션"):
    r = await client.post("/v1/place", headers=headers, json={"name": name, "category": 1})
    return r.json()["place"]["place_id"]


async def _set_template(client, headers, pid, template_id):
    return (await client.post(f"/v1/place/{pid}/site/template", headers=headers,
                              json={"template_id": template_id})).json()


async def _get_site(client, headers, pid):
    return (await client.get(f"/v1/place/{pid}/site", headers=headers)).json()


async def test_set_template_creates_site_row(auth_headers, client):
    """검증: 사이트 행이 없어도 템플릿을 먼저 고를 수 있다(고르는 건 발행보다 앞선 결정이다).
    기대결과: SUCCESS + 저장된 값이 조회에도 그대로 나온다."""
    h = await auth_headers("tpl1")
    pid = await _place(client, h)

    saved = await _set_template(client, h, pid, "stay-quiet-margin")
    assert saved["result"]["code"] == ErrorType.SUCCESS.value
    assert saved["site"]["template_id"] == "stay-quiet-margin"

    # 화면이 "지금 어느 템플릿으로 나가는지"를 조회로 다시 읽을 수 있어야 한다.
    assert (await _get_site(client, h, pid))["site"]["template_id"] == "stay-quiet-margin"


async def test_unknown_template_key_is_accepted(auth_headers, client):
    """검증: ★ 서버는 값을 검증하지 않는다 — 템플릿 목록은 프론트가 소유한다.
    기대결과: 모르는 키도 저장된다(화이트리스트를 두면 템플릿 하나 늘릴 때마다 백엔드를 고쳐야 한다)."""
    h = await auth_headers("tpl2")
    pid = await _place(client, h)

    saved = await _set_template(client, h, pid, "brand-new-template-nobody-knows")
    assert saved["result"]["code"] == ErrorType.SUCCESS.value
    assert saved["site"]["template_id"] == "brand-new-template-nobody-knows"


async def test_too_long_template_is_refused(auth_headers, client):
    """검증: 길이(varchar(100))만은 막는다 — 안 막으면 DB 가 트랜잭션째로 튕긴다.
    기대결과: INVALID_REQUEST_DATA."""
    h = await auth_headers("tpl3")
    pid = await _place(client, h)

    refused = await _set_template(client, h, pid, "t" * 101)
    assert refused["result"]["code"] == ErrorType.INVALID_REQUEST_DATA.value
    assert "site" not in refused


async def test_empty_value_clears_to_default(auth_headers, client):
    """검증: 빈 값은 '고르지 않음'이다 — NULL 로 되돌아가 업종 기본으로 떨어진다.
    기대결과: 저장 후 빈 문자열을 보내면 template_id 가 응답에서 사라진다(None)."""
    h = await auth_headers("tpl4")
    pid = await _place(client, h)
    await _set_template(client, h, pid, "stay-quiet-margin")

    cleared = await _set_template(client, h, pid, "")
    assert cleared["result"]["code"] == ErrorType.SUCCESS.value
    # RemoveNoneResponse 가 None 필드를 지운다 — 키가 없으면 NULL 이다.
    assert "template_id" not in cleared["site"]


async def test_published_site_template_is_not_locked(auth_headers, client, db_engine):
    """검증: ★ 발행된 사이트도 템플릿은 바꿀 수 있다(주소와 다르다 — URL 이 그대로라 색인이 안 깨진다).
    기대결과: SUCCESS + 재빌드 필요 표시(needs_rebuild)."""
    h = await auth_headers("tpl5")
    pid = await _place(client, h)
    await _set_template(client, h, pid, "stay-o2o-editorial")

    # 발행 상태를 만든다(빌드 잡을 돌리는 대신 상태만).
    async with db_engine.begin() as conn:
        await conn.execute(
            text("UPDATE sites SET status = :st, published_at = now() WHERE place_id = :p"),
            {"st": SiteStatus.PUBLISHED.value, "p": uuid.UUID(pid)},
        )

    changed = await _set_template(client, h, pid, "stay-quiet-margin")
    assert changed["result"]["code"] == ErrorType.SUCCESS.value
    assert changed["site"]["template_id"] == "stay-quiet-margin"
    # 나가 있는 페이지와 달라졌으므로 이 사업장만 다시 빌드하면 된다는 표시가 서야 한다.
    assert changed["needs_rebuild"] is True


async def test_other_company_place_is_blocked(auth_headers, client, other_company_id):
    """검증: 남의 회사 사업장의 템플릿은 바꿀 수 없다.
    기대결과: PLACE_NOT_FOUND(존재 여부조차 알려주지 않는다)."""
    h = await auth_headers("tpl6")
    intruder = await auth_headers("tpl7", other_company_id)
    pid = await _place(client, h)

    blocked = await _set_template(client, intruder, pid, "stay-quiet-margin")
    assert blocked["result"]["code"] == ErrorType.PLACE_NOT_FOUND.value


def test_payload_uses_saved_template_and_falls_back():
    """검증: 발행 payload 의 theme.templateId 는 저장된 값을 쓰고, 없으면 업종 기본으로 떨어진다.
    기대결과: 저장값 → 그대로 / NULL → 업종 기본. ★ 고르지 않은 값을 고른 것처럼 굽지 않는다."""
    place = {"place_id": uuid.uuid4(), "category": PlaceCategory.LODGING.value, "name": "스테이,머뭄"}
    snapshot = {"place": {"name": "스테이,머뭄", "category": PlaceCategory.LODGING.value}}
    version = {"version": 1}

    chosen = to_site_payload(place, snapshot, {"template_id": "stay-quiet-margin"}, version, [])
    assert chosen["theme"]["templateId"] == "stay-quiet-margin"

    default_id = _DEFAULT_THEME[PlaceCategory.LODGING.value]["templateId"]
    assert to_site_payload(place, snapshot, {"template_id": None}, version, [])["theme"]["templateId"] == default_id
    # 색·서체는 여전히 업종 기본이다(그 값들은 아직 저장되는 자리가 없다).
    assert chosen["theme"]["fontStyle"] == _DEFAULT_THEME[PlaceCategory.LODGING.value]["fontStyle"]