Aislo/resources/tester/test_529_3_plan_points.py

"""D02 기준 지정(PLAN 529-3 · 560-2 · 579-1) — 경위도 ↔ TM 변환 · 시점 · 종점 목록 정본 API ·
옛 한 점 꼴 감쌈 · 옛 연도 행 접기 · 저장된 좌표계 그대로.

./venv/Scripts/python.exe -m pytest resources/tester/test_529_3_plan_points.py -q
"""

import json

import pytest
from fastapi import FastAPI
from fastapi.testclient import TestClient

from common_util.common_util_auth import require_company
from common_util.common_util_crs import lonlat_to_tm, tm_epsg_for_lon, tm_to_lonlat
from D02_PlanBase import D02_PlanBase_Router as router_mod

PID = "12953f5a-5ba1-42ad-a427-8ac857a58bc2"
URL = f"/api/projects/{PID}/plan/points"


# --- 좌표 변환 ---


@pytest.mark.parametrize("epsg, lon0", [(5185, 125.0), (5186, 127.0), (5187, 129.0)])
def test_origin_is_false_origin(epsg, lon0) -> None:
    x, y = lonlat_to_tm(lon0, 38.0, epsg)  # 원점 = (200 000, 600 000)
    assert abs(x - 200000.0) < 0.001 and abs(y - 600000.0) < 0.001


@pytest.mark.parametrize("lon, lat", [(127.3851, 36.3504), (125.7, 34.1), (129.36, 36.02)])
def test_round_trip_within_1mm(lon, lat) -> None:
    epsg = tm_epsg_for_lon(lon)
    x, y = lonlat_to_tm(lon, lat, epsg)
    back = lonlat_to_tm(*tm_to_lonlat(x, y, epsg), epsg)
    assert abs(back[0] - x) < 0.001 and abs(back[1] - y) < 0.001


def test_belt_by_longitude() -> None:
    assert [tm_epsg_for_lon(v) for v in (124.5, 125.999, 126.0, 127.999, 128.0, 131.8)] == [
        5185,
        5185,
        5186,
        5186,
        5187,
        5187,
    ]


# --- 지점 정본 API ---


A, B, C = (
    {"lon": 127.10, "lat": 37.50},
    {"lon": 127.12, "lat": 37.51},
    {"lon": 127.15, "lat": 37.52},
)


@pytest.fixture
def api(monkeypatch):
    state = {
        "project": {"project_kind": "PLAN", "project_year": 2026, "project_year_end": 2027},
        "rows": {},
        "session": {"role": "USER", "email": "kim@forest.go.kr", "company_id": 1},
    }

    async def load(project_id):
        return state["project"], {y: dict(r) for y, r in state["rows"].items()}

    async def store_base(project_id, base):
        state["project"] = {**state["project"], "plan_base_json": json.dumps(base)}

    monkeypatch.setattr(router_mod, "_load", load)
    monkeypatch.setattr(router_mod, "_store_base", store_base)
    app = FastAPI()
    app.include_router(router_mod.router)
    app.dependency_overrides[require_company] = lambda: state["session"]
    client = TestClient(app)
    client.state = state
    return client


def test_put_lists(api) -> None:
    got = api.put(URL, json={"starts": [A, B], "ends": [C]})
    assert got.status_code == 200
    data = got.json()
    assert data["epsg"] == 5186 and len(data["starts"]) == 2 and len(data["ends"]) == 1
    assert (data["starts"][0]["x"], data["starts"][0]["y"]) == pytest.approx(
        lonlat_to_tm(127.10, 37.50, 5186)
    )
    end = data["ends"][0]
    assert abs(end["lon"] - C["lon"]) < 1e-8 and abs(end["lat"] - C["lat"]) < 1e-8
    stored = json.loads(api.state["project"]["plan_base_json"])
    assert set(stored) == {"epsg", "starts", "ends"} and set(stored["starts"][1]) == {"x", "y"}
    assert api.state["rows"] == {}  # 연도 행은 D02 가 안 만듦


def test_put_partial_and_empty(api) -> None:
    data = api.put(URL, json={"starts": [A]}).json()
    assert len(data["starts"]) == 1 and data["ends"] == [] and data["epsg"] == 5186
    data = api.put(URL, json={}).json()
    assert data == {"status": "success", "epsg": None, "starts": [], "ends": []}
    assert api.put(URL, json={"starts": [A] * 21}).status_code == 422  # 상한 20


def test_put_keeps_stored_epsg(api) -> None:
    api.put(URL, json={"starts": [A], "ends": [C]})
    far = {"lon": 128.9, "lat": 37.5}  # 평균이 128° 를 넘어도 저장된 5186 그대로
    data = api.put(URL, json={"starts": [A], "ends": [far, far]}).json()
    assert data["epsg"] == 5186


def test_get_wraps_old_single_points(api) -> None:
    a, c = (lonlat_to_tm(p["lon"], p["lat"], 5186) for p in (A, C))
    old = {"epsg": 5186, "start": {"x": a[0], "y": a[1]}, "end": {"x": c[0], "y": c[1]}}
    api.state["project"]["plan_base_json"] = json.dumps(old)
    data = api.get(URL).json()
    assert [p["x"] for p in data["starts"]] == pytest.approx([a[0]])
    assert [p["x"] for p in data["ends"]] == pytest.approx([c[0]])
    old["end"] = None
    api.state["project"]["plan_base_json"] = json.dumps(old)
    assert api.get(URL).json()["ends"] == []


def test_put_forest_only(api) -> None:
    api.state["session"] = {"role": "USER", "email": "lee@company.com", "company_id": 1}
    assert api.put(URL, json={"starts": [A], "ends": [B]}).status_code == 403
    api.state["session"] = {"role": "SYSTEM_ADMIN", "email": "admin@x.com", "company_id": None}
    assert api.put(URL, json={"starts": [A], "ends": [B]}).status_code == 200


def test_get_folds_old_year_rows(api) -> None:
    epsg = 5186
    a, b, c = (lonlat_to_tm(p["lon"], p["lat"], epsg) for p in (A, B, C))
    for year, s, e in ((2026, a, b), (2027, b, c)):
        api.state["rows"][year] = {
            "year": year,
            **{"start_x": s[0], "start_y": s[1], "end_x": e[0], "end_y": e[1]},
            "epsg": epsg,
        }
    data = api.get(URL).json()
    assert data["epsg"] == epsg
    assert (data["starts"][0]["x"], data["ends"][0]["x"]) == pytest.approx((a[0], c[0]))


def test_get_empty_new_project(api) -> None:
    assert api.get(URL).json() == {"status": "success", "epsg": None, "starts": [], "ends": []}