junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-08 11:17:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Set API key as session header on MagicPlanClient construction 🟥

Browse source
This commit is contained in:
Daniel Roth 2026-05-12 13:58:16 +00:00
parent a672c0dea0
commit da4f5f44c0
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
backend/magic_plan/tests/test_magic_plan_client.py
View file

@ -20,6 +20,7 @@ def _load_fixture(name: str) -> dict[str, Any]:
def _make_client(mock_session: MagicMock) -> MagicPlanClient: def _make_client(mock_session: MagicMock) -> MagicPlanClient:
mock_session.headers = {}
with patch( with patch(
"backend.magic_plan.magic_plan_client.requests.Session", "backend.magic_plan.magic_plan_client.requests.Session",
return_value=mock_session, return_value=mock_session,
@ -44,7 +45,14 @@ def test_customer_header_set_on_session(mock_session: MagicMock) -> None:
# Act # Act
_make_client(mock_session) _make_client(mock_session)
# Assert # Assert
mock_session.headers.update.assert_called_once_with({"customer": CUSTOMER_ID}) assert mock_session.headers["customer"] == CUSTOMER_ID
def test_api_key_header_set_on_session(mock_session: MagicMock) -> None:
# Act
_make_client(mock_session)
# Assert
assert mock_session.headers["key"] == API_KEY
# --- get_plans --- # --- get_plans ---