Add SEARCHMODE support for SIRIUS subunit

Add SiriusSearchMode enum with All Ch, Category, and Preset values
and wire it up as a searchmode attribute on the Sirius class.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: mvdwetering

src/ynca/__init__.py

@@ -30,6 +30,7 @@
     PwrB,
     Repeat,
     Shuffle,
+    SiriusSearchMode,
     Sleep,
     SoundPrg,
     SpeakerA,
@@ -116,6 +117,7 @@
     "Shuffle",
     "Sirius",
     "SiriusIr",
+    "SiriusSearchMode",
     "SiriusXm",
     "Sleep",
     "SoundPrg",

src/ynca/enums.py

@@ -427,6 +427,21 @@ def _missing_(cls, value: object) -> Self:
     """Unknown values in the enum are mapped to UNKNOWN"""
 
 
+@unique
+class SiriusSearchMode(StrEnum):
+    ALL_CH = "All Ch"
+    CATEGORY = "Category"
+    PRESET = "Preset"
+
+    @classmethod
+    def _missing_(cls, value: object) -> Self:
+        logger.warning("Unknown value '%s' in %s", value, cls.__name__)
+        return cls(cls.UNKNOWN)
+
+    UNKNOWN = UNKNOWN_STRING
+    """Unknown values in the enum are mapped to UNKNOWN"""
+
+
 @unique
 class Sleep(StrEnum):
     OFF = "Off"

src/ynca/subunits/sirius.py

@@ -1,6 +1,8 @@
 from __future__ import annotations
 
 from ..constants import Subunit
+from ..enums import SiriusSearchMode
+from ..function import EnumFunctionMixin
 from ..subunit import SubunitBase
 from . import (
     ArtistFunctionMixin,
@@ -25,6 +27,8 @@ class Sirius(
 ):
     id = Subunit.SIRIUS
 
+    searchmode = EnumFunctionMixin[SiriusSearchMode](SiriusSearchMode)
+
 
 class SiriusIr(
     ArtistFunctionMixin,

tests/test_enums.py

@@ -27,6 +27,7 @@
     PwrB,
     Repeat,
     Shuffle,
+    SiriusSearchMode,
     Sleep,
     SoundPrg,
     SpeakerA,
@@ -69,6 +70,7 @@ def test_invalid_values_on_enums() -> None:
     assert PwrB("x") is PwrB.UNKNOWN
     assert Repeat("x") is Repeat.UNKNOWN
     assert Shuffle("x") is Shuffle.UNKNOWN
+    assert SiriusSearchMode("x") is SiriusSearchMode.UNKNOWN
     assert Sleep("x") is Sleep.UNKNOWN
     assert SoundPrg("x") is SoundPrg.UNKNOWN
     assert SpeakerA("x") is SpeakerA.UNKNOWN

tests/test_sirius.py

@@ -2,7 +2,7 @@
 from typing import Any
 
 from tests.mock_yncaconnection import YncaConnectionMock
-from ynca import Sirius
+from ynca import Sirius, SiriusSearchMode
 
 SYS = "SYS"
 SUBUNIT = "SIRIUS"
@@ -25,6 +25,12 @@
             (SUBUNIT, "AVAIL", "Ready"),
         ],
     ),
+    (
+        (SUBUNIT, "SEARCHMODE"),
+        [
+            (SUBUNIT, "SEARCHMODE", "Preset"),
+        ],
+    ),
     (
         (SYS, "VERSION"),
         [
@@ -47,3 +53,29 @@ def test_initialize(
     assert sirius.artist == "Artist"
     assert sirius.song == "Song"
     assert sirius.chname == "ChName"
+    assert sirius.searchmode == SiriusSearchMode.PRESET
+
+
+def test_searchmode(
+    connection: YncaConnectionMock, update_callback: Callable[[str, Any], None]
+) -> None:
+    connection.get_response_list = INITIALIZE_FULL_RESPONSES
+    sirius = Sirius(connection)
+    sirius.initialize()
+
+    # Updates from device
+    connection.send_protocol_message(SUBUNIT, "SEARCHMODE", "All Ch")
+    assert sirius.searchmode == SiriusSearchMode.ALL_CH
+
+    connection.send_protocol_message(SUBUNIT, "SEARCHMODE", "Category")
+    assert sirius.searchmode == SiriusSearchMode.CATEGORY
+
+    connection.send_protocol_message(SUBUNIT, "SEARCHMODE", "Preset")
+    assert sirius.searchmode == SiriusSearchMode.PRESET
+
+    # Set
+    sirius.searchmode = SiriusSearchMode.ALL_CH
+    connection.put.assert_called_with(SUBUNIT, "SEARCHMODE", "All Ch")
+
+    sirius.searchmode = SiriusSearchMode.CATEGORY
+    connection.put.assert_called_with(SUBUNIT, "SEARCHMODE", "Category")