jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-21 20:27:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix hvv_departures config flow patches (#40095)

Browse Source
This commit is contained in:
Martin Hjelmare 2020-09-15 12:37:31 +02:00 committed by GitHub
parent fffc7e2e8e
commit 487b56ab69
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 29 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
tests/components/hvv_departures/test_config_flow.py
View File

@ -32,8 +32,11 @@ async def test_user_flow(hass):
with patch( with patch(
"homeassistant.components.hvv_departures.hub.GTI.init", "homeassistant.components.hvv_departures.hub.GTI.init",
return_value=FIXTURE_INIT, return_value=FIXTURE_INIT,
), patch("pygti.gti.GTI.checkName", return_value=FIXTURE_CHECK_NAME,), patch( ), patch(
"pygti.gti.GTI.stationInformation", "homeassistant.components.hvv_departures.hub.GTI.checkName",
return_value=FIXTURE_CHECK_NAME,
), patch(
"homeassistant.components.hvv_departures.hub.GTI.stationInformation",
return_value=FIXTURE_STATION_INFORMATION, return_value=FIXTURE_STATION_INFORMATION,
), patch( ), patch(
"homeassistant.components.hvv_departures.async_setup", return_value=True "homeassistant.components.hvv_departures.async_setup", return_value=True
@ -96,7 +99,7 @@ async def test_user_flow_no_results(hass):
"homeassistant.components.hvv_departures.hub.GTI.init", "homeassistant.components.hvv_departures.hub.GTI.init",
return_value=FIXTURE_INIT, return_value=FIXTURE_INIT,
), patch( ), patch(
"pygti.gti.GTI.checkName", "homeassistant.components.hvv_departures.hub.GTI.checkName",
return_value={"returnCode": "OK", "results": []}, return_value={"returnCode": "OK", "results": []},
), patch( ), patch(
"homeassistant.components.hvv_departures.async_setup", return_value=True "homeassistant.components.hvv_departures.async_setup", return_value=True
@ -186,7 +189,7 @@ async def test_user_flow_station(hass):
"homeassistant.components.hvv_departures.hub.GTI.init", "homeassistant.components.hvv_departures.hub.GTI.init",
return_value=True, return_value=True,
), patch( ), patch(
"pygti.gti.GTI.checkName", "homeassistant.components.hvv_departures.hub.GTI.checkName",
return_value={"returnCode": "OK", "results": []}, return_value={"returnCode": "OK", "results": []},
): ):
@ -220,7 +223,7 @@ async def test_user_flow_station_select(hass):
"homeassistant.components.hvv_departures.hub.GTI.init", "homeassistant.components.hvv_departures.hub.GTI.init",
return_value=True, return_value=True,
), patch( ), patch(
"pygti.gti.GTI.checkName", "homeassistant.components.hvv_departures.hub.GTI.checkName",
return_value=FIXTURE_CHECK_NAME, return_value=FIXTURE_CHECK_NAME,
): ):
result_user = await hass.config_entries.flow.async_init( result_user = await hass.config_entries.flow.async_init(
@ -268,10 +271,11 @@ async def test_options_flow(hass):
"homeassistant.components.hvv_departures.hub.GTI.init", "homeassistant.components.hvv_departures.hub.GTI.init",
return_value=True, return_value=True,
), patch( ), patch(
"pygti.gti.GTI.departureList", "homeassistant.components.hvv_departures.hub.GTI.departureList",
return_value=FIXTURE_DEPARTURE_LIST, return_value=FIXTURE_DEPARTURE_LIST,
): ):
assert await hass.config_entries.async_setup(config_entry.entry_id) assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()
result = await hass.config_entries.options.async_init(config_entry.entry_id) result = await hass.config_entries.options.async_init(config_entry.entry_id)
@ -315,14 +319,22 @@ async def test_options_flow_invalid_auth(hass):
config_entry.add_to_hass(hass) config_entry.add_to_hass(hass)
with patch( with patch(
"homeassistant.components.hvv_departures.hub.GTI.init", "homeassistant.components.hvv_departures.hub.GTI.init", return_value=True
), patch(
"homeassistant.components.hvv_departures.hub.GTI.departureList",
return_value=FIXTURE_DEPARTURE_LIST,
):
assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()
with patch(
"homeassistant.components.hvv_departures.hub.GTI.departureList",
side_effect=InvalidAuth( side_effect=InvalidAuth(
"ERROR_TEXT", "ERROR_TEXT",
"Bei der Verarbeitung der Anfrage ist ein technisches Problem aufgetreten.", "Bei der Verarbeitung der Anfrage ist ein technisches Problem aufgetreten.",
"Authentication failed!", "Authentication failed!",
), ),
): ):
assert await hass.config_entries.async_setup(config_entry.entry_id)
result = await hass.config_entries.options.async_init(config_entry.entry_id) result = await hass.config_entries.options.async_init(config_entry.entry_id)
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
@ -348,11 +360,18 @@ async def test_options_flow_cannot_connect(hass):
config_entry.add_to_hass(hass) config_entry.add_to_hass(hass)
with patch( with patch(
"pygti.gti.GTI.departureList", "homeassistant.components.hvv_departures.hub.GTI.init", return_value=True
side_effect=CannotConnect(), ), patch(
"homeassistant.components.hvv_departures.hub.GTI.departureList",
return_value=FIXTURE_DEPARTURE_LIST,
): ):
assert await hass.config_entries.async_setup(config_entry.entry_id) assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()
with patch(
"homeassistant.components.hvv_departures.hub.GTI.departureList",
side_effect=CannotConnect(),
):
result = await hass.config_entries.options.async_init(config_entry.entry_id) result = await hass.config_entries.options.async_init(config_entry.entry_id)
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM assert result["type"] == data_entry_flow.RESULT_TYPE_FORM