diff --git a/tests/components/freedompro/test_config_flow.py b/tests/components/freedompro/test_config_flow.py index 0999f1576619..d482dab78a13 100644 --- a/tests/components/freedompro/test_config_flow.py +++ b/tests/components/freedompro/test_config_flow.py @@ -39,9 +39,15 @@ async def test_invalid_auth(hass: HomeAssistant) -> None: }, ): result = await hass.config_entries.flow.async_init( - DOMAIN, - context={"source": SOURCE_USER}, - data=VALID_CONFIG, + DOMAIN, context={"source": SOURCE_USER} + ) + + assert result["type"] is FlowResultType.FORM + assert result["step_id"] == "user" + + result = await hass.config_entries.flow.async_configure( + result["flow_id"], + user_input=VALID_CONFIG, ) assert result["errors"] == {"base": "invalid_auth"} @@ -57,9 +63,15 @@ async def test_connection_error(hass: HomeAssistant) -> None: }, ): result = await hass.config_entries.flow.async_init( - DOMAIN, - context={"source": SOURCE_USER}, - data=VALID_CONFIG, + DOMAIN, context={"source": SOURCE_USER} + ) + + assert result["type"] is FlowResultType.FORM + assert result["step_id"] == "user" + + result = await hass.config_entries.flow.async_configure( + result["flow_id"], + user_input=VALID_CONFIG, ) assert result["errors"] == {"base": "cannot_connect"} @@ -75,9 +87,15 @@ async def test_create_entry(hass: HomeAssistant) -> None: }, ): result = await hass.config_entries.flow.async_init( - DOMAIN, - context={"source": SOURCE_USER}, - data=VALID_CONFIG, + DOMAIN, context={"source": SOURCE_USER} + ) + + assert result["type"] is FlowResultType.FORM + assert result["step_id"] == "user" + + result = await hass.config_entries.flow.async_configure( + result["flow_id"], + user_input=VALID_CONFIG, ) assert result["type"] is FlowResultType.CREATE_ENTRY