jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-14 16:57:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix custom account config flow setup (#127750)

Browse Source
This commit is contained in:
Ricardo Marques 2024-10-08 16:44:59 +01:00 committed by GitHub
parent 00a037c786
commit d41b9beb71
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
homeassistant/components/ovo_energy/__init__.py
View File

@ -32,7 +32,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
client_session=async_get_clientsession(hass), client_session=async_get_clientsession(hass),
) )
if custom_account := entry.data.get(CONF_ACCOUNT) is not None: if (custom_account := entry.data.get(CONF_ACCOUNT)) is not None:
client.custom_account_id = custom_account client.custom_account_id = custom_account
try: try:
@ -49,7 +49,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async def async_update_data() -> OVODailyUsage: async def async_update_data() -> OVODailyUsage:
"""Fetch data from OVO Energy.""" """Fetch data from OVO Energy."""
if custom_account := entry.data.get(CONF_ACCOUNT) is not None: if (custom_account := entry.data.get(CONF_ACCOUNT)) is not None:
client.custom_account_id = custom_account client.custom_account_id = custom_account
async with asyncio.timeout(10): async with asyncio.timeout(10):

2
homeassistant/components/ovo_energy/config_flow.py
View File

@ -46,7 +46,7 @@ class OVOEnergyFlowHandler(ConfigFlow, domain=DOMAIN):
client_session=async_get_clientsession(self.hass), client_session=async_get_clientsession(self.hass),
) )
if custom_account := user_input.get(CONF_ACCOUNT) is not None: if (custom_account := user_input.get(CONF_ACCOUNT)) is not None:
client.custom_account_id = custom_account client.custom_account_id = custom_account
try: try:

1
tests/components/ovo_energy/test_config_flow.py
View File

@ -117,6 +117,7 @@ async def test_full_flow_implementation(hass: HomeAssistant) -> None:
assert result2["type"] is FlowResultType.CREATE_ENTRY assert result2["type"] is FlowResultType.CREATE_ENTRY
assert result2["data"][CONF_USERNAME] == FIXTURE_USER_INPUT[CONF_USERNAME] assert result2["data"][CONF_USERNAME] == FIXTURE_USER_INPUT[CONF_USERNAME]
assert result2["data"][CONF_PASSWORD] == FIXTURE_USER_INPUT[CONF_PASSWORD] assert result2["data"][CONF_PASSWORD] == FIXTURE_USER_INPUT[CONF_PASSWORD]
assert result2["data"][CONF_ACCOUNT] == FIXTURE_USER_INPUT[CONF_ACCOUNT]
async def test_reauth_authorization_error(hass: HomeAssistant) -> None: async def test_reauth_authorization_error(hass: HomeAssistant) -> None: