diff --git a/tests/components/egauge/test_config_flow.py b/tests/components/egauge/test_config_flow.py index 7c2b094afeab..c9e6b21e7a3f 100644 --- a/tests/components/egauge/test_config_flow.py +++ b/tests/components/egauge/test_config_flow.py @@ -73,9 +73,15 @@ async def test_user_flow_errors( mock_egauge_client.get_device_serial_number.side_effect = side_effect result = await hass.config_entries.flow.async_init( - DOMAIN, - context={"source": SOURCE_USER}, - data={ + 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={ CONF_HOST: "192.168.1.100", CONF_USERNAME: "admin", CONF_PASSWORD: "wrong", @@ -119,9 +125,15 @@ async def test_user_flow_already_configured( mock_config_entry.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["type"] is FlowResultType.FORM + assert result["step_id"] == "user" + + result = await hass.config_entries.flow.async_configure( + result["flow_id"], + user_input={ CONF_HOST: "http://192.168.1.200", CONF_USERNAME: "admin", CONF_PASSWORD: "secret",