138 lines
4.5 KiB
Python
138 lines
4.5 KiB
Python
"""Test the Rachio config flow."""
|
|
from homeassistant import config_entries, setup
|
|
from homeassistant.components.rachio.const import (
|
|
CONF_CUSTOM_URL,
|
|
CONF_MANUAL_RUN_MINS,
|
|
DOMAIN,
|
|
)
|
|
from homeassistant.const import CONF_API_KEY
|
|
|
|
from tests.async_mock import MagicMock, patch
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
def _mock_rachio_return_value(get=None, getInfo=None):
|
|
rachio_mock = MagicMock()
|
|
person_mock = MagicMock()
|
|
type(person_mock).get = MagicMock(return_value=get)
|
|
type(person_mock).getInfo = MagicMock(return_value=getInfo)
|
|
type(rachio_mock).person = person_mock
|
|
return rachio_mock
|
|
|
|
|
|
async def test_form(hass):
|
|
"""Test we get the form."""
|
|
await setup.async_setup_component(hass, "persistent_notification", {})
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
|
)
|
|
assert result["type"] == "form"
|
|
assert result["errors"] == {}
|
|
|
|
rachio_mock = _mock_rachio_return_value(
|
|
get=({"status": 200}, {"username": "myusername"}),
|
|
getInfo=({"status": 200}, {"id": "myid"}),
|
|
)
|
|
|
|
with patch(
|
|
"homeassistant.components.rachio.config_flow.Rachio", return_value=rachio_mock
|
|
), patch(
|
|
"homeassistant.components.rachio.async_setup", return_value=True
|
|
) as mock_setup, patch(
|
|
"homeassistant.components.rachio.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_API_KEY: "api_key",
|
|
CONF_CUSTOM_URL: "http://custom.url",
|
|
CONF_MANUAL_RUN_MINS: 5,
|
|
},
|
|
)
|
|
|
|
assert result2["type"] == "create_entry"
|
|
assert result2["title"] == "myusername"
|
|
assert result2["data"] == {
|
|
CONF_API_KEY: "api_key",
|
|
CONF_CUSTOM_URL: "http://custom.url",
|
|
CONF_MANUAL_RUN_MINS: 5,
|
|
}
|
|
await hass.async_block_till_done()
|
|
assert len(mock_setup.mock_calls) == 1
|
|
assert len(mock_setup_entry.mock_calls) == 1
|
|
|
|
|
|
async def test_form_invalid_auth(hass):
|
|
"""Test we handle invalid auth."""
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
|
)
|
|
|
|
rachio_mock = _mock_rachio_return_value(
|
|
get=({"status": 200}, {"username": "myusername"}),
|
|
getInfo=({"status": 412}, {"error": "auth fail"}),
|
|
)
|
|
with patch(
|
|
"homeassistant.components.rachio.config_flow.Rachio", return_value=rachio_mock
|
|
):
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{CONF_API_KEY: "api_key"},
|
|
)
|
|
|
|
assert result2["type"] == "form"
|
|
assert result2["errors"] == {"base": "invalid_auth"}
|
|
|
|
|
|
async def test_form_cannot_connect(hass):
|
|
"""Test we handle cannot connect error."""
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
|
)
|
|
|
|
rachio_mock = _mock_rachio_return_value(
|
|
get=({"status": 599}, {"username": "myusername"}),
|
|
getInfo=({"status": 200}, {"id": "myid"}),
|
|
)
|
|
with patch(
|
|
"homeassistant.components.rachio.config_flow.Rachio", return_value=rachio_mock
|
|
):
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{CONF_API_KEY: "api_key"},
|
|
)
|
|
|
|
assert result2["type"] == "form"
|
|
assert result2["errors"] == {"base": "cannot_connect"}
|
|
|
|
|
|
async def test_form_homekit(hass):
|
|
"""Test that we abort from homekit if rachio is already setup."""
|
|
await setup.async_setup_component(hass, "persistent_notification", {})
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": "homekit"},
|
|
data={"properties": {"id": "AA:BB:CC:DD:EE:FF"}},
|
|
)
|
|
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_API_KEY: "api_key"})
|
|
entry.add_to_hass(hass)
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": "homekit"},
|
|
data={"properties": {"id": "AA:BB:CC:DD:EE:FF"}},
|
|
)
|
|
assert result["type"] == "abort"
|
|
assert result["reason"] == "already_configured"
|