mirror of
https://github.com/home-assistant/core.git
synced 2025-04-25 09:47:52 +00:00
107 lines
3.7 KiB
Python
107 lines
3.7 KiB
Python
"""Test for vesync config flow."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.components.vesync import DOMAIN, config_flow
|
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.data_entry_flow import FlowResultType
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_abort_already_setup(hass: HomeAssistant) -> None:
|
|
"""Test if we abort because component is already setup."""
|
|
flow = config_flow.VeSyncFlowHandler()
|
|
flow.hass = hass
|
|
MockConfigEntry(domain=DOMAIN, title="user", data={"user": "pass"}).add_to_hass(
|
|
hass
|
|
)
|
|
result = await flow.async_step_user()
|
|
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "single_instance_allowed"
|
|
|
|
|
|
async def test_invalid_login_error(hass: HomeAssistant) -> None:
|
|
"""Test if we return error for invalid username and password."""
|
|
test_dict = {CONF_USERNAME: "user", CONF_PASSWORD: "pass"}
|
|
flow = config_flow.VeSyncFlowHandler()
|
|
flow.hass = hass
|
|
with patch("pyvesync.vesync.VeSync.login", return_value=False):
|
|
result = await flow.async_step_user(user_input=test_dict)
|
|
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == {"base": "invalid_auth"}
|
|
|
|
|
|
async def test_config_flow_user_input(hass: HomeAssistant) -> None:
|
|
"""Test config flow with user input."""
|
|
flow = config_flow.VeSyncFlowHandler()
|
|
flow.hass = hass
|
|
result = await flow.async_step_user()
|
|
assert result["type"] is FlowResultType.FORM
|
|
with patch("pyvesync.vesync.VeSync.login", return_value=True):
|
|
result = await flow.async_step_user(
|
|
{CONF_USERNAME: "user", CONF_PASSWORD: "pass"}
|
|
)
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result["data"][CONF_USERNAME] == "user"
|
|
assert result["data"][CONF_PASSWORD] == "pass"
|
|
|
|
|
|
async def test_reauth_flow(hass: HomeAssistant) -> None:
|
|
"""Test a successful reauth flow."""
|
|
mock_entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
unique_id="test-username",
|
|
)
|
|
mock_entry.add_to_hass(hass)
|
|
|
|
result = await mock_entry.start_reauth_flow(hass)
|
|
|
|
assert result["step_id"] == "reauth_confirm"
|
|
assert result["type"] is FlowResultType.FORM
|
|
with patch("pyvesync.vesync.VeSync.login", return_value=True):
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{CONF_USERNAME: "new-username", CONF_PASSWORD: "new-password"},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "reauth_successful"
|
|
assert mock_entry.data == {
|
|
CONF_USERNAME: "new-username",
|
|
CONF_PASSWORD: "new-password",
|
|
}
|
|
|
|
|
|
async def test_reauth_flow_invalid_auth(hass: HomeAssistant) -> None:
|
|
"""Test an authorization error reauth flow."""
|
|
|
|
mock_entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
unique_id="test-username",
|
|
)
|
|
mock_entry.add_to_hass(hass)
|
|
|
|
result = await mock_entry.start_reauth_flow(hass)
|
|
assert result["step_id"] == "reauth_confirm"
|
|
assert result["type"] is FlowResultType.FORM
|
|
|
|
with patch("pyvesync.vesync.VeSync.login", return_value=False):
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{CONF_USERNAME: "new-username", CONF_PASSWORD: "new-password"},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.FORM
|
|
with patch("pyvesync.vesync.VeSync.login", return_value=True):
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{CONF_USERNAME: "new-username", CONF_PASSWORD: "new-password"},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "reauth_successful"
|