Fix kettle turn on for skykettle integration (dext0r#548)

ALEX36RU · Nov 1, 2024 · 4deefb5 · 4deefb5
1 parent e00ef65
commit 4deefb5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/custom_components/yandex_smart_home/capability_onoff.py b/custom_components/yandex_smart_home/capability_onoff.py
@@ -44,6 +44,7 @@
 from homeassistant.exceptions import ServiceNotFound
 from homeassistant.helpers.service import async_call_from_config
 
+from . import const
 from .capability import STATE_CAPABILITIES_REGISTRY, ActionOnlyCapabilityMixin, StateCapability
 from .const import CONF_FEATURES, CONF_STATE_UNKNOWN, CONF_TURN_OFF, CONF_TURN_ON, MediaPlayerFeature
 from .helpers import ActionNotAllowed, APIError
@@ -449,7 +450,7 @@ class OnOffCapabilityWaterHeater(OnOffCapability):
     """Capability to turn on or off a water heater."""
 
     _water_heater_operations = {
-        STATE_ON: [STATE_ON, "On", "ON", water_heater.STATE_ELECTRIC],
+        STATE_ON: [STATE_ON, "On", "ON", water_heater.STATE_ELECTRIC, const.SKYKETTLE_MODE_BOIL],
         STATE_OFF: [STATE_OFF, "Off", "OFF"],
     }
 

diff --git a/custom_components/yandex_smart_home/const.py b/custom_components/yandex_smart_home/const.py
@@ -144,6 +144,9 @@
 TION_FAN_SPEED_5 = "5"
 TION_FAN_SPEED_6 = "6"
 
+# https://github.com/ClusterM/skykettle-ha/blob/c1b61c4a22693d6e2b7c2f57a989df418011f2c2/custom_components/skykettle/skykettle.py#L53
+SKYKETTLE_MODE_BOIL = "Boil"
+
 # https://github.com/home-assistant/core/pull/67743
 FAN_SPEED_OFF = "off"
 FAN_SPEED_LOW = "low"

diff --git a/tests/test_capability_onoff.py b/tests/test_capability_onoff.py
@@ -703,7 +703,7 @@ async def test_capability_onoff_water_heater(hass):
     assert cap.parameters.dict() == {"split": True}
 
 
-@pytest.mark.parametrize("op_on", ["on", "On", "ON", "electric"])
+@pytest.mark.parametrize("op_on", ["on", "On", "ON", "electric", "Boil"])
 @pytest.mark.parametrize("op_off", ["off", "Off", "OFF"])
 async def test_capability_onoff_water_heater_set_op_mode(hass, op_on, op_off):
     state = State(