diff --git a/homeassistant/components/anglian_water/config_flow.py b/homeassistant/components/anglian_water/config_flow.py index 246291c5b31..93f72a3b0fc 100644 --- a/homeassistant/components/anglian_water/config_flow.py +++ b/homeassistant/components/anglian_water/config_flow.py @@ -30,6 +30,7 @@ STEP_USER_DATA_SCHEMA = vol.Schema( vol.Required(CONF_PASSWORD): selector.TextSelector( selector.TextSelectorConfig(type=selector.TextSelectorType.PASSWORD) ), + vol.Required(CONF_ACCOUNT_NUMBER): selector.TextSelector(), } ) @@ -68,34 +69,19 @@ class AnglianWaterConfigFlow(ConfigFlow, domain=DOMAIN): self.hass, cookie_jar=CookieJar(quote_cookie=False), ), - account_number=user_input.get(CONF_ACCOUNT_NUMBER), + account_number=user_input[CONF_ACCOUNT_NUMBER], ) ) if isinstance(validation_response, BaseAuth): - account_number = ( - user_input.get(CONF_ACCOUNT_NUMBER) - or validation_response.account_number - ) - await self.async_set_unique_id(account_number) + await self.async_set_unique_id(user_input[CONF_ACCOUNT_NUMBER]) self._abort_if_unique_id_configured() return self.async_create_entry( - title=account_number, + title=user_input[CONF_ACCOUNT_NUMBER], data={ **user_input, CONF_ACCESS_TOKEN: validation_response.refresh_token, - CONF_ACCOUNT_NUMBER: account_number, }, ) - if validation_response == "smart_meter_unavailable": - return self.async_show_form( - step_id="user", - data_schema=STEP_USER_DATA_SCHEMA.extend( - { - vol.Required(CONF_ACCOUNT_NUMBER): selector.TextSelector(), - } - ), - errors={"base": validation_response}, - ) errors["base"] = validation_response return self.async_show_form( diff --git a/tests/components/anglian_water/test_config_flow.py b/tests/components/anglian_water/test_config_flow.py index 86b205987f3..d577a35880a 100644 --- a/tests/components/anglian_water/test_config_flow.py +++ b/tests/components/anglian_water/test_config_flow.py @@ -40,6 +40,7 @@ async def test_full_flow( user_input={ CONF_USERNAME: USERNAME, CONF_PASSWORD: PASSWORD, + CONF_ACCOUNT_NUMBER: ACCOUNT_NUMBER, }, ) @@ -74,6 +75,7 @@ async def test_already_configured( user_input={ CONF_USERNAME: USERNAME, CONF_PASSWORD: PASSWORD, + CONF_ACCOUNT_NUMBER: ACCOUNT_NUMBER, }, ) @@ -107,6 +109,7 @@ async def test_auth_recover_exception( user_input={ CONF_USERNAME: USERNAME, CONF_PASSWORD: PASSWORD, + CONF_ACCOUNT_NUMBER: ACCOUNT_NUMBER, }, ) @@ -123,6 +126,7 @@ async def test_auth_recover_exception( user_input={ CONF_USERNAME: USERNAME, CONF_PASSWORD: PASSWORD, + CONF_ACCOUNT_NUMBER: ACCOUNT_NUMBER, }, ) @@ -164,6 +168,7 @@ async def test_account_recover_exception( user_input={ CONF_USERNAME: USERNAME, CONF_PASSWORD: PASSWORD, + CONF_ACCOUNT_NUMBER: ACCOUNT_NUMBER, }, )