Skip to content

Commit 95e898d

Browse files
committed
refactor: delete redundant properties from V1VacuumSimulator and update trigger_push_update
1 parent dc26f35 commit 95e898d

1 file changed

Lines changed: 4 additions & 84 deletions

File tree

roborock/testing/v1_simulator.py

Lines changed: 4 additions & 84 deletions
Original file line numberDiff line numberDiff line change
@@ -238,86 +238,6 @@ def v1_channel(self) -> V1Channel:
238238
"""Returns the real V1Channel bound to the fake channels."""
239239
return self._v1_channel
240240

241-
@property
242-
def battery(self) -> int:
243-
"""Get battery level."""
244-
return self.status.battery or 0
245-
246-
@battery.setter
247-
def battery(self, value: int) -> None:
248-
"""Set battery level."""
249-
self.status.battery = value
250-
251-
@property
252-
def state(self) -> RoborockStateCode:
253-
"""Get device state code."""
254-
return self.status.state or RoborockStateCode.charging
255-
256-
@state.setter
257-
def state(self, value: RoborockStateCode | int) -> None:
258-
"""Set device state code."""
259-
self.status.state = RoborockStateCode(value)
260-
261-
@property
262-
def fan_power(self) -> int:
263-
"""Get fan power speed."""
264-
return self.status.fan_power or 0
265-
266-
@fan_power.setter
267-
def fan_power(self, value: int) -> None:
268-
"""Set fan power speed."""
269-
self.status.fan_power = value
270-
271-
@property
272-
def dnd_enabled(self) -> int:
273-
"""Get DND enabled state."""
274-
return self.status.dnd_enabled or 0
275-
276-
@dnd_enabled.setter
277-
def dnd_enabled(self, value: int) -> None:
278-
"""Set DND enabled state."""
279-
self.status.dnd_enabled = value
280-
281-
@property
282-
def mop_mode(self) -> int:
283-
"""Get mop route mode."""
284-
return self.status.mop_mode or 0
285-
286-
@mop_mode.setter
287-
def mop_mode(self, value: int) -> None:
288-
"""Set mop route mode."""
289-
self.status.mop_mode = value
290-
291-
@property
292-
def water_box_mode(self) -> int:
293-
"""Get water box mode."""
294-
return self.status.water_box_mode or 0
295-
296-
@water_box_mode.setter
297-
def water_box_mode(self, value: int) -> None:
298-
"""Set water box mode."""
299-
self.status.water_box_mode = value
300-
301-
@property
302-
def dss(self) -> int:
303-
"""Get dock sensor status."""
304-
return self.status.dss or 0
305-
306-
@dss.setter
307-
def dss(self, value: int) -> None:
308-
"""Set dock sensor status."""
309-
self.status.dss = value
310-
311-
@property
312-
def dock_type(self) -> RoborockDockTypeCode:
313-
"""Get dock type."""
314-
return self.status.dock_type or RoborockDockTypeCode.no_dock
315-
316-
@dock_type.setter
317-
def dock_type(self, value: RoborockDockTypeCode | int) -> None:
318-
"""Set dock type."""
319-
self.status.dock_type = RoborockDockTypeCode(value)
320-
321241
@property
322242
def in_cleaning(self) -> RoborockInCleaning:
323243
"""Return global_clean_not_complete if cleaning, else complete."""
@@ -458,10 +378,10 @@ async def _handle_publish(self, message: RoborockMessage, channel: FakeChannel)
458378
def trigger_push_update(self) -> None:
459379
"""Trigger an unsolicited push state update to all subscribers."""
460380
dps_payload = {
461-
str(int(RoborockDataProtocol.STATE)): self.state,
462-
str(int(RoborockDataProtocol.BATTERY)): self.battery,
463-
str(int(RoborockDataProtocol.FAN_POWER)): self.fan_power,
464-
str(int(RoborockDataProtocol.WATER_BOX_MODE)): self.water_box_mode,
381+
str(int(RoborockDataProtocol.STATE)): self.status.state.value if self.status.state else 0,
382+
str(int(RoborockDataProtocol.BATTERY)): self.status.battery,
383+
str(int(RoborockDataProtocol.FAN_POWER)): self.status.fan_power,
384+
str(int(RoborockDataProtocol.WATER_BOX_MODE)): self.status.water_box_mode,
465385
}
466386

467387
payload = {"dps": dps_payload, "t": int(time.time())}

0 commit comments

Comments
 (0)