feat: Allow RoborockModeEnum parsing by either enum name, value name, or code.

This will be used for command line parsing for Q10 commands. Adds the first tests in the tests/data directory since we moved many containers here.

Author: allenporter

roborock/data/code_mappings.py

@@ -70,6 +70,22 @@ def from_code(cls, code: int):
                 return member
         raise ValueError(f"{code} is not a valid code for {cls.__name__}")
 
+    @classmethod
+    def from_value(cls, value: str):
+        """Find enum member by string value (case-insensitive)."""
+        for member in cls:
+            if member.value.lower() == value.lower():
+                return member
+        raise ValueError(f"{value} is not a valid value for {cls.__name__}")
+
+    @classmethod
+    def from_name(cls, name: str):
+        """Find enum member by name (case-insensitive)."""
+        for member in cls:
+            if member.name.lower() == name.lower():
+                return member
+        raise ValueError(f"{name} is not a valid name for {cls.__name__}")
+
     @classmethod
     def keys(cls) -> list[str]:
         """Returns a list of all member values."""

tests/data/test_code_mappings.py

@@ -0,0 +1,26 @@
+"""Tests for code mappings.
+These tests exercise the custom enum methods using arbitrary enum values.
+"""
+
+from roborock.data.b01_q10.b01_q10_code_mappings import B01_Q10_DP
+
+
+def test_from_code():
+    """Test from_code method."""
+    assert B01_Q10_DP.START_CLEAN == B01_Q10_DP.from_code(201)
+    assert B01_Q10_DP.PAUSE == B01_Q10_DP.from_code(204)
+    assert B01_Q10_DP.STOP == B01_Q10_DP.from_code(206)
+
+
+def test_from_name():
+    """Test from_name method."""
+    assert B01_Q10_DP.START_CLEAN == B01_Q10_DP.from_name("START_CLEAN")
+    assert B01_Q10_DP.PAUSE == B01_Q10_DP.from_name("pause")
+    assert B01_Q10_DP.STOP == B01_Q10_DP.from_name("Stop")
+
+
+def test_from_value():
+    """Test from_value method."""
+    assert B01_Q10_DP.START_CLEAN == B01_Q10_DP.from_value("dpStartClean")
+    assert B01_Q10_DP.PAUSE == B01_Q10_DP.from_value("dpPause")
+    assert B01_Q10_DP.STOP == B01_Q10_DP.from_value("dpStop")