From 49bef01ab8a9ef2db49b9a435199aa285a61e007 Mon Sep 17 00:00:00 2001
From: Alex X <alexey.khit@gmail.com>
Date: Mon, 15 Jan 2024 20:22:50 +0300
Subject: [PATCH] Fix test for kettle

---
 tests/test_water_heater.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_water_heater.py b/tests/test_water_heater.py
index 8e7ef44..1c0d910 100644
--- a/tests/test_water_heater.py
+++ b/tests/test_water_heater.py
@@ -102,7 +102,7 @@ def test_polaris():
     }
 
     state = update_ha_state(YandexKettle, device)
-    assert state.state == "black_tea"
+    assert state.state == "on"
     assert state.attributes == {
         "min_temp": 30,
         "max_temp": 100,
@@ -122,7 +122,7 @@ def test_polaris():
         "temperature": 65,
         "target_temp_high": null,
         "target_temp_low": null,
-        "operation_mode": "black_tea",
+        "operation_mode": "on",
         "friendly_name": "Чайник",
         "supported_features": (
             WaterHeaterEntityFeature.TARGET_TEMPERATURE