From bf1c138a3c4ab29a80f5f81c37fb70e7f06873b3 Mon Sep 17 00:00:00 2001
From: Manu <4445816+tr4nt0r@users.noreply.github.com>
Date: Tue, 22 Apr 2025 20:53:09 +0200
Subject: [PATCH] Fix some mistakes in the Habitica tests (#143465)

---
 tests/components/habitica/test_config_flow.py | 40 +++++++++----------
 1 file changed, 20 insertions(+), 20 deletions(-)

diff --git a/tests/components/habitica/test_config_flow.py b/tests/components/habitica/test_config_flow.py
index 07678b031bc..5ec998ec82e 100644
--- a/tests/components/habitica/test_config_flow.py
+++ b/tests/components/habitica/test_config_flow.py
@@ -76,8 +76,9 @@ async def test_form_login(hass: HomeAssistant, mock_setup_entry: AsyncMock) -> N
     assert "login" in result["menu_options"]
     assert result["step_id"] == "user"
 
-    result = await hass.config_entries.flow.async_init(
-        DOMAIN, context={"source": "login"}
+    result = await hass.config_entries.flow.async_configure(
+        result["flow_id"],
+        {"next_step_id": "login"},
     )
     assert result["type"] is FlowResultType.FORM
     assert result["errors"] == {}
@@ -123,8 +124,9 @@ async def test_form_login_errors(
     assert result["type"] is FlowResultType.MENU
     assert result["step_id"] == "user"
 
-    result = await hass.config_entries.flow.async_init(
-        DOMAIN, context={"source": "login"}
+    result = await hass.config_entries.flow.async_configure(
+        result["flow_id"],
+        {"next_step_id": "login"},
     )
 
     habitica.login.side_effect = raise_error
@@ -156,7 +158,7 @@ async def test_form_login_errors(
 
 
 @pytest.mark.usefixtures("habitica")
-async def test_form__already_configured(
+async def test_form_already_configured(
     hass: HomeAssistant,
     config_entry: MockConfigEntry,
 ) -> None:
@@ -171,13 +173,14 @@ async def test_form__already_configured(
     assert result["type"] is FlowResultType.MENU
     assert result["step_id"] == "user"
 
-    result = await hass.config_entries.flow.async_init(
-        DOMAIN, context={"source": "advanced"}
+    result = await hass.config_entries.flow.async_configure(
+        result["flow_id"],
+        {"next_step_id": "login"},
     )
 
     result = await hass.config_entries.flow.async_configure(
         result["flow_id"],
-        user_input=MOCK_DATA_ADVANCED_STEP,
+        user_input=MOCK_DATA_LOGIN_STEP,
     )
 
     assert result["type"] is FlowResultType.ABORT
@@ -196,19 +199,14 @@ async def test_form_advanced(hass: HomeAssistant, mock_setup_entry: AsyncMock) -
     assert "advanced" in result["menu_options"]
     assert result["step_id"] == "user"
 
-    result = await hass.config_entries.flow.async_init(
-        DOMAIN, context={"source": "advanced"}
+    result = await hass.config_entries.flow.async_configure(
+        result["flow_id"],
+        {"next_step_id": "advanced"},
     )
     assert result["type"] is FlowResultType.FORM
     assert result["errors"] == {}
     assert result["step_id"] == "advanced"
 
-    result = await hass.config_entries.flow.async_init(
-        DOMAIN, context={"source": "advanced"}
-    )
-    assert result["type"] is FlowResultType.FORM
-    assert result["errors"] == {}
-
     result = await hass.config_entries.flow.async_configure(
         result["flow_id"],
         user_input=MOCK_DATA_ADVANCED_STEP,
@@ -249,8 +247,9 @@ async def test_form_advanced_errors(
     assert result["type"] is FlowResultType.MENU
     assert result["step_id"] == "user"
 
-    result = await hass.config_entries.flow.async_init(
-        DOMAIN, context={"source": "advanced"}
+    result = await hass.config_entries.flow.async_configure(
+        result["flow_id"],
+        {"next_step_id": "advanced"},
     )
 
     habitica.get_user.side_effect = raise_error
@@ -298,8 +297,9 @@ async def test_form_advanced_already_configured(
     assert result["type"] is FlowResultType.MENU
     assert result["step_id"] == "user"
 
-    result = await hass.config_entries.flow.async_init(
-        DOMAIN, context={"source": "advanced"}
+    result = await hass.config_entries.flow.async_configure(
+        result["flow_id"],
+        {"next_step_id": "advanced"},
     )
 
     result = await hass.config_entries.flow.async_configure(