jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-08-02 10:08:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Patch both connection objects

Browse Source
This commit is contained in:
Robin Lintermann 2025-05-15 14:32:21 +00:00
parent 6bbd357547
commit f0a3ecd27b
2 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/components/smarla/conftest.py
View File

@ -27,11 +27,17 @@ def mock_config_entry() -> MockConfigEntry:
@pytest.fixture @pytest.fixture
def mock_cf_connection(): def mock_connection():
"""Patch config_flow Connection object.""" """Patch Connection object."""
with patch( with (
"homeassistant.components.smarla.config_flow.Connection" patch(
) as mock_connection: "homeassistant.components.smarla.config_flow.Connection"
) as mock_connection,
patch(
"homeassistant.components.smarla.Connection",
mock_connection,
),
):
connection = MagicMock() connection = MagicMock()
connection.token = AuthToken.from_json(MOCK_ACCESS_TOKEN_JSON) connection.token = AuthToken.from_json(MOCK_ACCESS_TOKEN_JSON)
connection.refresh_token = AsyncMock(return_value=True) connection.refresh_token = AsyncMock(return_value=True)

8
tests/components/smarla/test_config_flow.py
View File

@ -14,7 +14,7 @@ from . import MOCK_SERIAL_NUMBER, MOCK_USER_INPUT
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
async def test_config_flow(hass: HomeAssistant, mock_cf_connection) -> None: async def test_config_flow(hass: HomeAssistant, mock_connection) -> None:
"""Test creating a config entry.""" """Test creating a config entry."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": SOURCE_USER} DOMAIN, context={"source": SOURCE_USER}
@ -34,7 +34,7 @@ async def test_config_flow(hass: HomeAssistant, mock_cf_connection) -> None:
@pytest.mark.parametrize("error", ["malformed_token", "invalid_auth"]) @pytest.mark.parametrize("error", ["malformed_token", "invalid_auth"])
async def test_form_error(hass: HomeAssistant, error: str, mock_cf_connection) -> None: async def test_form_error(hass: HomeAssistant, error: str, mock_connection) -> None:
"""Test we show user form on invalid auth.""" """Test we show user form on invalid auth."""
match error: match error:
case "malformed_token": case "malformed_token":
@ -44,7 +44,7 @@ async def test_form_error(hass: HomeAssistant, error: str, mock_cf_connection) -
) )
case "invalid_auth": case "invalid_auth":
error_patch = patch.object( error_patch = patch.object(
mock_cf_connection, mock_connection,
"refresh_token", "refresh_token",
new=AsyncMock(return_value=False), new=AsyncMock(return_value=False),
) )
@ -68,7 +68,7 @@ async def test_form_error(hass: HomeAssistant, error: str, mock_cf_connection) -
async def test_device_exists_abort( async def test_device_exists_abort(
hass: HomeAssistant, mock_config_entry: MockConfigEntry, mock_cf_connection hass: HomeAssistant, mock_config_entry: MockConfigEntry, mock_connection
) -> None: ) -> None:
"""Test we abort config flow if Smarla device already configured.""" """Test we abort config flow if Smarla device already configured."""
mock_config_entry.add_to_hass(hass) mock_config_entry.add_to_hass(hass)