mirror of
https://github.com/home-assistant/core.git
synced 2025-08-11 08:35:15 +02:00
Delete test_config_flow.py
Reverting the tests, which will be placed in a different PR.
This commit is contained in:
Erwin Douna
GitHub
@@ -1,214 +0,0 @@
|
|||||||
"""Test the Tado config flow."""
|
|
||||||
from http import HTTPStatus
|
|
||||||
from unittest.mock import MagicMock, patch
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
import requests
|
|
||||||
|
|
||||||
from homeassistant import config_entries
|
|
||||||
from homeassistant.components import zeroconf
|
|
||||||
from homeassistant.components.tado.const import DOMAIN
|
|
||||||
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
|
||||||
from homeassistant.core import HomeAssistant
|
|
||||||
|
|
||||||
from tests.common import MockConfigEntry
|
|
||||||
|
|
||||||
|
|
||||||
def _get_mock_tado_api(getMe=None):
|
|
||||||
mock_tado = MagicMock()
|
|
||||||
if isinstance(getMe, Exception):
|
|
||||||
type(mock_tado).getMe = MagicMock(side_effect=getMe)
|
|
||||||
else:
|
|
||||||
type(mock_tado).getMe = MagicMock(return_value=getMe)
|
|
||||||
return mock_tado
|
|
||||||
|
|
||||||
|
|
||||||
async def test_form(hass: HomeAssistant) -> None:
|
|
||||||
"""Test we can setup though the user path."""
|
|
||||||
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
|
||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
|
||||||
)
|
|
||||||
assert result["type"] == "form"
|
|
||||||
assert result["errors"] == {}
|
|
||||||
|
|
||||||
mock_tado_api = _get_mock_tado_api(getMe={"homes": [{"id": 1, "name": "myhome"}]})
|
|
||||||
|
|
||||||
with patch(
|
|
||||||
"homeassistant.components.tado.config_flow.Tado",
|
|
||||||
return_value=mock_tado_api,
|
|
||||||
), patch(
|
|
||||||
"homeassistant.components.tado.async_setup_entry",
|
|
||||||
return_value=True,
|
|
||||||
) as mock_setup_entry:
|
|
||||||
result2 = await hass.config_entries.flow.async_configure(
|
|
||||||
result["flow_id"],
|
|
||||||
{"username": "test-username", "password": "test-password"},
|
|
||||||
)
|
|
||||||
await hass.async_block_till_done()
|
|
||||||
|
|
||||||
assert result2["type"] == "create_entry"
|
|
||||||
assert result2["title"] == "myhome"
|
|
||||||
assert result2["data"] == {
|
|
||||||
"username": "test-username",
|
|
||||||
"password": "test-password",
|
|
||||||
}
|
|
||||||
assert len(mock_setup_entry.mock_calls) == 1
|
|
||||||
|
|
||||||
|
|
||||||
async def test_form_invalid_auth(hass: HomeAssistant) -> None:
|
|
||||||
"""Test we handle invalid auth."""
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
|
||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
|
||||||
)
|
|
||||||
|
|
||||||
response_mock = MagicMock()
|
|
||||||
type(response_mock).status_code = HTTPStatus.UNAUTHORIZED
|
|
||||||
mock_tado_api = _get_mock_tado_api(getMe=requests.HTTPError(response=response_mock))
|
|
||||||
|
|
||||||
with patch(
|
|
||||||
"homeassistant.components.tado.config_flow.Tado",
|
|
||||||
return_value=mock_tado_api,
|
|
||||||
):
|
|
||||||
result2 = await hass.config_entries.flow.async_configure(
|
|
||||||
result["flow_id"],
|
|
||||||
{"username": "test-username", "password": "test-password"},
|
|
||||||
)
|
|
||||||
|
|
||||||
assert result2["type"] == "form"
|
|
||||||
assert result2["errors"] == {"base": "invalid_auth"}
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
"error",
|
|
||||||
[
|
|
||||||
(KeyError, "invalid_auth"),
|
|
||||||
(RuntimeError, "cannot_connect"),
|
|
||||||
(ValueError, "unknown"),
|
|
||||||
],
|
|
||||||
)
|
|
||||||
async def test_form_exceptions(hass: HomeAssistant, error) -> None:
|
|
||||||
"""Test we handle Exception."""
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
|
||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
|
||||||
)
|
|
||||||
|
|
||||||
exc, base_error = error
|
|
||||||
with patch(
|
|
||||||
"homeassistant.components.tado.config_flow.Tado",
|
|
||||||
side_effect=exc,
|
|
||||||
):
|
|
||||||
result2 = await hass.config_entries.flow.async_configure(
|
|
||||||
result["flow_id"],
|
|
||||||
{"username": "test-username", "password": "test-password"},
|
|
||||||
)
|
|
||||||
|
|
||||||
assert result2["type"] == "form"
|
|
||||||
assert result2["errors"] == {"base": base_error}
|
|
||||||
|
|
||||||
|
|
||||||
async def test_options_flow(hass: HomeAssistant) -> None:
|
|
||||||
"""Test config flow options."""
|
|
||||||
entry = MockConfigEntry(domain=DOMAIN, data={"username": "test-username"})
|
|
||||||
entry.add_to_hass(hass)
|
|
||||||
|
|
||||||
with patch(
|
|
||||||
"homeassistant.components.tado.async_setup_entry",
|
|
||||||
return_value=True,
|
|
||||||
):
|
|
||||||
await hass.config_entries.async_setup(entry.entry_id)
|
|
||||||
await hass.async_block_till_done()
|
|
||||||
|
|
||||||
result = await hass.config_entries.options.async_init(entry.entry_id)
|
|
||||||
assert result["type"] == "form"
|
|
||||||
assert result["step_id"] == "init"
|
|
||||||
|
|
||||||
|
|
||||||
async def test_form_cannot_connect(hass: HomeAssistant) -> None:
|
|
||||||
"""Test we handle cannot connect error."""
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
|
||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
|
||||||
)
|
|
||||||
|
|
||||||
response_mock = MagicMock()
|
|
||||||
type(response_mock).status_code = HTTPStatus.INTERNAL_SERVER_ERROR
|
|
||||||
mock_tado_api = _get_mock_tado_api(getMe=requests.HTTPError(response=response_mock))
|
|
||||||
|
|
||||||
with patch(
|
|
||||||
"homeassistant.components.tado.config_flow.Tado",
|
|
||||||
return_value=mock_tado_api,
|
|
||||||
):
|
|
||||||
result2 = await hass.config_entries.flow.async_configure(
|
|
||||||
result["flow_id"],
|
|
||||||
{"username": "test-username", "password": "test-password"},
|
|
||||||
)
|
|
||||||
|
|
||||||
assert result2["type"] == "form"
|
|
||||||
assert result2["errors"] == {"base": "cannot_connect"}
|
|
||||||
|
|
||||||
|
|
||||||
async def test_no_homes(hass: HomeAssistant) -> None:
|
|
||||||
"""Test we handle no homes error."""
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
|
||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
|
||||||
)
|
|
||||||
|
|
||||||
mock_tado_api = _get_mock_tado_api(getMe={"homes": []})
|
|
||||||
|
|
||||||
with patch(
|
|
||||||
"homeassistant.components.tado.config_flow.Tado",
|
|
||||||
return_value=mock_tado_api,
|
|
||||||
):
|
|
||||||
result2 = await hass.config_entries.flow.async_configure(
|
|
||||||
result["flow_id"],
|
|
||||||
{"username": "test-username", "password": "test-password"},
|
|
||||||
)
|
|
||||||
|
|
||||||
assert result2["type"] == "form"
|
|
||||||
assert result2["errors"] == {"base": "no_homes"}
|
|
||||||
|
|
||||||
|
|
||||||
async def test_form_homekit(hass: HomeAssistant) -> None:
|
|
||||||
"""Test that we abort from homekit if tado is already setup."""
|
|
||||||
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
|
||||||
DOMAIN,
|
|
||||||
context={"source": config_entries.SOURCE_HOMEKIT},
|
|
||||||
data=zeroconf.ZeroconfServiceInfo(
|
|
||||||
host="mock_host",
|
|
||||||
addresses=["mock_host"],
|
|
||||||
hostname="mock_hostname",
|
|
||||||
name="mock_name",
|
|
||||||
port=None,
|
|
||||||
properties={zeroconf.ATTR_PROPERTIES_ID: "AA:BB:CC:DD:EE:FF"},
|
|
||||||
type="mock_type",
|
|
||||||
),
|
|
||||||
)
|
|
||||||
assert result["type"] == "form"
|
|
||||||
assert result["errors"] == {}
|
|
||||||
flow = next(
|
|
||||||
flow
|
|
||||||
for flow in hass.config_entries.flow.async_progress()
|
|
||||||
if flow["flow_id"] == result["flow_id"]
|
|
||||||
)
|
|
||||||
assert flow["context"]["unique_id"] == "AA:BB:CC:DD:EE:FF"
|
|
||||||
|
|
||||||
entry = MockConfigEntry(
|
|
||||||
domain=DOMAIN, data={CONF_USERNAME: "mock", CONF_PASSWORD: "mock"}
|
|
||||||
)
|
|
||||||
entry.add_to_hass(hass)
|
|
||||||
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
|
||||||
DOMAIN,
|
|
||||||
context={"source": config_entries.SOURCE_HOMEKIT},
|
|
||||||
data=zeroconf.ZeroconfServiceInfo(
|
|
||||||
host="mock_host",
|
|
||||||
addresses=["mock_host"],
|
|
||||||
hostname="mock_hostname",
|
|
||||||
name="mock_name",
|
|
||||||
port=None,
|
|
||||||
properties={zeroconf.ATTR_PROPERTIES_ID: "AA:BB:CC:DD:EE:FF"},
|
|
||||||
type="mock_type",
|
|
||||||
),
|
|
||||||
)
|
|
||||||
assert result["type"] == "abort"
|
|
||||||
Reference in New Issue
Block a user