Add initial version for the YouLess integration (#41942)

Co-authored-by: Franck Nijhof <frenck@frenck.nl>
Co-authored-by: Franck Nijhof <git@frenck.dev>
This commit is contained in:
gjong 2021-07-27 17:42:15 +02:00 committed by GitHub
parent ea9d312b45
commit 9f495fd200
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
14 changed files with 437 additions and 0 deletions

View file

@ -0,0 +1,72 @@
"""Test the youless config flow."""
from unittest.mock import MagicMock, patch
from urllib.error import URLError
from homeassistant.components.youless import DOMAIN
from homeassistant.config_entries import SOURCE_USER
from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import RESULT_TYPE_CREATE_ENTRY, RESULT_TYPE_FORM
def _get_mock_youless_api(initialize=None):
mock_youless = MagicMock()
if isinstance(initialize, Exception):
type(mock_youless).initialize = MagicMock(side_effect=initialize)
else:
type(mock_youless).initialize = MagicMock(return_value=initialize)
type(mock_youless).mac_address = None
return mock_youless
async def test_full_flow(hass: HomeAssistant) -> None:
"""Check setup."""
result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": SOURCE_USER}
)
assert result.get("type") == RESULT_TYPE_FORM
assert result.get("errors") == {}
assert result.get("step_id") == SOURCE_USER
assert "flow_id" in result
mock_youless = _get_mock_youless_api(
initialize={"homes": [{"id": 1, "name": "myhome"}]}
)
with patch(
"homeassistant.components.youless.config_flow.YoulessAPI",
return_value=mock_youless,
) as mocked_youless:
result2 = await hass.config_entries.flow.async_configure(
result["flow_id"],
{"host": "localhost"},
)
assert result2.get("type") == RESULT_TYPE_CREATE_ENTRY
assert result2.get("title") == "localhost"
assert len(mocked_youless.mock_calls) == 1
async def test_not_found(hass: HomeAssistant) -> None:
"""Check setup."""
result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": SOURCE_USER}
)
assert result.get("type") == RESULT_TYPE_FORM
assert result.get("errors") == {}
assert result.get("step_id") == SOURCE_USER
assert "flow_id" in result
mock_youless = _get_mock_youless_api(initialize=URLError(""))
with patch(
"homeassistant.components.youless.config_flow.YoulessAPI",
return_value=mock_youless,
) as mocked_youless:
result2 = await hass.config_entries.flow.async_configure(
result["flow_id"],
{"host": "localhost"},
)
assert result2.get("type") == RESULT_TYPE_FORM
assert len(mocked_youless.mock_calls) == 1