diff --git a/backend/magic_plan/tests/test_handler.py b/backend/magic_plan/tests/test_handler.py index 366f3ded..b0365f5b 100644 --- a/backend/magic_plan/tests/test_handler.py +++ b/backend/magic_plan/tests/test_handler.py @@ -54,7 +54,7 @@ def test_handler_raises_on_missing_address(mock_plan: MagicMock) -> None: def test_handler_constructs_client_from_settings(mock_service: MagicMock) -> None: # Arrange - body = {"address": ADDRESS} + body = {"address": ADDRESS, "hubspot_deal_id": "deal-123"} with patch("backend.magic_plan.handler.get_settings", return_value=_make_settings(customer_id="cust-xyz", api_key="key-xyz")), \ patch("backend.magic_plan.handler.MagicPlanClient") as MockClient, \ patch("backend.magic_plan.handler.MagicPlanService", return_value=mock_service): @@ -69,31 +69,37 @@ def test_handler_constructs_client_from_settings(mock_service: MagicMock) -> Non def test_handler_calls_service_run_with_address(mock_service: MagicMock) -> None: # Arrange - body = {"address": ADDRESS} + body = {"address": ADDRESS, "hubspot_deal_id": "deal-123"} with patch("backend.magic_plan.handler.get_settings", return_value=_make_settings()), \ patch("backend.magic_plan.handler.MagicPlanClient"), \ patch("backend.magic_plan.handler.MagicPlanService", return_value=mock_service): # Act _call_handler(body) # Assert - mock_service.run.assert_called_once_with(ADDRESS, None) + mock_service.run.assert_called_once() + request = mock_service.run.call_args.args[0] + assert request.address == ADDRESS + assert request.uprn is None def test_handler_passes_uprn_to_service(mock_service: MagicMock) -> None: # Arrange - body = {"address": ADDRESS, "uprn": "100023336956"} + body = {"address": ADDRESS, "uprn": "100023336956", "hubspot_deal_id": "deal-123"} with patch("backend.magic_plan.handler.get_settings", return_value=_make_settings()), \ patch("backend.magic_plan.handler.MagicPlanClient"), \ patch("backend.magic_plan.handler.MagicPlanService", return_value=mock_service): # Act _call_handler(body) # Assert - mock_service.run.assert_called_once_with(ADDRESS, "100023336956") + mock_service.run.assert_called_once() + request = mock_service.run.call_args.args[0] + assert request.address == ADDRESS + assert request.uprn == "100023336956" def test_handler_returns_plan_uid(mock_service: MagicMock) -> None: # Arrange - body = {"address": ADDRESS} + body = {"address": ADDRESS, "hubspot_deal_id": "deal-123"} with patch("backend.magic_plan.handler.get_settings", return_value=_make_settings()), \ patch("backend.magic_plan.handler.MagicPlanClient"), \ patch("backend.magic_plan.handler.MagicPlanService", return_value=mock_service):