fix(vendor.viomi): Fix manual control state not being reported correctly

Hypfer · Jan 7, 2023 · 3fe2998 · 3fe2998
1 parent d2637fd
commit 3fe2998
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/backend/lib/robots/viomi/capabilities/ViomiManualControlCapability.js b/backend/lib/robots/viomi/capabilities/ViomiManualControlCapability.js
@@ -48,6 +48,8 @@ class ViomiManualControlCapability extends ManualControlCapability {
     async enableManualControl() {
         if (!this.isInManualControlMode()) {
             await this.viomiMove(ViomiManualControlDirection.ENTER_EXIT);
+
+            await this.robot.pollState();
         }
     }
 
@@ -57,6 +59,8 @@ class ViomiManualControlCapability extends ManualControlCapability {
     async disableManualControl() {
         if (this.isInManualControlMode()) {
             await this.viomiMove(ViomiManualControlDirection.ENTER_EXIT);
+
+            await this.robot.pollState();
         }
     }