Fix LifeSpan enum (#705)

DeebotUniverse · Dec 20, 2024 · 95a1590 · 95a1590
1 parent 0238549
commit 95a1590
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/deebot_client/events/__init__.py b/deebot_client/events/__init__.py
@@ -128,7 +128,7 @@ class LifeSpan(StrEnum):
     xml_value: str
 
     def __new__(cls, value: str, xml_value: str = "") -> Self:
-        obj = str.__new__(cls)
+        obj = str.__new__(cls, value)
         obj._value_ = value
         obj.xml_value = xml_value
         return obj

diff --git a/tests/events/test_events.py b/tests/events/test_events.py
@@ -0,0 +1,11 @@
+"""Test events."""
+
+from __future__ import annotations
+
+from deebot_client.events import LifeSpan
+
+
+def test_life_span() -> None:
+    """Test life span events."""
+    assert LifeSpan.BRUSH != LifeSpan.FILTER
+    assert LifeSpan.FILTER not in {LifeSpan.BLADE, LifeSpan.BRUSH, LifeSpan.SIDE_BRUSH}