diff --git a/tests/components/azure_event_hub/test_config_flow.py b/tests/components/azure_event_hub/test_config_flow.py
index 52685c36bbe6..a8492d566c39 100644
--- a/tests/components/azure_event_hub/test_config_flow.py
+++ b/tests/components/azure_event_hub/test_config_flow.py
@@ -55,7 +55,15 @@ async def test_form(
 ) -> None:
     """Test we get the form."""
     result = await hass.config_entries.flow.async_init(
-        DOMAIN, context={"source": config_entries.SOURCE_USER}, data=None
+        DOMAIN, context={"source": config_entries.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=None,
     )
     assert result["type"] is FlowResultType.FORM
     assert result["errors"] is None
@@ -133,9 +141,15 @@ async def test_connection_error_sas(
 ) -> None:
     """Test we handle connection errors."""
     result = await hass.config_entries.flow.async_init(
-        DOMAIN,
-        context={"source": config_entries.SOURCE_USER},
-        data=BASE_CONFIG_SAS.copy(),
+        DOMAIN, context={"source": config_entries.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=BASE_CONFIG_SAS.copy(),
     )
     assert result["type"] is FlowResultType.FORM
     assert result["errors"] is None
@@ -162,9 +176,15 @@ async def test_connection_error_cs(
 ) -> None:
     """Test we handle connection errors."""
     result = await hass.config_entries.flow.async_init(
-        DOMAIN,
-        context={"source": config_entries.SOURCE_USER},
-        data=BASE_CONFIG_CS.copy(),
+        DOMAIN, context={"source": config_entries.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=BASE_CONFIG_CS.copy(),
     )
     assert result["type"] is FlowResultType.FORM
     assert result["errors"] is None