From c3aae8fd51373a9b38cbdb023275bbd12172519c Mon Sep 17 00:00:00 2001
From: Daniel Roth <daniel_roth@hotmail.co.uk>
Date: Fri, 8 May 2026 14:08:56 +0000
Subject: [PATCH] =?UTF-8?q?Expose=20get=5Fplan=5Fraw=20method=20on=20Magic?=
 =?UTF-8?q?PlanClient=20=F0=9F=9F=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/magic_plan/magic_plan_client.py       |  3 ++
 .../tests/test_magic_plan_client.py           | 52 +++++++++++++++++++
 2 files changed, 55 insertions(+)

diff --git a/backend/magic_plan/magic_plan_client.py b/backend/magic_plan/magic_plan_client.py
index 60f70fb1..172190fd 100644
--- a/backend/magic_plan/magic_plan_client.py
+++ b/backend/magic_plan/magic_plan_client.py
@@ -22,3 +22,6 @@ class MagicPlanClient:
         )
         r.raise_for_status()
         return MagicPlanPlan.model_validate(r.json()["data"])
+
+    def get_plan_raw(self, plan_id: str) -> bytes:
+        raise NotImplementedError
diff --git a/backend/magic_plan/tests/test_magic_plan_client.py b/backend/magic_plan/tests/test_magic_plan_client.py
index 1be1448f..c96b9cdf 100644
--- a/backend/magic_plan/tests/test_magic_plan_client.py
+++ b/backend/magic_plan/tests/test_magic_plan_client.py
@@ -172,3 +172,55 @@ def test_get_plan_propagates_http_error(
     # Act / Assert
     with pytest.raises(requests.HTTPError):
         client.get_plan("some-id")
+
+
+# --- get_plan_raw ---
+
+
+def test_get_plan_raw_returns_bytes(
+    client: MagicPlanClient, mock_session: MagicMock
+) -> None:
+    # Arrange
+    mock_session.get.return_value.content = b'{"data": "raw"}'
+    plan_id = "a7285ed1-878d-47eb-8aa6-85ef9e187516"
+    # Act
+    result = client.get_plan_raw(plan_id)
+    # Assert
+    assert isinstance(result, bytes)
+
+
+def test_get_plan_raw_calls_correct_url(
+    client: MagicPlanClient, mock_session: MagicMock
+) -> None:
+    # Arrange
+    mock_session.get.return_value.content = b"{}"
+    plan_id = "a7285ed1-878d-47eb-8aa6-85ef9e187516"
+    # Act
+    client.get_plan_raw(plan_id)
+    # Assert
+    mock_session.get.assert_called_once_with(
+        f"{BASE_URL}/plans/{plan_id}", params={"key": API_KEY}
+    )
+
+
+def test_get_plan_raw_calls_raise_for_status(
+    client: MagicPlanClient, mock_session: MagicMock
+) -> None:
+    # Arrange
+    mock_session.get.return_value.content = b"{}"
+    # Act
+    client.get_plan_raw("a7285ed1-878d-47eb-8aa6-85ef9e187516")
+    # Assert
+    mock_session.get.return_value.raise_for_status.assert_called_once()
+
+
+def test_get_plan_raw_propagates_http_error(
+    client: MagicPlanClient, mock_session: MagicMock
+) -> None:
+    # Arrange
+    mock_session.get.return_value.raise_for_status.side_effect = requests.HTTPError(
+        "500"
+    )
+    # Act / Assert
+    with pytest.raises(requests.HTTPError):
+        client.get_plan_raw("some-id")