fix: forward to post library step

bramstroker · Jan 1, 2025 · c3845e4 · c3845e4
1 parent 8bed3b6
commit c3845e4
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/custom_components/powercalc/config_flow.py b/custom_components/powercalc/config_flow.py
@@ -1014,7 +1014,7 @@ async def _process_user_input(user_input: dict[str, Any]) -> dict[str, Any]:
             PowercalcFormStep(
                 step=Step.LIBRARY_CUSTOM_FIELDS,
                 schema=build_dynamic_field_schema(self.selected_profile),  # type: ignore
-                # next_step=Step.POST_LIBRARY,
+                next_step=Step.POST_LIBRARY,
                 validate_user_input=_process_user_input,
             ),
             user_input,

diff --git a/tests/config_flow/test_virtual_power_library.py b/tests/config_flow/test_virtual_power_library.py
@@ -7,7 +7,12 @@
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.selector import SelectSelector
 
-from custom_components.powercalc import CONF_CREATE_ENERGY_SENSOR, CONF_CREATE_UTILITY_METERS
+from custom_components.powercalc import (
+    CONF_CREATE_ENERGY_SENSOR,
+    CONF_CREATE_UTILITY_METERS,
+    CONF_ENERGY_INTEGRATION_METHOD,
+    DEFAULT_ENERGY_INTEGRATION_METHOD,
+)
 from custom_components.powercalc.config_flow import (
     CONF_CONFIRM_AUTODISCOVERED_MODEL,
     Step,
@@ -229,10 +234,18 @@ async def test_profile_with_custom_fields(
         {"some_entity": "sensor.foobar"},
     )
 
+    assert result["type"] == data_entry_flow.FlowResultType.FORM
+    assert result["step_id"] == Step.POWER_ADVANCED
+    result = await hass.config_entries.flow.async_configure(
+        result["flow_id"],
+        {},
+    )
+
     assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY
     assert result["data"] == {
         CONF_CREATE_ENERGY_SENSOR: True,
         CONF_CREATE_UTILITY_METERS: False,
+        CONF_ENERGY_INTEGRATION_METHOD: DEFAULT_ENERGY_INTEGRATION_METHOD,
         CONF_ENTITY_ID: "sensor.test",
         CONF_NAME: "test",
         CONF_MANUFACTURER: "test",