Skip to content

Conversation

@r1cht3r
Copy link
Contributor

@r1cht3r r1cht3r commented Dec 5, 2024

This is a relatively new model, this provides more features than I was previously getting with the fallback vacuum, including functioning work mode & clean count. Water flow rate was the only one that didn't work.

Summary by CodeRabbit

  • New Features

    • Introduced a new robotic vacuum device configuration with advanced capabilities, including battery management, cleaning operations, fan speed adjustments, mapping, and sound playback.
  • Bug Fixes

    • Expanded device testing coverage by adding the new device identifier and updating expected commands for existing device classes, enhancing the robustness of the testing framework.

@coderabbitai
Copy link
Contributor

coderabbitai bot commented Dec 5, 2024

Walkthrough

The pull request introduces a new file, zwkcqc.py, within the deebot_client framework, which defines a comprehensive configuration for a robotic vacuum device. This file establishes various capabilities, including battery management, cleaning operations, fan speed adjustments, and mapping functionalities. Additionally, it modifies the test suite to include the new device identifier and updates expected commands for existing device classes, enhancing the testing coverage and ensuring the robustness of the framework.

Changes

File Path Change Summary
deebot_client/hardware/deebot/zwkcqc.py New file defining capabilities for a robotic vacuum device, including battery, cleaning, mapping, and error handling functionalities.
tests/hardware/test_init.py Added new device identifier "zwkcqc" and updated expected commands for device class "p95mgv" in the test suite.

Possibly related PRs

Suggested labels

pr: new-feature

Suggested reviewers

  • edenhaus

🐰 In the land of dust and cheer,
A vacuum's dance is now quite clear.
With capabilities galore,
It cleans and maps, and so much more!
The tests are set, the models load,
A hopping success on this new road! 🌟


Thank you for using CodeRabbit. We offer it for free to the OSS community and would appreciate your support in helping us grow. If you find it useful, would you consider giving us a shout-out on your favorite social media?

❤️ Share
🪧 Tips

Chat

There are 3 ways to chat with CodeRabbit:

  • Review comments: Directly reply to a review comment made by CodeRabbit. Example:
    • I pushed a fix in commit <commit_id>, please review it.
    • Generate unit testing code for this file.
    • Open a follow-up GitHub issue for this discussion.
  • Files and specific lines of code (under the "Files changed" tab): Tag @coderabbitai in a new review comment at the desired location with your query. Examples:
    • @coderabbitai generate unit testing code for this file.
    • @coderabbitai modularize this function.
  • PR comments: Tag @coderabbitai in a new PR comment to ask questions about the PR branch. For the best results, please provide a very specific query, as very limited context is provided in this mode. Examples:
    • @coderabbitai gather interesting stats about this repository and render them as a table. Additionally, render a pie chart showing the language distribution in the codebase.
    • @coderabbitai read src/utils.ts and generate unit testing code.
    • @coderabbitai read the files in the src/scheduler package and generate a class diagram using mermaid and a README in the markdown format.
    • @coderabbitai help me debug CodeRabbit configuration file.

Note: Be mindful of the bot's finite context window. It's strongly recommended to break down tasks such as reading entire modules into smaller chunks. For a focused discussion, use review comments to chat about specific files and their changes, instead of using the PR comments.

CodeRabbit Commands (Invoked using PR comments)

  • @coderabbitai pause to pause the reviews on a PR.
  • @coderabbitai resume to resume the paused reviews.
  • @coderabbitai review to trigger an incremental review. This is useful when automatic reviews are disabled for the repository.
  • @coderabbitai full review to do a full review from scratch and review all the files again.
  • @coderabbitai summary to regenerate the summary of the PR.
  • @coderabbitai resolve resolve all the CodeRabbit review comments.
  • @coderabbitai configuration to show the current CodeRabbit configuration for the repository.
  • @coderabbitai help to get help.

Other keywords and placeholders

  • Add @coderabbitai ignore anywhere in the PR description to prevent this PR from being reviewed.
  • Add @coderabbitai summary to generate the high-level summary at a specific location in the PR description.
  • Add @coderabbitai anywhere in the PR title to generate the title automatically.

CodeRabbit Configuration File (.coderabbit.yaml)

  • You can programmatically configure CodeRabbit by adding a .coderabbit.yaml file to the root of your repository.
  • Please see the configuration documentation for more information.
  • If your editor has YAML language server enabled, you can add the path at the top of this file to enable auto-completion and validation: # yaml-language-server: $schema=https://coderabbit.ai/integrations/schema.v2.json

Documentation and Community

  • Visit our Documentation for detailed information on how to use CodeRabbit.
  • Join our Discord Community to get help, request features, and share feedback.
  • Follow us on X/Twitter for updates and announcements.

@codecov
Copy link

codecov bot commented Dec 5, 2024

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 86.42%. Comparing base (65b935c) to head (85f9426).
Report is 2 commits behind head on dev.

Additional details and impacted files
@@           Coverage Diff           @@
##              dev     #622   +/-   ##
=======================================
  Coverage   86.42%   86.42%           
=======================================
  Files          88       88           
  Lines        3301     3301           
  Branches      298      298           
=======================================
  Hits         2853     2853           
  Misses        394      394           
  Partials       54       54           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

Copy link
Contributor

@coderabbitai coderabbitai bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 0

🧹 Outside diff range and nitpick comments (3)
deebot_client/hardware/deebot/zwkcqc.py (1)

Line range hint 137-146: Remove unsupported 'water' capability

According to the PR description, the water flow rate feature does not work with the Deebot N30 Omni. Including the water capability may lead to functionality issues or user confusion. It would be best to remove this capability to accurately reflect the device's supported features.

Apply this diff to remove the water capability:

-        water=CapabilitySetTypes(
-            event=WaterInfoEvent,
-            get=[GetWaterInfo()],
-            set=SetWaterInfo,
-            types=(
-                WaterAmount.LOW,
-                WaterAmount.MEDIUM,
-                WaterAmount.HIGH,
-                WaterAmount.ULTRAHIGH,
-            ),
-        ),
tests/hardware/test_init.py (2)

Line range hint 70-133: Add test coverage for the new device 'zwkcqc'

The new device 'zwkcqc' is not included in the test_capabilities_event_extraction test cases. To ensure that all capabilities of this device are correctly tested, please add an entry for 'zwkcqc' in the parameterized test. This will validate that the capabilities defined in zwkcqc.py are properly loaded and functioning as expected.

Apply this diff to add the new device to the test cases:

     @pytest.mark.parametrize(
         ("class_", "expected"),
         [
             (
                 "5xu9h3",
                 {
                     AdvancedModeEvent: [GetAdvancedMode()],
                     AvailabilityEvent: [GetBattery(is_available_check=True)],
                     BatteryEvent: [GetBattery()],
                     BorderSwitchEvent: [GetBorderSwitch()],
                     CutDirectionEvent: [GetCutDirection()],
                     ChildLockEvent: [GetChildLock()],
                     CrossMapBorderWarningEvent: [GetCrossMapBorderWarning()],
                     CustomCommandEvent: [],
                     ErrorEvent: [GetError()],
                     LifeSpanEvent: [GetLifeSpan([LifeSpan.BLADE, LifeSpan.LENS_BRUSH])],
                     MoveUpWarningEvent: [GetMoveUpWarning()],
                     NetworkInfoEvent: [GetNetInfo()],
                     ReportStatsEvent: [],
                     SafeProtectEvent: [GetSafeProtect()],
                     StateEvent: [GetChargeState(), GetCleanInfoV2()],
                     StatsEvent: [GetStats()],
                     TotalStatsEvent: [GetTotalStats()],
                     TrueDetectEvent: [GetTrueDetect()],
                     VolumeEvent: [GetVolume()],
                 },
             ),
             (
                 "itk04l",
                 {
                     AdvancedModeEvent: [GetAdvancedMode()],
                     AvailabilityEvent: [GetBattery(is_available_check=True)],
                     BatteryEvent: [GetBattery()],
                     BorderSwitchEvent: [GetBorderSwitch()],
                     CutDirectionEvent: [GetCutDirection()],
                     ChildLockEvent: [GetChildLock()],
                     CrossMapBorderWarningEvent: [GetCrossMapBorderWarning()],
                     CustomCommandEvent: [],
                     ErrorEvent: [GetError()],
                     LifeSpanEvent: [GetLifeSpan([LifeSpan.BLADE, LifeSpan.LENS_BRUSH])],
                     MoveUpWarningEvent: [GetMoveUpWarning()],
                     NetworkInfoEvent: [GetNetInfo()],
                     ReportStatsEvent: [],
                     SafeProtectEvent: [GetSafeProtect()],
                     StateEvent: [GetChargeState(), GetCleanInfoV2()],
                     StatsEvent: [GetStats()],
                     TotalStatsEvent: [GetTotalStats()],
                     TrueDetectEvent: [GetTrueDetect()],
                     VolumeEvent: [GetVolume()],
                 },
             ),
             (
                 "yna5xi",
                 {
                     AdvancedModeEvent: [GetAdvancedMode()],
                     AvailabilityEvent: [GetBattery(is_available_check=True)],
                     BatteryEvent: [GetBattery()],
                     CachedMapInfoEvent: [GetCachedMapInfo()],
                     CarpetAutoFanBoostEvent: [GetCarpetAutoFanBoost()],
                     CleanLogEvent: [GetCleanLogs()],
                     ContinuousCleaningEvent: [GetContinuousCleaning()],
                     CustomCommandEvent: [],
                     ErrorEvent: [GetError()],
                     FanSpeedEvent: [GetFanSpeed()],
                     LifeSpanEvent: [
                         GetLifeSpan([LifeSpan.BRUSH, LifeSpan.FILTER, LifeSpan.SIDE_BRUSH])
                     ],
                     MapChangedEvent: [],
                     MajorMapEvent: [GetMajorMap()],
                     MapTraceEvent: [GetMapTrace()],
                     MultimapStateEvent: [GetMultimapState()],
                     NetworkInfoEvent: [GetNetInfo()],
                     OtaEvent: [GetOta()],
                     PositionsEvent: [GetPos()],
                     ReportStatsEvent: [],
                     RoomsEvent: [GetCachedMapInfo()],
                     StateEvent: [GetChargeState(), GetCleanInfo()],
                     StatsEvent: [GetStats()],
                     TotalStatsEvent: [GetTotalStats()],
                     VolumeEvent: [GetVolume()],
                     WaterInfoEvent: [GetWaterInfo()],
                 },
             ),
             (
                 "p95mgv",
                 {
                     AdvancedModeEvent: [GetAdvancedMode()],
                     AvailabilityEvent: [GetBattery(is_available_check=True)],
                     BatteryEvent: [GetBattery()],
                     CachedMapInfoEvent: [GetCachedMapInfo()],
                     CarpetAutoFanBoostEvent: [GetCarpetAutoFanBoost()],
                     CleanCountEvent: [GetCleanCount()],
                     CleanPreferenceEvent: [GetCleanPreference()],
                     ContinuousCleaningEvent: [GetContinuousCleaning()],
                     CustomCommandEvent: [],
                     EfficiencyModeEvent: [GetEfficiencyMode()],
                     ErrorEvent: [GetError()],
                     FanSpeedEvent: [GetFanSpeed()],
                     LifeSpanEvent: [
                         GetLifeSpan(
                             [
                                 LifeSpan.BRUSH,
                                 LifeSpan.FILTER,
                                 LifeSpan.SIDE_BRUSH,
                                 LifeSpan.UNIT_CARE,
                             ]
                         )
                     ],
                     MajorMapEvent: [GetMajorMap()],
                     MapChangedEvent: [],
                     MapTraceEvent: [GetMapTrace()],
                     MultimapStateEvent: [GetMultimapState()],
                     NetworkInfoEvent: [GetNetInfo()],
                     OtaEvent: [GetOta()],
                     PositionsEvent: [GetPos()],
                     ReportStatsEvent: [],
                     RoomsEvent: [GetCachedMapInfo()],
                     StateEvent: [GetChargeState(), GetCleanInfo()],
                     StatsEvent: [GetStats()],
                     TotalStatsEvent: [GetTotalStats()],
                     TrueDetectEvent: [GetTrueDetect()],
                     VoiceAssistantStateEvent: [GetVoiceAssistantState()],
                     VolumeEvent: [GetVolume()],
                     WaterInfoEvent: [GetWaterInfo()],
                 },
             ),
+            (
+                "zwkcqc",
+                {
+                    AdvancedModeEvent: [GetAdvancedMode()],
+                    AvailabilityEvent: [GetBattery(is_available_check=True)],
+                    BatteryEvent: [GetBattery()],
+                    CachedMapInfoEvent: [GetCachedMapInfo()],
+                    CarpetAutoFanBoostEvent: [GetCarpetAutoFanBoost()],
+                    CleanCountEvent: [GetCleanCount()],
+                    CleanLogEvent: [GetCleanLogs()],
+                    CleanPreferenceEvent: [GetCleanPreference()],
+                    ContinuousCleaningEvent: [GetContinuousCleaning()],
+                    CustomCommandEvent: [],
+                    ErrorEvent: [GetError()],
+                    FanSpeedEvent: [GetFanSpeed()],
+                    LifeSpanEvent: [
+                        GetLifeSpan([LifeSpan.BRUSH, LifeSpan.FILTER, LifeSpan.SIDE_BRUSH])
+                    ],
+                    MapChangedEvent: [],
+                    MajorMapEvent: [GetMajorMap()],
+                    MapTraceEvent: [GetMapTrace()],
+                    MultimapStateEvent: [GetMultimapState()],
+                    NetworkInfoEvent: [GetNetInfo()],
+                    PositionsEvent: [GetPos()],
+                    ReportStatsEvent: [],
+                    RoomsEvent: [GetCachedMapInfo()],
+                    StateEvent: [GetChargeState(), GetCleanInfo()],
+                    StatsEvent: [GetStats()],
+                    TotalStatsEvent: [GetTotalStats()],
+                    TrueDetectEvent: [GetTrueDetect()],
+                    VolumeEvent: [GetVolume()],
+                },
+            ),
         ],
-        ids=["5xu9h3", "itk04l", "yna5xi", "p95mgv"],
+        ids=["5xu9h3", "itk04l", "yna5xi", "p95mgv", "zwkcqc"],
     )
     async def test_capabilities_event_extraction(
         class_: str, expected: dict[type[Event], list[Command]]
     ) -> None:
         info = await get_static_device_info(class_)
         assert info is not None
         capabilities = info.capabilities
         assert capabilities._events.keys() == expected.keys()
         for event, expected_commands in expected.items():
             assert (
                 capabilities.get_refresh_commands(event) == expected_commands
             ), f"Refresh commands doesn't match for {event}"

Line range hint 38-47: Include 'zwkcqc' in 'test_get_static_device_info' test cases

To ensure that get_static_device_info correctly retrieves information for the new device 'zwkcqc', consider adding a test case for it. This will help validate that the device information is properly loaded and can be accessed as expected.

Apply this diff to add the new test case:

     @pytest.mark.parametrize(
         ("class_", "expected"),
         [
             ("not_specified", lambda: None),
             ("yna5xi", lambda: DEVICES["yna5xi"]),
+            ("zwkcqc", lambda: DEVICES["zwkcqc"]),
         ],
     )
     async def test_get_static_device_info(
         class_: str, expected: Callable[[], StaticDeviceInfo]
     ) -> None:
         """Test get_static_device_info."""
         static_device_info = await get_static_device_info(class_)
         assert static_device_info == expected()
📜 Review details

Configuration used: CodeRabbit UI
Review profile: CHILL

📥 Commits

Reviewing files that changed from the base of the PR and between 65b935c and 85f9426.

📒 Files selected for processing (2)
  • deebot_client/hardware/deebot/zwkcqc.py (1 hunks)
  • tests/hardware/test_init.py (1 hunks)
🔇 Additional comments (2)
deebot_client/hardware/deebot/zwkcqc.py (2)

Line range hint 61-83: Capabilities for 'work_mode' and 'clean count' are correctly configured

The work_mode and count capabilities are properly implemented, aligning with the enhanced functionality for the Deebot N30 Omni as described in the PR. This ensures that the device can utilize the work mode settings and track the clean count effectively.


1-1: ⚠️ Potential issue

Remove unintended filename declaration at the top of the file

Line 1 contains p1jij8.py, which appears to be an accidental inclusion from another file. This line is unnecessary and should be removed to avoid confusion.

Apply this diff to remove the unintended line:

- p1jij8.py

Likely invalid or redundant comment.

@edenhaus
Copy link
Member

edenhaus commented Dec 5, 2024

This is a relatively new model, this provides more features than I was previously getting with the fallback vacuum, including functioning work mode & clean count. Water flow rate was the only one that didn't work.

That's the actual reason, why I removed the fallback so models get add with all supported features instead of only some

Copy link
Member

@edenhaus edenhaus left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @r1cht3r 👍

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

pr: new-feature PR, which adds a new feature

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants