diff --git a/tests/components/rova/test_config_flow.py b/tests/components/rova/test_config_flow.py index 608f4ec105b9..3f8c456981b9 100644 --- a/tests/components/rova/test_config_flow.py +++ b/tests/components/rova/test_config_flow.py @@ -31,10 +31,9 @@ async def test_user(hass: HomeAssistant, mock_rova: MagicMock) -> None: assert result.get("step_id") == "user" # test with all information provided - result = await hass.config_entries.flow.async_init( - DOMAIN, - context={"source": SOURCE_USER}, - data={ + result = await hass.config_entries.flow.async_configure( + result["flow_id"], + user_input={ CONF_ZIP_CODE: ZIP_CODE, CONF_HOUSE_NUMBER: HOUSE_NUMBER, CONF_HOUSE_NUMBER_SUFFIX: HOUSE_NUMBER_SUFFIX, @@ -106,9 +105,15 @@ async def test_abort_if_already_setup(hass: HomeAssistant) -> None: ).add_to_hass(hass) result = await hass.config_entries.flow.async_init( - DOMAIN, - context={"source": SOURCE_USER}, - data={ + DOMAIN, context={"source": SOURCE_USER} + ) + + assert result.get("type") is FlowResultType.FORM + assert result.get("step_id") == "user" + + result = await hass.config_entries.flow.async_configure( + result["flow_id"], + user_input={ CONF_ZIP_CODE: ZIP_CODE, CONF_HOUSE_NUMBER: HOUSE_NUMBER, CONF_HOUSE_NUMBER_SUFFIX: HOUSE_NUMBER_SUFFIX,