2020-05-27 14:55:18 +00:00
|
|
|
import pytest
|
|
|
|
|
|
|
|
from kasa import DeviceType, SmartDeviceException
|
|
|
|
|
|
|
|
from .conftest import (
|
|
|
|
bulb,
|
|
|
|
color_bulb,
|
|
|
|
dimmable,
|
|
|
|
handle_turn_on,
|
|
|
|
non_color_bulb,
|
|
|
|
non_dimmable,
|
|
|
|
non_variable_temp,
|
|
|
|
turn_on,
|
|
|
|
variable_temp,
|
|
|
|
)
|
|
|
|
from .newfakes import BULB_SCHEMA, LIGHT_STATE_SCHEMA
|
|
|
|
|
|
|
|
|
|
|
|
@bulb
|
|
|
|
async def test_bulb_sysinfo(dev):
|
|
|
|
assert dev.sys_info is not None
|
|
|
|
BULB_SCHEMA(dev.sys_info)
|
|
|
|
|
|
|
|
assert dev.model is not None
|
|
|
|
|
|
|
|
assert dev.device_type == DeviceType.Bulb
|
|
|
|
assert dev.is_bulb
|
|
|
|
|
|
|
|
|
|
|
|
@bulb
|
|
|
|
async def test_state_attributes(dev):
|
|
|
|
assert "Brightness" in dev.state_information
|
|
|
|
assert dev.state_information["Brightness"] == dev.brightness
|
|
|
|
|
|
|
|
assert "Is dimmable" in dev.state_information
|
|
|
|
assert dev.state_information["Is dimmable"] == dev.is_dimmable
|
|
|
|
|
|
|
|
|
|
|
|
@bulb
|
|
|
|
async def test_light_state_without_update(dev, monkeypatch):
|
|
|
|
with pytest.raises(SmartDeviceException):
|
|
|
|
monkeypatch.setitem(
|
|
|
|
dev._last_update["system"]["get_sysinfo"], "light_state", None
|
|
|
|
)
|
|
|
|
print(dev.light_state)
|
|
|
|
|
|
|
|
|
|
|
|
@bulb
|
|
|
|
async def test_get_light_state(dev):
|
|
|
|
LIGHT_STATE_SCHEMA(await dev.get_light_state())
|
|
|
|
|
|
|
|
|
|
|
|
@color_bulb
|
|
|
|
@turn_on
|
|
|
|
async def test_hsv(dev, turn_on):
|
|
|
|
await handle_turn_on(dev, turn_on)
|
|
|
|
assert dev.is_color
|
|
|
|
|
|
|
|
hue, saturation, brightness = dev.hsv
|
|
|
|
assert 0 <= hue <= 255
|
|
|
|
assert 0 <= saturation <= 100
|
|
|
|
assert 0 <= brightness <= 100
|
|
|
|
|
|
|
|
await dev.set_hsv(hue=1, saturation=1, value=1)
|
|
|
|
|
|
|
|
hue, saturation, brightness = dev.hsv
|
|
|
|
assert hue == 1
|
|
|
|
assert saturation == 1
|
|
|
|
assert brightness == 1
|
|
|
|
|
|
|
|
|
2020-06-14 18:21:55 +00:00
|
|
|
@color_bulb
|
|
|
|
async def test_set_hsv_transition(dev, mocker):
|
|
|
|
set_light_state = mocker.patch("kasa.SmartBulb.set_light_state")
|
|
|
|
await dev.set_hsv(10, 10, 100, transition=1000)
|
|
|
|
|
|
|
|
set_light_state.assert_called_with(
|
|
|
|
{"hue": 10, "saturation": 10, "brightness": 100, "color_temp": 0},
|
|
|
|
transition=1000,
|
|
|
|
)
|
|
|
|
|
|
|
|
|
2020-05-27 14:55:18 +00:00
|
|
|
@color_bulb
|
|
|
|
@turn_on
|
|
|
|
async def test_invalid_hsv(dev, turn_on):
|
|
|
|
await handle_turn_on(dev, turn_on)
|
|
|
|
assert dev.is_color
|
|
|
|
|
|
|
|
for invalid_hue in [-1, 361, 0.5]:
|
|
|
|
with pytest.raises(ValueError):
|
|
|
|
await dev.set_hsv(invalid_hue, 0, 0)
|
|
|
|
|
|
|
|
for invalid_saturation in [-1, 101, 0.5]:
|
|
|
|
with pytest.raises(ValueError):
|
|
|
|
await dev.set_hsv(0, invalid_saturation, 0)
|
|
|
|
|
|
|
|
for invalid_brightness in [-1, 101, 0.5]:
|
|
|
|
with pytest.raises(ValueError):
|
|
|
|
await dev.set_hsv(0, 0, invalid_brightness)
|
|
|
|
|
|
|
|
|
|
|
|
@color_bulb
|
|
|
|
async def test_color_state_information(dev):
|
|
|
|
assert "HSV" in dev.state_information
|
|
|
|
assert dev.state_information["HSV"] == dev.hsv
|
|
|
|
|
|
|
|
|
|
|
|
@non_color_bulb
|
|
|
|
async def test_hsv_on_non_color(dev):
|
|
|
|
assert not dev.is_color
|
|
|
|
|
|
|
|
with pytest.raises(SmartDeviceException):
|
|
|
|
await dev.set_hsv(0, 0, 0)
|
|
|
|
with pytest.raises(SmartDeviceException):
|
|
|
|
print(dev.hsv)
|
|
|
|
|
|
|
|
|
|
|
|
@variable_temp
|
|
|
|
async def test_variable_temp_state_information(dev):
|
|
|
|
assert "Color temperature" in dev.state_information
|
|
|
|
assert dev.state_information["Color temperature"] == dev.color_temp
|
|
|
|
|
|
|
|
assert "Valid temperature range" in dev.state_information
|
|
|
|
assert (
|
|
|
|
dev.state_information["Valid temperature range"] == dev.valid_temperature_range
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
@variable_temp
|
|
|
|
@turn_on
|
|
|
|
async def test_try_set_colortemp(dev, turn_on):
|
|
|
|
await handle_turn_on(dev, turn_on)
|
|
|
|
await dev.set_color_temp(2700)
|
|
|
|
assert dev.color_temp == 2700
|
|
|
|
|
|
|
|
|
2020-06-14 18:21:55 +00:00
|
|
|
@variable_temp
|
|
|
|
async def test_set_color_temp_transition(dev, mocker):
|
|
|
|
set_light_state = mocker.patch("kasa.SmartBulb.set_light_state")
|
|
|
|
await dev.set_color_temp(2700, transition=100)
|
|
|
|
|
|
|
|
set_light_state.assert_called_with({"color_temp": 2700}, transition=100)
|
|
|
|
|
|
|
|
|
2020-05-27 14:55:18 +00:00
|
|
|
@variable_temp
|
|
|
|
async def test_unknown_temp_range(dev, monkeypatch):
|
|
|
|
with pytest.raises(SmartDeviceException):
|
|
|
|
monkeypatch.setitem(dev._sys_info, "model", "unknown bulb")
|
|
|
|
dev.valid_temperature_range()
|
|
|
|
|
|
|
|
|
|
|
|
@variable_temp
|
|
|
|
async def test_out_of_range_temperature(dev):
|
|
|
|
with pytest.raises(ValueError):
|
|
|
|
await dev.set_color_temp(1000)
|
|
|
|
with pytest.raises(ValueError):
|
|
|
|
await dev.set_color_temp(10000)
|
|
|
|
|
|
|
|
|
|
|
|
@non_variable_temp
|
|
|
|
async def test_non_variable_temp(dev):
|
|
|
|
with pytest.raises(SmartDeviceException):
|
|
|
|
await dev.set_color_temp(2700)
|
|
|
|
|
|
|
|
with pytest.raises(SmartDeviceException):
|
|
|
|
dev.valid_temperature_range()
|
|
|
|
|
|
|
|
with pytest.raises(SmartDeviceException):
|
|
|
|
print(dev.color_temp)
|
|
|
|
|
|
|
|
|
|
|
|
@dimmable
|
|
|
|
@turn_on
|
|
|
|
async def test_dimmable_brightness(dev, turn_on):
|
|
|
|
await handle_turn_on(dev, turn_on)
|
|
|
|
assert dev.is_dimmable
|
|
|
|
|
|
|
|
await dev.set_brightness(50)
|
|
|
|
assert dev.brightness == 50
|
|
|
|
|
|
|
|
await dev.set_brightness(10)
|
|
|
|
assert dev.brightness == 10
|
|
|
|
|
|
|
|
with pytest.raises(ValueError):
|
|
|
|
await dev.set_brightness("foo")
|
|
|
|
|
|
|
|
|
2020-06-14 18:21:55 +00:00
|
|
|
@bulb
|
|
|
|
async def test_turn_on_transition(dev, mocker):
|
|
|
|
set_light_state = mocker.patch("kasa.SmartBulb.set_light_state")
|
|
|
|
await dev.turn_on(transition=1000)
|
|
|
|
|
|
|
|
set_light_state.assert_called_with({"on_off": 1}, transition=1000)
|
|
|
|
|
|
|
|
await dev.turn_off(transition=100)
|
|
|
|
|
|
|
|
set_light_state.assert_called_with({"on_off": 0}, transition=100)
|
|
|
|
|
|
|
|
|
|
|
|
@bulb
|
|
|
|
async def test_dimmable_brightness_transition(dev, mocker):
|
|
|
|
set_light_state = mocker.patch("kasa.SmartBulb.set_light_state")
|
|
|
|
await dev.set_brightness(10, transition=1000)
|
|
|
|
|
|
|
|
set_light_state.assert_called_with({"brightness": 10}, transition=1000)
|
|
|
|
|
|
|
|
|
2020-05-27 14:55:18 +00:00
|
|
|
@dimmable
|
|
|
|
async def test_invalid_brightness(dev):
|
|
|
|
assert dev.is_dimmable
|
|
|
|
|
|
|
|
with pytest.raises(ValueError):
|
|
|
|
await dev.set_brightness(110)
|
|
|
|
|
|
|
|
with pytest.raises(ValueError):
|
|
|
|
await dev.set_brightness(-100)
|
|
|
|
|
|
|
|
|
|
|
|
@non_dimmable
|
|
|
|
async def test_non_dimmable(dev):
|
|
|
|
assert not dev.is_dimmable
|
|
|
|
|
|
|
|
with pytest.raises(SmartDeviceException):
|
|
|
|
assert dev.brightness == 0
|
|
|
|
with pytest.raises(SmartDeviceException):
|
|
|
|
await dev.set_brightness(100)
|