From 8e12ac705b798899239af2983ab23381db0fb970 Mon Sep 17 00:00:00 2001
From: Daniel Roth <daniel_roth@hotmail.co.uk>
Date: Thu, 7 May 2026 12:41:15 +0000
Subject: [PATCH] =?UTF-8?q?match=20address=20string=20to=20magicplan=20pla?=
 =?UTF-8?q?n=20=F0=9F=9F=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../magic_plan/tests/test_address_matcher.py  | 125 ++++++++++++++++++
 1 file changed, 125 insertions(+)
 create mode 100644 backend/magic_plan/tests/test_address_matcher.py

diff --git a/backend/magic_plan/tests/test_address_matcher.py b/backend/magic_plan/tests/test_address_matcher.py
new file mode 100644
index 00000000..2ee1f687
--- /dev/null
+++ b/backend/magic_plan/tests/test_address_matcher.py
@@ -0,0 +1,125 @@
+import pytest
+
+from datatypes.magicplan.api.response import Address, PlanSummary
+from backend.magic_plan.address_matcher import find_matching_plan, _extract_postcode
+
+
+def _make_plan(
+    plan_id: str,
+    street: str | None = None,
+    street_number: str | None = None,
+    postal_code: str | None = None,
+) -> PlanSummary:
+    return PlanSummary.model_validate(
+        {
+            "id": plan_id,
+            "name": f"Plan {plan_id}",
+            "address": {
+                "street": street,
+                "street_number": street_number,
+                "postal_code": postal_code,
+            },
+        }
+    )
+
+
+# --- _extract_postcode ---
+
+
+def test_extract_postcode_standard_format() -> None:
+    assert _extract_postcode("2 Laburnum Way Bromley BR2 8BZ") == "BR28BZ"
+
+
+def test_extract_postcode_no_space_in_postcode() -> None:
+    assert _extract_postcode("123 High St London SW1A1AA") == "SW1A1AA"
+
+
+def test_extract_postcode_lowercase_input() -> None:
+    assert _extract_postcode("2 laburnum way br2 8bz") == "BR28BZ"
+
+
+def test_extract_postcode_none_when_absent() -> None:
+    assert _extract_postcode("123 High Street London") is None
+
+
+def test_extract_postcode_none_for_empty_string() -> None:
+    assert _extract_postcode("") is None
+
+
+# --- find_matching_plan ---
+
+
+PLAN_A = _make_plan("plan-a", street="Laburnum Way", street_number="2", postal_code="BR2 8BZ")
+PLAN_B = _make_plan("plan-b", street="Station Road", street_number="11", postal_code="BR1 3LP")
+
+
+def test_find_matching_plan_returns_match() -> None:
+    # Arrange
+    plans = [PLAN_A, PLAN_B]
+    # Act
+    result = find_matching_plan(plans, "2 Laburnum Way Bromley BR2 8BZ")
+    # Assert
+    assert result is not None
+    assert result.id == "plan-a"
+
+
+def test_find_matching_plan_postcode_mismatch_returns_none() -> None:
+    # Arrange
+    plans = [PLAN_A]
+    # Act
+    result = find_matching_plan(plans, "2 Laburnum Way Bromley SW1A 1AA")
+    # Assert
+    assert result is None
+
+
+def test_find_matching_plan_street_mismatch_returns_none() -> None:
+    # Arrange
+    plans = [PLAN_A]
+    # Act
+    result = find_matching_plan(plans, "99 Other Road Bromley BR2 8BZ")
+    # Assert
+    assert result is None
+
+
+def test_find_matching_plan_empty_list_returns_none() -> None:
+    # Act
+    result = find_matching_plan([], "2 Laburnum Way Bromley BR2 8BZ")
+    # Assert
+    assert result is None
+
+
+def test_find_matching_plan_postcode_with_no_space_in_address() -> None:
+    # Arrange - address has postcode without internal space
+    plans = [PLAN_A]
+    # Act
+    result = find_matching_plan(plans, "2 Laburnum Way Bromley BR28BZ")
+    # Assert
+    assert result is not None
+    assert result.id == "plan-a"
+
+
+def test_find_matching_plan_plan_postcode_with_no_space() -> None:
+    # Arrange - plan has postcode without space
+    plan = _make_plan("plan-c", street="Laburnum Way", street_number="2", postal_code="BR28BZ")
+    # Act
+    result = find_matching_plan([plan], "2 Laburnum Way Bromley BR2 8BZ")
+    # Assert
+    assert result is not None
+    assert result.id == "plan-c"
+
+
+def test_find_matching_plan_no_postcode_in_address_returns_none() -> None:
+    # Act
+    result = find_matching_plan([PLAN_A], "2 Laburnum Way Bromley")
+    # Assert
+    assert result is None
+
+
+def test_find_matching_plan_second_plan_matches() -> None:
+    # Arrange
+    plans = [PLAN_A, PLAN_B]
+    # Act
+    result = find_matching_plan(plans, "11 Station Road Bromley BR1 3LP")
+    # Assert
+    assert result is not None
+    assert result.id == "plan-b"