mirror of
https://github.com/python-kasa/python-kasa.git
synced 2025-10-10 09:28:02 +00:00
Add reboot and factory_reset to tapodevice (#686)
* Add reboot and factory_reset to tapodevice * Add test for reboot command * Fix mocking as different protocols use different methods for comms..
This commit is contained in:
@@ -21,6 +21,7 @@ from kasa.cli import (
|
||||
cli,
|
||||
emeter,
|
||||
raw_command,
|
||||
reboot,
|
||||
state,
|
||||
sysinfo,
|
||||
toggle,
|
||||
@@ -103,6 +104,21 @@ async def test_raw_command(dev):
|
||||
assert "Usage" in res.output
|
||||
|
||||
|
||||
@device_smart
|
||||
async def test_reboot(dev, mocker):
|
||||
"""Test that reboot works on SMART devices."""
|
||||
runner = CliRunner()
|
||||
query_mock = mocker.patch.object(dev.protocol, "query")
|
||||
|
||||
res = await runner.invoke(
|
||||
reboot,
|
||||
obj=dev,
|
||||
)
|
||||
|
||||
query_mock.assert_called()
|
||||
assert res.exit_code == 0
|
||||
|
||||
|
||||
@device_smart
|
||||
async def test_wifi_scan(dev):
|
||||
runner = CliRunner()
|
||||
|
Reference in New Issue
Block a user