fix: identification of Genius X and 10000 (#39)

Bluetooth-Devices · Feb 6, 2023 · 93ea567 · 93ea567
1 parent b7592ef
commit 93ea567
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/src/oralb_ble/parser.py b/src/oralb_ble/parser.py
@@ -62,6 +62,7 @@ class Models(Enum):
     SmartSeries7000 = auto()
     SmartSeries8000 = auto()
     SmartSeries9000 = auto()
+    GeniusX = auto()
 
 
 @dataclass
@@ -140,9 +141,10 @@ class ModelDescription:
         modes=SMART_SERIES_MODES,
     ),
     Models.SmartSeries9000: ModelDescription(
-        device_type="Smart Series 9000",
+        device_type="Smart Series 9000/10000",
         modes=SMART_SERIES_MODES,
     ),
+    Models.GeniusX: ModelDescription(device_type="Genius X", modes=SMART_SERIES_MODES),
 }
 
 STATES = {
@@ -205,6 +207,7 @@ class ModelDescription:
     b"\x061\x16": Models.IOSeries89,
     b"\x02\x02\x06": Models.TriumphV2,
     b"\x01\x02\x05": Models.Pro6000,
+    b"\x04q\x04": Models.GeniusX,
 }
 
 SECTOR_MAP = {

diff --git a/tests/test_parser.py b/tests/test_parser.py
@@ -1515,11 +1515,11 @@ def test_9000_series():
     service_info = ORALB_9000_SERIES
     result = parser.update(service_info)
     assert result == SensorUpdate(
-        title="Smart Series 9000 48BE",
+        title="Smart Series 9000/10000 48BE",
         devices={
             None: SensorDeviceInfo(
-                name="Smart Series 9000 48BE",
-                model="Smart Series 9000",
+                name="Smart Series 9000/10000 48BE",
+                model="Smart Series 9000/10000",
                 manufacturer="Oral-B",
                 sw_version=None,
                 hw_version=None,
@@ -1630,11 +1630,11 @@ def test_9000_black_series():
     service_info = ORALB_9000_BLACK_SERIES
     result = parser.update(service_info)
     assert result == SensorUpdate(
-        title="Smart Series 9000 48BE",
+        title="Smart Series 9000/10000 48BE",
         devices={
             None: SensorDeviceInfo(
-                name="Smart Series 9000 48BE",
-                model="Smart Series 9000",
+                name="Smart Series 9000/10000 48BE",
+                model="Smart Series 9000/10000",
                 manufacturer="Oral-B",
                 sw_version=None,
                 hw_version=None,