feat(MockRobot): add MapResetCapability

Hypfer · Mar 31, 2021 · 531498b · 531498b
1 parent 2957e25
commit 531498b
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/lib/robots/mock/MockRobot.js b/lib/robots/mock/MockRobot.js
@@ -22,6 +22,7 @@ class MockRobot extends ValetudoRobot {
         this.registerCapability(new capabilities.MockLocateCapability({robot: this}));
         this.registerCapability(new capabilities.MockWifiConfigurationCapability({robot: this}));
         this.registerCapability(new capabilities.MockGoToLocationCapability({robot: this}));
+        this.registerCapability(new capabilities.MockMapResetCapability({robot: this}));
     }
 
     getManufacturer() {
@@ -47,7 +48,7 @@ class MockRobot extends ValetudoRobot {
     }
 
     /**
-     * @private
+     * @public
      */
     buildMap() {
         this.mockMap = {

diff --git a/lib/robots/mock/capabilities/MockMapResetCapability.js b/lib/robots/mock/capabilities/MockMapResetCapability.js
@@ -0,0 +1,15 @@
+const MapResetCapability = require("../../../core/capabilities/MapResetCapability");
+
+/**
+ * @extends MapResetCapability<import("../MockRobot")>
+ */
+class MockMapResetCapability extends MapResetCapability {
+    /**
+     * @returns {Promise<void>}
+     */
+    async reset() {
+        await this.robot.buildMap();
+    }
+}
+
+module.exports = MockMapResetCapability;
diff --git a/lib/robots/mock/capabilities/index.js b/lib/robots/mock/capabilities/index.js
@@ -6,6 +6,7 @@ module.exports = {
     MockFanSpeedControlCapability: require("./MockFanSpeedControlCapability"),
     MockGoToLocationCapability: require("./MockGoToLocationCapability"),
     MockLocateCapability: require("./MockLocateCapability"),
+    MockMapResetCapability: require("./MockMapResetCapability"),
     MockSpeakerTestCapability: require("./MockSpeakerTestCapability"),
     MockSpeakerVolumeControlCapability: require("./MockSpeakerVolumeControlCapability"),
     MockWifiConfigurationCapability: require("./MockWifiConfigurationCapability"),