Skip to content

Commit ad8cb91

Browse files
committed
add tests for previous and changed models
1 parent e154de6 commit ad8cb91

1 file changed

Lines changed: 54 additions & 1 deletion

File tree

tests/hardware/test_init.py

Lines changed: 54 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@
7171
TrueDetectEvent,
7272
VoiceAssistantStateEvent,
7373
VolumeEvent,
74+
water_info,
7475
)
7576
from deebot_client.events.efficiency_mode import EfficiencyModeEvent
7677
from deebot_client.events.fan_speed import FanSpeedEvent
@@ -238,8 +239,60 @@ async def test_get_static_device_info(
238239
WaterAmountEvent: [GetWaterInfo()],
239240
},
240241
),
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+
),
241294
],
242-
ids=["5xu9h3", "itk04l", "yna5xi", "p95mgv"],
295+
ids=["5xu9h3", "itk04l", "yna5xi", "p95mgv", "x5d34r", "8kwdb4"],
243296
)
244297
async def test_capabilities_event_extraction(
245298
class_: str, expected: dict[type[Event], list[Command]]

0 commit comments

Comments
 (0)