self-hosted/home-assistant-core
self-hosted/home-assistant-core
1
0
Fork 0
mirror of https://github.com/home-assistant/core synced 2024-10-05 22:48:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use last_step marker on UniFi options flow (#50053)

Browse source
This commit is contained in:
Robert Svensson 2021-05-04 08:54:45 +02:00 committed by GitHub
parent 016a4433d2
commit d579e3427f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
homeassistant/components/unifi/config_flow.py
View file

@ -299,6 +299,7 @@ class UnifiOptionsFlowHandler(config_entries.OptionsFlow):
): cv.multi_select(clients_to_block),
}
),
last_step=True,
)
async def async_step_device_tracker(self, user_input=None):
@ -354,6 +355,7 @@ class UnifiOptionsFlowHandler(config_entries.OptionsFlow):
): bool,
}
),
last_step=False,
)
async def async_step_client_control(self, user_input=None):
@ -391,6 +393,7 @@ class UnifiOptionsFlowHandler(config_entries.OptionsFlow):
}
),
errors=errors,
last_step=False,
)
async def async_step_statistics_sensors(self, user_input=None):
@ -413,6 +416,7 @@ class UnifiOptionsFlowHandler(config_entries.OptionsFlow):
): bool,
}
),
last_step=True,
)
async def _update_options(self):

4
tests/components/unifi/test_config_flow.py
View file

@ -448,6 +448,7 @@ async def test_advanced_option_flow(hass, aioclient_mock):
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
assert result["step_id"] == "device_tracker"
assert not result["last_step"]
assert set(
result["data_schema"].schema[CONF_SSID_FILTER].options.keys()
).intersection(("SSID 1", "SSID 2", "SSID 2_IOT", "SSID 3"))
@ -465,6 +466,7 @@ async def test_advanced_option_flow(hass, aioclient_mock):
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
assert result["step_id"] == "client_control"
assert not result["last_step"]
result = await hass.config_entries.options.async_configure(
result["flow_id"],
@ -477,6 +479,7 @@ async def test_advanced_option_flow(hass, aioclient_mock):
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
assert result["step_id"] == "statistics_sensors"
assert result["last_step"]
result = await hass.config_entries.options.async_configure(
result["flow_id"],
@ -519,6 +522,7 @@ async def test_simple_option_flow(hass, aioclient_mock):
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
assert result["step_id"] == "simple_options"
assert result["last_step"]
result = await hass.config_entries.options.async_configure(
result["flow_id"],