evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Allow next_step to be a string (#82647)

Browse source 
* Allow next_step to be a string

* Adjust docstrings

* Add test
This commit is contained in:
epenet 2022-11-24 18:55:53 +01:00 • committed by GitHub
parent 285aff154d
commit caa99ea9fb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 25 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/helpers/test_schema_config_entry_flow.py
View file

@ -29,7 +29,8 @@ async def test_menu_step(hass: HomeAssistant) -> None:
"user": SchemaFlowMenuStep(MENU_1),
"option1": SchemaFlowFormStep(vol.Schema({}), next_step=lambda _: "menu2"),
"menu2": SchemaFlowMenuStep(MENU_2),
"option3": SchemaFlowFormStep(vol.Schema({})),
"option3": SchemaFlowFormStep(vol.Schema({}), next_step="option4"),
"option4": SchemaFlowFormStep(vol.Schema({})),
}
class TestConfigFlow(SchemaConfigFlowHandler, domain=TEST_DOMAIN):
@ -63,5 +64,9 @@ async def test_menu_step(hass: HomeAssistant) -> None:
assert result["type"] == FlowResultType.FORM
assert result["step_id"] == "option3"
result = await hass.config_entries.flow.async_configure(result["flow_id"], {})
assert result["type"] == FlowResultType.FORM
assert result["step_id"] == "option4"
result = await hass.config_entries.flow.async_configure(result["flow_id"], {})
assert result["type"] == FlowResultType.CREATE_ENTRY