And fix the tests

dan-r · Apr 13, 2024 · fa9e6cc · fa9e6cc
1 parent 33b0eae
commit fa9e6cc
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/tests/test_config_flow.py b/tests/test_config_flow.py
@@ -31,18 +31,13 @@ async def test_options_flow(hass):
     entry.add_to_hass(hass)
 
     flow = config_flow.OhmeOptionsFlow(entry)
-    result = await config_entry_flow.async_init(
-        hass, flow, context={"source": "test"}, data={}
-    )
+    result = await flow.async_step_init()
 
     assert result["type"] == "form"
     assert result["step_id"] == "init"
     assert result["data_schema"] is not None
 
-    result = await config_entry_flow.async_configure(
-        hass,
-        flow,
-        result["step_id"],
+    result = await flow.async_step_init(
         {
             "email": "[email protected]",
             "password": "password123",
@@ -55,4 +50,4 @@ async def test_options_flow(hass):
     assert result["data"] == {
         "never_session_specific": True,
         "enable_accumulative_energy": False,
-    }
+    }