Model/domain/magicplan/tests/test_mapper.py

import json
from pathlib import Path
from typing import Any

import pytest

from datatypes.magicplan.api.response import MagicPlanPlan
from domain.magicplan.mapper import map_plan
from domain.magicplan.models import Plan

FIXTURE_DIR = Path(__file__).parents[3] / "backend" / "magic_plan"
PLAN_ID = "a7285ed1-878d-47eb-8aa6-85ef9e187516"
PLAN_ID_2 = "9f9889ff-793e-4e9a-a6f0-e22f5b0f5365"


@pytest.fixture(scope="module")
def raw_data() -> dict[str, Any]:
    payload = json.loads(
        (FIXTURE_DIR / "magicplan_api_plan_response_example.json").read_text()
    )
    return payload["data"]


@pytest.fixture(scope="module")
def mp(raw_data: dict[str, Any]) -> MagicPlanPlan:
    return MagicPlanPlan.model_validate(raw_data)


@pytest.fixture(scope="module")
def plan(mp: MagicPlanPlan) -> Plan:
    return map_plan(mp)


def test_plan_uid(plan: Plan):
    assert plan.uid == PLAN_ID


def test_floor_count(plan: Plan):
    assert len(plan.floors) == 2


def test_first_room_name(plan: Plan):
    assert plan.floors[0].rooms[0].name == "Kitchen"


def test_room_dimensions_are_floats(plan: Plan):
    room = plan.floors[0].rooms[0]
    assert isinstance(room.width_m, float)
    assert isinstance(room.length_m, float)
    assert isinstance(room.area_m2, float)


def test_room_area_rounded_to_2dp(plan: Plan):
    room = plan.floors[0].rooms[0]
    assert room.area_m2 == 7.95


def test_room_dimensions_parsed_from_string(plan: Plan):
    room = plan.floors[0].rooms[0]
    assert room.width_m == pytest.approx(2.67)
    assert room.length_m == pytest.approx(2.98)


def test_kitchen_has_windows(plan: Plan):
    room = plan.floors[0].rooms[0]
    assert len(room.windows) >= 1


def test_window_fields_are_floats(plan: Plan):
    window = plan.floors[0].rooms[0].windows[0]
    assert isinstance(window.width_m, float)
    assert isinstance(window.height_m, float)
    assert isinstance(window.area_m2, float)


def test_window_opening_type_prefix_stripped(plan: Plan):
    window = plan.floors[0].rooms[0].windows[0]
    assert not window.opening_type.startswith("window")
    assert window.opening_type == "casement"


def test_window_area_is_width_times_height(plan: Plan):
    window = plan.floors[0].rooms[0].windows[0]
    assert window.area_m2 == pytest.approx(window.width_m * window.height_m, rel=1e-2)


def test_window_dimensions_rounded_to_2dp(plan: Plan):
    window = plan.floors[0].rooms[0].windows[0]
    assert window.width_m == 1.40
    assert window.height_m == 1.20
    assert window.area_m2 == 1.68


def test_door_width_rounded_to_2dp(plan: Plan):
    door = plan.floors[0].rooms[0].doors[0]
    assert door.width_mm == 0.79


def test_kitchen_has_doors(plan: Plan):
    room = plan.floors[0].rooms[0]
    assert len(room.doors) >= 1


def test_door_width_is_float(plan: Plan):
    door = plan.floors[0].rooms[0].doors[0]
    assert isinstance(door.width_mm, float)


# --- Fixture 2: magicplan_api_plan_response_example_2.json ---


@pytest.fixture(scope="module")
def raw_data_2() -> dict[str, Any]:
    payload = json.loads(
        (FIXTURE_DIR / "magicplan_api_plan_response_example_2.json").read_text()
    )
    return payload["data"]


@pytest.fixture(scope="module")
def plan2(raw_data_2: dict[str, Any]) -> Plan:
    return map_plan(MagicPlanPlan.model_validate(raw_data_2))


def test_plan2_uid(plan2: Plan):
    assert plan2.uid == PLAN_ID_2


def test_plan2_floor_count(plan2: Plan):
    assert len(plan2.floors) == 3


def test_plan2_first_room_name(plan2: Plan):
    assert plan2.floors[0].rooms[0].name == "Toilet"


def test_plan2_room_area_rounded_to_2dp(plan2: Plan):
    room = plan2.floors[0].rooms[0]
    assert room.area_m2 == 0.96


def test_plan2_room_dimensions_parsed_from_string(plan2: Plan):
    room = plan2.floors[0].rooms[0]
    assert room.width_m == pytest.approx(1.12)
    assert room.length_m == pytest.approx(0.86)


def test_plan2_room_with_no_windows(plan2: Plan):
    hall = plan2.floors[0].rooms[1]
    assert hall.name == "Hall"
    assert hall.windows == []


def test_plan2_window_dimensions_rounded_to_2dp(plan2: Plan):
    window = plan2.floors[0].rooms[0].windows[0]
    assert window.width_m == 0.39
    assert window.height_m == 0.67
    assert window.area_m2 == 0.26


def test_plan2_window_opening_type_casement(plan2: Plan):
    window = plan2.floors[0].rooms[0].windows[0]
    assert window.opening_type == "casement"


def test_plan2_window_opening_type_hung(plan2: Plan):
    bathroom1 = plan2.floors[1].rooms[1]
    assert bathroom1.name == "Bathroom 1"
    assert bathroom1.windows[0].opening_type == "hung"


def test_plan2_door_width_rounded_to_2dp(plan2: Plan):
    door = plan2.floors[0].rooms[0].doors[0]
    assert door.width_mm == 0.71


# --- Address and postcode fields ---


def test_plan_postcode(plan: Plan):
    assert plan.postcode == "BR2 8BZ"


def test_plan_address(plan: Plan):
    assert plan.address == "2 Laburnum Way, Bromley, GB"


def test_plan2_postcode(plan2: Plan):
    assert plan2.postcode == "BR1 3LP"


def test_plan2_address(plan2: Plan):
    assert plan2.address == "11 Station Road, Bromley, GB"


# --- Fixture 3: street_number set, city absent ---


@pytest.fixture(scope="module")
def plan3() -> Plan:
    payload = json.loads(
        (FIXTURE_DIR / "magicplan_api_plan_response_example_3.json").read_text()
    )
    return map_plan(MagicPlanPlan.model_validate(payload["data"]))


def test_plan3_address_uses_street_number_and_omits_city(plan3: Plan):
    assert plan3.address == "2 Laburnum Way, GB"


def test_plan3_postcode(plan3: Plan):
    assert plan3.postcode == "BR2 8BZ"


# --- Fixture 4: street_number set, street absent ---


@pytest.fixture(scope="module")
def plan4() -> Plan:
    payload = json.loads(
        (FIXTURE_DIR / "magicplan_api_plan_response_example_4.json").read_text()
    )
    return map_plan(MagicPlanPlan.model_validate(payload["data"]))


def test_plan4_address_uses_street_number_when_street_absent(plan4: Plan):
    assert plan4.address == "2, Bromley, GB"