mirror of
https://github.com/python-kasa/python-kasa.git
synced 2024-12-23 03:33:35 +00:00
644a10a0d1
* Move tests to device-type specific test files to make improvements more approachable * protocol: remove the port parameter from query, as there are no other known ports, fix docstrings * Revise docstrings, remove superfluous information and remove unused methods ({set,get_icon} and set_time) * cli: indent device output to make it more easily readable when having multiple devices * remove adjust flake8 ignores (we have no setup.py anymore) * pyproject: include cli tool to coverage, add config for interrogate (docstring coverage) * bulb: raise exception on color_temp error cases instead of returning zero values * improve bulb tests, simplify conftest * strip: rename plugs property to children and move it to smartdevice
118 lines
2.8 KiB
Python
118 lines
2.8 KiB
Python
import pytest
|
|
from asyncclick.testing import CliRunner
|
|
|
|
from kasa import SmartDevice
|
|
from kasa.cli import alias, brightness, emeter, raw_command, state, sysinfo
|
|
|
|
from .conftest import handle_turn_on, turn_on
|
|
|
|
pytestmark = pytest.mark.asyncio
|
|
|
|
|
|
async def test_sysinfo(dev):
|
|
runner = CliRunner()
|
|
res = await runner.invoke(sysinfo, obj=dev)
|
|
assert "System info" in res.output
|
|
assert dev.alias in res.output
|
|
|
|
|
|
@turn_on
|
|
async def test_state(dev, turn_on):
|
|
await handle_turn_on(dev, turn_on)
|
|
runner = CliRunner()
|
|
res = await runner.invoke(state, obj=dev)
|
|
print(res.output)
|
|
|
|
if dev.is_on:
|
|
assert "Device state: ON" in res.output
|
|
else:
|
|
assert "Device state: OFF" in res.output
|
|
|
|
|
|
async def test_alias(dev):
|
|
runner = CliRunner()
|
|
|
|
res = await runner.invoke(alias, obj=dev)
|
|
assert f"Alias: {dev.alias}" in res.output
|
|
|
|
new_alias = "new alias"
|
|
res = await runner.invoke(alias, [new_alias], obj=dev)
|
|
assert f"Setting alias to {new_alias}" in res.output
|
|
|
|
res = await runner.invoke(alias, obj=dev)
|
|
assert f"Alias: {new_alias}" in res.output
|
|
|
|
|
|
async def test_raw_command(dev):
|
|
runner = CliRunner()
|
|
res = await runner.invoke(raw_command, ["system", "get_sysinfo"], obj=dev)
|
|
|
|
assert res.exit_code == 0
|
|
assert dev.alias in res.output
|
|
|
|
res = await runner.invoke(raw_command, obj=dev)
|
|
assert res.exit_code != 0
|
|
assert "Usage" in res.output
|
|
|
|
|
|
async def test_emeter(dev: SmartDevice, mocker):
|
|
runner = CliRunner()
|
|
|
|
res = await runner.invoke(emeter, obj=dev)
|
|
if not dev.has_emeter:
|
|
assert "Device has no emeter" in res.output
|
|
return
|
|
|
|
assert "== Emeter ==" in res.output
|
|
|
|
monthly = mocker.patch.object(dev, "get_emeter_monthly")
|
|
res = await runner.invoke(emeter, ["--year", "1900"], obj=dev)
|
|
assert "For year" in res.output
|
|
monthly.assert_called()
|
|
|
|
daily = mocker.patch.object(dev, "get_emeter_daily")
|
|
res = await runner.invoke(emeter, ["--month", "1900-12"], obj=dev)
|
|
assert "For month" in res.output
|
|
daily.assert_called()
|
|
|
|
|
|
async def test_brightness(dev):
|
|
runner = CliRunner()
|
|
res = await runner.invoke(brightness, obj=dev)
|
|
if not dev.is_dimmable:
|
|
assert "This device does not support brightness." in res.output
|
|
return
|
|
|
|
res = await runner.invoke(brightness, obj=dev)
|
|
assert f"Brightness: {dev.brightness}" in res.output
|
|
|
|
res = await runner.invoke(brightness, ["12"], obj=dev)
|
|
assert "Setting brightness" in res.output
|
|
|
|
res = await runner.invoke(brightness, obj=dev)
|
|
assert "Brightness: 12" in res.output
|
|
|
|
|
|
async def test_temperature(dev):
|
|
pass
|
|
|
|
|
|
async def test_hsv(dev):
|
|
pass
|
|
|
|
|
|
async def test_led(dev):
|
|
pass
|
|
|
|
|
|
async def test_on(dev):
|
|
pass
|
|
|
|
|
|
async def test_off(dev):
|
|
pass
|
|
|
|
|
|
async def test_reboot(dev):
|
|
pass
|