|
71 | 71 | TrueDetectEvent, |
72 | 72 | VoiceAssistantStateEvent, |
73 | 73 | VolumeEvent, |
| 74 | + water_info, |
74 | 75 | ) |
75 | 76 | from deebot_client.events.efficiency_mode import EfficiencyModeEvent |
76 | 77 | from deebot_client.events.fan_speed import FanSpeedEvent |
@@ -238,8 +239,60 @@ async def test_get_static_device_info( |
238 | 239 | WaterAmountEvent: [GetWaterInfo()], |
239 | 240 | }, |
240 | 241 | ), |
| 242 | + ( |
| 243 | + "x5d34r", |
| 244 | + { |
| 245 | + AdvancedModeEvent: [GetAdvancedMode()], |
| 246 | + AvailabilityEvent: [GetBattery(is_available_check=True)], |
| 247 | + BatteryEvent: [GetBattery()], |
| 248 | + CarpetAutoFanBoostEvent: [GetCarpetAutoFanBoost()], |
| 249 | + CleanCountEvent: [GetCleanCount()], |
| 250 | + CleanLogEvent: [GetCleanLogs()], |
| 251 | + CleanPreferenceEvent: [GetCleanPreference()], |
| 252 | + ContinuousCleaningEvent: [GetContinuousCleaning()], |
| 253 | + CustomCommandEvent: [], |
| 254 | + ErrorEvent: [GetError()], |
| 255 | + FanSpeedEvent: [GetFanSpeed()], |
| 256 | + LifeSpanEvent: [GetLifeSpan([LifeSpan.BRUSH, LifeSpan.FILTER, LifeSpan.SIDE_BRUSH, LifeSpan.UNIT_CARE])], |
| 257 | + NetworkInfoEvent: [GetNetInfo()], |
| 258 | + ReportStatsEvent: [], |
| 259 | + StateEvent: [GetChargeState(), GetCleanInfo()], |
| 260 | + StatsEvent: [GetStats()], |
| 261 | + TotalStatsEvent: [GetTotalStats()], |
| 262 | + TrueDetectEvent: [GetTrueDetect()], |
| 263 | + VolumeEvent: [GetVolume()], |
| 264 | + water_info.MopAttachedEvent: [GetWaterInfo()], |
| 265 | + water_info.WaterAmountEvent: [GetWaterInfo()], |
| 266 | + }, |
| 267 | + ), |
| 268 | + ( |
| 269 | + "8kwdb4", |
| 270 | + { |
| 271 | + AdvancedModeEvent: [GetAdvancedMode()], |
| 272 | + AvailabilityEvent: [GetBattery(is_available_check=True)], |
| 273 | + BatteryEvent: [GetBattery()], |
| 274 | + CarpetAutoFanBoostEvent: [GetCarpetAutoFanBoost()], |
| 275 | + CleanCountEvent: [GetCleanCount()], |
| 276 | + CleanLogEvent: [GetCleanLogs()], |
| 277 | + CleanPreferenceEvent: [GetCleanPreference()], |
| 278 | + ContinuousCleaningEvent: [GetContinuousCleaning()], |
| 279 | + CustomCommandEvent: [], |
| 280 | + ErrorEvent: [GetError()], |
| 281 | + FanSpeedEvent: [GetFanSpeed()], |
| 282 | + LifeSpanEvent: [GetLifeSpan([LifeSpan.BRUSH, LifeSpan.FILTER, LifeSpan.SIDE_BRUSH, LifeSpan.UNIT_CARE])], |
| 283 | + NetworkInfoEvent: [GetNetInfo()], |
| 284 | + ReportStatsEvent: [], |
| 285 | + StateEvent: [GetChargeState(), GetCleanInfo()], |
| 286 | + StatsEvent: [GetStats()], |
| 287 | + TotalStatsEvent: [GetTotalStats()], |
| 288 | + TrueDetectEvent: [GetTrueDetect()], |
| 289 | + VolumeEvent: [GetVolume()], |
| 290 | + water_info.MopAttachedEvent: [GetWaterInfo()], |
| 291 | + water_info.WaterAmountEvent: [GetWaterInfo()], |
| 292 | + }, |
| 293 | + ), |
241 | 294 | ], |
242 | | - ids=["5xu9h3", "itk04l", "yna5xi", "p95mgv"], |
| 295 | + ids=["5xu9h3", "itk04l", "yna5xi", "p95mgv", "x5d34r", "8kwdb4"], |
243 | 296 | ) |
244 | 297 | async def test_capabilities_event_extraction( |
245 | 298 | class_: str, expected: dict[type[Event], list[Command]] |
|
0 commit comments