mirror of
https://github.com/python-kasa/python-kasa.git
synced 2024-12-23 03:33:35 +00:00
f9a987ca18
* Add doctests to SmartBulb * Add SmartDevice doctests, cleanup README.md * add doctests for smartplug and smartstrip * add discover doctests * Fix bulb mock * add smartdimmer doctests * add sphinx-generated docs, cleanup readme a bit * remove sphinx-click as it does not work with asyncclick * in preparation for rtd hooking, move doc deps to be separate from dev deps * pytestmark needs to be applied separately for each and every file, this fixes the tests * use pathlib for resolving relative paths * Skip discovery doctest on python3.7 The code is just fine, but some reason the mocking behaves differently between 3.7 and 3.8. The latter seems to accept a discrete object for asyncio.run where the former expects a coroutine..
135 lines
3.5 KiB
Python
135 lines
3.5 KiB
Python
import pytest
|
|
|
|
from kasa import SmartDimmer
|
|
|
|
from .conftest import dimmer, handle_turn_on, pytestmark, turn_on
|
|
|
|
|
|
@dimmer
|
|
@turn_on
|
|
async def test_set_brightness(dev, turn_on):
|
|
await handle_turn_on(dev, turn_on)
|
|
|
|
await dev.set_brightness(99)
|
|
assert dev.brightness == 99
|
|
assert dev.is_on == turn_on
|
|
|
|
await dev.set_brightness(0)
|
|
assert dev.brightness == 1
|
|
assert dev.is_on == turn_on
|
|
|
|
|
|
@dimmer
|
|
@turn_on
|
|
async def test_set_brightness_transition(dev, turn_on, mocker):
|
|
await handle_turn_on(dev, turn_on)
|
|
query_helper = mocker.spy(SmartDimmer, "_query_helper")
|
|
|
|
await dev.set_brightness(99, transition=1000)
|
|
|
|
assert dev.brightness == 99
|
|
assert dev.is_on
|
|
query_helper.assert_called_with(
|
|
mocker.ANY,
|
|
"smartlife.iot.dimmer",
|
|
"set_dimmer_transition",
|
|
{"brightness": 99, "duration": 1000},
|
|
)
|
|
|
|
await dev.set_brightness(0, transition=1000)
|
|
assert dev.brightness == 1
|
|
|
|
|
|
@dimmer
|
|
async def test_set_brightness_invalid(dev):
|
|
for invalid_brightness in [-1, 101, 0.5]:
|
|
with pytest.raises(ValueError):
|
|
await dev.set_brightness(invalid_brightness)
|
|
|
|
for invalid_transition in [-1, 0, 0.5]:
|
|
with pytest.raises(ValueError):
|
|
await dev.set_brightness(1, transition=invalid_transition)
|
|
|
|
|
|
@dimmer
|
|
async def test_turn_on_transition(dev, mocker):
|
|
query_helper = mocker.spy(SmartDimmer, "_query_helper")
|
|
original_brightness = dev.brightness
|
|
|
|
await dev.turn_on(transition=1000)
|
|
|
|
assert dev.is_on
|
|
assert dev.brightness == original_brightness
|
|
query_helper.assert_called_with(
|
|
mocker.ANY,
|
|
"smartlife.iot.dimmer",
|
|
"set_dimmer_transition",
|
|
{"brightness": original_brightness, "duration": 1000},
|
|
)
|
|
|
|
|
|
@dimmer
|
|
async def test_turn_off_transition(dev, mocker):
|
|
await handle_turn_on(dev, True)
|
|
query_helper = mocker.spy(SmartDimmer, "_query_helper")
|
|
original_brightness = dev.brightness
|
|
|
|
await dev.turn_off(transition=1000)
|
|
|
|
assert dev.is_off
|
|
assert dev.brightness == original_brightness
|
|
query_helper.assert_called_with(
|
|
mocker.ANY,
|
|
"smartlife.iot.dimmer",
|
|
"set_dimmer_transition",
|
|
{"brightness": 0, "duration": 1000},
|
|
)
|
|
|
|
|
|
@dimmer
|
|
@turn_on
|
|
async def test_set_dimmer_transition(dev, turn_on, mocker):
|
|
await handle_turn_on(dev, turn_on)
|
|
query_helper = mocker.spy(SmartDimmer, "_query_helper")
|
|
|
|
await dev.set_dimmer_transition(99, 1000)
|
|
|
|
assert dev.is_on
|
|
assert dev.brightness == 99
|
|
query_helper.assert_called_with(
|
|
mocker.ANY,
|
|
"smartlife.iot.dimmer",
|
|
"set_dimmer_transition",
|
|
{"brightness": 99, "duration": 1000},
|
|
)
|
|
|
|
|
|
@dimmer
|
|
@turn_on
|
|
async def test_set_dimmer_transition_to_off(dev, turn_on, mocker):
|
|
await handle_turn_on(dev, turn_on)
|
|
original_brightness = dev.brightness
|
|
query_helper = mocker.spy(SmartDimmer, "_query_helper")
|
|
|
|
await dev.set_dimmer_transition(0, 1000)
|
|
|
|
assert dev.is_off
|
|
assert dev.brightness == original_brightness
|
|
query_helper.assert_called_with(
|
|
mocker.ANY,
|
|
"smartlife.iot.dimmer",
|
|
"set_dimmer_transition",
|
|
{"brightness": 0, "duration": 1000},
|
|
)
|
|
|
|
|
|
@dimmer
|
|
async def test_set_dimmer_transition_invalid(dev):
|
|
for invalid_brightness in [-1, 101, 0.5]:
|
|
with pytest.raises(ValueError):
|
|
await dev.set_dimmer_transition(invalid_brightness, 1000)
|
|
|
|
for invalid_transition in [-1, 0, 0.5]:
|
|
with pytest.raises(ValueError):
|
|
await dev.set_dimmer_transition(1, invalid_transition)
|