Add factory-reset command to cli (python-kasa#1108)

rytilahti · web-flow · commit b7fa0d2040f9 · 2024-09-21T16:52:52.000+02:00
Allow reseting devices to factory settings using the cli: `kasa device factory-reset`.
diff --git a/kasa/cli/device.py b/kasa/cli/device.py
@@ -166,6 +166,17 @@ async def reboot(plug, delay):
     return await plug.reboot(delay)
 
 
+@device.command()
+@pass_dev
+async def factory_reset(plug):
+    """Reset device to factory settings."""
+    click.confirm(
+        "Do you really want to reset the device to factory settings?", abort=True
+    )
+
+    return await plug.factory_reset()
+
+
 @device.command()
 @pass_dev
 @click.option(
diff --git a/kasa/tests/test_cli.py b/kasa/tests/test_cli.py
@@ -19,6 +19,7 @@
 )
 from kasa.cli.device import (
     alias,
+    factory_reset,
     led,
     reboot,
     state,
@@ -215,6 +216,21 @@ async def test_reboot(dev, mocker, runner):
     assert res.exit_code == 0
 
 
+@device_smart
+async def test_factory_reset(dev, mocker, runner):
+    """Test that factory reset works on SMART devices."""
+    query_mock = mocker.patch.object(dev.protocol, "query")
+
+    res = await runner.invoke(
+        factory_reset,
+        obj=dev,
+        input="y\n",
+    )
+
+    query_mock.assert_called()
+    assert res.exit_code == 0
+
+
 @device_smart
 async def test_wifi_scan(dev, runner):
     res = await runner.invoke(wifi, ["scan"], obj=dev)
