With new menu. Testus KO

tests/commons.py

@@ -23,9 +23,7 @@ from homeassistant.components.switch import (
     SwitchEntity,
 )
 
-from homeassistant.components.number import (
-    NumberEntity,
-)
+from homeassistant.components.number import NumberEntity, DOMAIN as NUMBER_DOMAIN
 
 from pytest_homeassistant_custom_component.common import MockConfigEntry
 
@@ -854,3 +852,25 @@ def cancel_switchs_cycles(entity: BaseThermostat):
         return
     for under in entity._underlyings:
         under._cancel_cycle()
+
+
+async def set_climate_preset_temp(
+    entity: BaseThermostat, temp_number_name: str, temp: float
+):
+    """Set a preset value in the temp Number entity"""
+    number_entity_id = (
+        NUMBER_DOMAIN
+        + "."
+        + entity.entity_id.split(".")[1]
+        + "_"
+        + temp_number_name
+        + PRESET_TEMP_SUFFIX
+    )
+
+    temp_entity = search_entity(
+        entity.hass,
+        number_entity_id,
+        NUMBER_DOMAIN,
+    )
+    if temp_entity:
+        await temp_entity.async_set_native_value(temp)

tests/const.py

@@ -138,6 +138,7 @@ MOCK_TH_OVER_CLIMATE_TYPE_AC_CONFIG = {
     CONF_AUTO_REGULATION_PERIOD_MIN: 1,
 }
 
+# TODO remove this later
 MOCK_PRESETS_CONFIG = {
     PRESET_FROST_PROTECTION + PRESET_TEMP_SUFFIX: 7,
     PRESET_ECO + PRESET_TEMP_SUFFIX: 16,
@@ -145,6 +146,7 @@ MOCK_PRESETS_CONFIG = {
     PRESET_BOOST + PRESET_TEMP_SUFFIX: 18,
 }
 
+# TODO remove this later
 MOCK_PRESETS_AC_CONFIG = {
     PRESET_FROST_PROTECTION + PRESET_TEMP_SUFFIX: 7,
     PRESET_ECO + PRESET_TEMP_SUFFIX: 17,
@@ -188,20 +190,10 @@ MOCK_POWER_CONFIG = {
 
 MOCK_PRESENCE_CONFIG = {
     CONF_PRESENCE_SENSOR: "person.presence_sensor",
-    PRESET_ECO + PRESET_AWAY_SUFFIX + PRESET_TEMP_SUFFIX: 16,
-    PRESET_COMFORT + PRESET_AWAY_SUFFIX + PRESET_TEMP_SUFFIX: 17,
-    PRESET_BOOST + PRESET_AWAY_SUFFIX + PRESET_TEMP_SUFFIX: 18,
 }
 
 MOCK_PRESENCE_AC_CONFIG = {
     CONF_PRESENCE_SENSOR: "person.presence_sensor",
-    PRESET_FROST_PROTECTION + PRESET_AWAY_SUFFIX + PRESET_TEMP_SUFFIX: 7,
-    PRESET_ECO + PRESET_AWAY_SUFFIX + PRESET_TEMP_SUFFIX: 16,
-    PRESET_COMFORT + PRESET_AWAY_SUFFIX + PRESET_TEMP_SUFFIX: 17,
-    PRESET_BOOST + PRESET_AWAY_SUFFIX + PRESET_TEMP_SUFFIX: 18,
-    PRESET_ECO + PRESET_AC_SUFFIX + PRESET_AWAY_SUFFIX + PRESET_TEMP_SUFFIX: 27,
-    PRESET_COMFORT + PRESET_AC_SUFFIX + PRESET_AWAY_SUFFIX + PRESET_TEMP_SUFFIX: 26,
-    PRESET_BOOST + PRESET_AC_SUFFIX + PRESET_AWAY_SUFFIX + PRESET_TEMP_SUFFIX: 25,
 }
 
 MOCK_ADVANCED_CONFIG = {

tests/test_config_flow.py

@@ -205,14 +205,6 @@ async def test_user_config_flow_over_climate(
         result["flow_id"], user_input={CONF_USE_PRESETS_CENTRAL_CONFIG: False}
     )
 
-    assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
-    assert result["step_id"] == "presets"
-    assert result["errors"] == {}
-
-    result = await hass.config_entries.flow.async_configure(
-        result["flow_id"], user_input=MOCK_PRESETS_CONFIG
-    )
-
     assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
     assert result["step_id"] == "advanced"
     assert result["errors"] == {}
@@ -232,7 +224,7 @@ async def test_user_config_flow_over_climate(
     assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
     assert result[
         "data"
-    ] == MOCK_TH_OVER_CLIMATE_USER_CONFIG | MOCK_TH_OVER_CLIMATE_MAIN_CONFIG | MOCK_TH_OVER_CLIMATE_CENTRAL_MAIN_CONFIG | MOCK_TH_OVER_CLIMATE_TYPE_CONFIG | MOCK_PRESETS_CONFIG | MOCK_ADVANCED_CONFIG | MOCK_DEFAULT_FEATURE_CONFIG | {
+    ] == MOCK_TH_OVER_CLIMATE_USER_CONFIG | MOCK_TH_OVER_CLIMATE_MAIN_CONFIG | MOCK_TH_OVER_CLIMATE_CENTRAL_MAIN_CONFIG | MOCK_TH_OVER_CLIMATE_TYPE_CONFIG | MOCK_ADVANCED_CONFIG | MOCK_DEFAULT_FEATURE_CONFIG | {
         CONF_USE_MAIN_CENTRAL_CONFIG: False,
         CONF_USE_TPI_CENTRAL_CONFIG: False,
         CONF_USE_PRESETS_CENTRAL_CONFIG: False,
@@ -502,11 +494,11 @@ async def test_user_config_flow_over_4_switches(
 
     await create_central_config(hass)
 
-    SOURCE_CONFIG = {
+    SOURCE_CONFIG = {  # pylint: disable=invalid-name
         CONF_THERMOSTAT_TYPE: CONF_THERMOSTAT_SWITCH,
     }
 
-    MAIN_CONFIG = {  # pylint: disable=wildcard-import, invalid-name
+    MAIN_CONFIG = {  # pylint: disable=invalid-name
         CONF_NAME: "TheOver4SwitchMockName",
         CONF_TEMP_SENSOR: "sensor.mock_temp_sensor",
         CONF_CYCLE_MIN: 5,
@@ -520,7 +512,7 @@ async def test_user_config_flow_over_4_switches(
         CONF_USED_BY_CENTRAL_BOILER: False,
     }
 
-    TYPE_CONFIG = {  # pylint: disable=wildcard-import, invalid-name
+    TYPE_CONFIG = {  # pylint: disable=invalid-name
         CONF_HEATER: "switch.mock_switch1",
         CONF_HEATER_2: "switch.mock_switch2",
         CONF_HEATER_3: "switch.mock_switch3",

tests/test_switch_ac.py

@@ -56,6 +56,23 @@ async def test_over_switch_ac_full_start(
         assert entity
         assert isinstance(entity, ThermostatOverSwitch)
 
+        # Initialise the preset temp
+        await set_climate_preset_temp(
+            entity, PRESET_FROST_PROTECTION + PRESET_AWAY_SUFFIX, 7
+        )
+        await set_climate_preset_temp(entity, PRESET_ECO + PRESET_AWAY_SUFFIX, 16)
+        await set_climate_preset_temp(entity, PRESET_COMFORT + PRESET_AWAY_SUFFIX, 17)
+        await set_climate_preset_temp(entity, PRESET_BOOST + PRESET_AWAY_SUFFIX, 18)
+        await set_climate_preset_temp(
+            entity, PRESET_ECO + PRESET_AC_SUFFIX + PRESET_AWAY_SUFFIX, 27
+        )
+        await set_climate_preset_temp(
+            entity, PRESET_COMFORT + PRESET_AC_SUFFIX + PRESET_AWAY_SUFFIX, 26
+        )
+        await set_climate_preset_temp(
+            entity, PRESET_BOOST + PRESET_AC_SUFFIX + PRESET_AWAY_SUFFIX, 25
+        )
+
         assert entity.name == "TheOverSwitchMockName"
         assert entity.is_over_climate is False  # pylint: disable=protected-access
         assert entity.ac_mode is True