feat(MockRobot): MockLocateCapability (#752)

Hypfer · Mar 15, 2021 · 5c41d49 · 5c41d49
1 parent ea7cce4
commit 5c41d49
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/lib/robots/mock/MockRobot.js b/lib/robots/mock/MockRobot.js
@@ -12,6 +12,7 @@ class MockRobot extends ValetudoRobot {
 
         this.registerCapability(new capabilities.MockBasicControlCapability({robot: this}));
         this.registerCapability(new capabilities.MockFanSpeedControlCapability({robot: this}));
+        this.registerCapability(new capabilities.MockLocateCapability({robot: this}));
     }
 
     getManufacturer() {
@@ -23,4 +24,4 @@ class MockRobot extends ValetudoRobot {
     }
 }
 
-module.exports = MockRobot;
+module.exports = MockRobot;
diff --git a/lib/robots/mock/capabilities/MockLocateCapability.js b/lib/robots/mock/capabilities/MockLocateCapability.js
@@ -0,0 +1,13 @@
+const LocateCapability = require("../../../core/capabilities/LocateCapability");
+
+class MockLocateCapability extends LocateCapability {
+    /**
+     * @returns {Promise<void>}
+     */
+    async locate() {
+        // Ring the terminal bell, just for fun
+        process.stdout.write("\x07");
+    }
+}
+
+module.exports = MockLocateCapability;
diff --git a/lib/robots/mock/capabilities/index.js b/lib/robots/mock/capabilities/index.js
@@ -1,4 +1,5 @@
 module.exports = {
     MockBasicControlCapability: require("./MockBasicControlCapability"),
-    MockFanSpeedControlCapability: require("./MockFanSpeedControlCapability")
-};
+    MockFanSpeedControlCapability: require("./MockFanSpeedControlCapability"),
+    MockLocateCapability: require("./MockLocateCapability")
+};