async def test_step_user(hass):
"""Test that the user step works."""
conf = {
CONF_USERNAME: "******",
CONF_PASSWORD: "******",
CONF_SCAN_INTERVAL: timedelta(seconds=90),
}
flow = config_flow.SimpliSafeFlowHandler()
flow.hass = hass
mop = mock_open(read_data=json.dumps({"refresh_token": "12345"}))
with patch(
"simplipy.API.login_via_credentials",
return_value=mock_coro(return_value=mock_api()),
):
with patch("homeassistant.util.json.open", mop, create=True):
with patch("homeassistant.util.json.os.open", return_value=0):
with patch("homeassistant.util.json.os.replace"):
result = await flow.async_step_user(user_input=conf)
assert result[
"type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
assert result["title"] == "*****@*****.**"
assert result["data"] == {
CONF_USERNAME: "******",
CONF_TOKEN: "12345abc",
CONF_SCAN_INTERVAL: 90,
}
async def test_step_import(hass):
"""Test that the import step works."""
conf = {CONF_USERNAME: "******", CONF_PASSWORD: "******"}
flow = config_flow.SimpliSafeFlowHandler()
flow.hass = hass
flow.context = {"source": SOURCE_USER}
mop = mock_open(read_data=json.dumps({"refresh_token": "12345"}))
with patch(
"simplipy.API.login_via_credentials",
return_value=mock_coro(return_value=mock_api()),
):
with patch("homeassistant.util.json.open", mop, create=True):
with patch("homeassistant.util.json.os.open", return_value=0):
with patch("homeassistant.util.json.os.replace"):
result = await flow.async_step_import(import_config=conf)
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
assert result["title"] == "*****@*****.**"
assert result["data"] == {
CONF_USERNAME: "******",
CONF_TOKEN: "12345abc",
}
async def test_show_form(hass):
"""Test that the form is served with no input."""
flow = config_flow.SimpliSafeFlowHandler()
flow.hass = hass
result = await flow.async_step_user(user_input=None)
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
assert result["step_id"] == "user"
async def test_duplicate_error(hass):
"""Test that errors are shown when duplicates are added."""
conf = {CONF_USERNAME: "******", CONF_PASSWORD: "******"}
MockConfigEntry(domain=DOMAIN, data=conf).add_to_hass(hass)
flow = config_flow.SimpliSafeFlowHandler()
flow.hass = hass
result = await flow.async_step_user(user_input=conf)
assert result["errors"] == {CONF_USERNAME: "******"}
async def test_invalid_credentials(hass):
"""Test that invalid credentials throws an error."""
conf = {CONF_USERNAME: "******", CONF_PASSWORD: "******"}
flow = config_flow.SimpliSafeFlowHandler()
flow.hass = hass
flow.context = {"source": SOURCE_USER}
with patch(
"simplipy.API.login_via_credentials",
return_value=mock_coro(exception=SimplipyError),
):
result = await flow.async_step_user(user_input=conf)
assert result["errors"] == {"base": "invalid_credentials"}
