mirror of
https://github.com/python-kasa/python-kasa.git
synced 2026-10-06 21:03:50 +00:00
b36c795cbc
Add type annotations (parameter types and `-> None` return types) to all top-level test functions across 22 test files. This enables mypy to check test function bodies, catching type errors that were previously hidden.
257 lines
7.2 KiB
Python
257 lines
7.2 KiB
Python
from __future__ import annotations
|
|
|
|
import re
|
|
from collections.abc import Callable
|
|
from contextlib import nullcontext
|
|
|
|
import pytest
|
|
|
|
from kasa import Device, DeviceType, KasaException, Module
|
|
from tests.conftest import handle_turn_on, turn_on
|
|
from tests.device_fixtures import (
|
|
bulb,
|
|
color_bulb,
|
|
non_color_bulb,
|
|
non_variable_temp,
|
|
variable_temp,
|
|
)
|
|
|
|
|
|
@bulb
|
|
async def test_state_attributes(dev: Device) -> None:
|
|
assert "Cloud connection" in dev.state_information
|
|
assert isinstance(dev.state_information["Cloud connection"], bool)
|
|
|
|
|
|
@color_bulb
|
|
@turn_on
|
|
async def test_hsv(dev: Device, turn_on: bool) -> None:
|
|
light = dev.modules.get(Module.Light)
|
|
assert light
|
|
await handle_turn_on(dev, turn_on)
|
|
assert light.has_feature("hsv")
|
|
|
|
hue, saturation, brightness = light.hsv
|
|
assert 0 <= hue <= 360
|
|
assert 0 <= saturation <= 100
|
|
assert 0 <= brightness <= 100
|
|
|
|
await light.set_hsv(hue=1, saturation=1, value=1)
|
|
|
|
await dev.update()
|
|
hue, saturation, brightness = light.hsv
|
|
assert hue == 1
|
|
assert saturation == 1
|
|
assert brightness == 1
|
|
|
|
|
|
@color_bulb
|
|
@turn_on
|
|
@pytest.mark.parametrize(
|
|
("hue", "sat", "brightness", "exception_cls", "error"),
|
|
[
|
|
pytest.param(-1, 0, 0, ValueError, "Invalid hue", id="hue out of range"),
|
|
pytest.param(361, 0, 0, ValueError, "Invalid hue", id="hue out of range"),
|
|
pytest.param(
|
|
0.5, 0, 0, TypeError, "Hue must be an integer", id="hue invalid type"
|
|
),
|
|
pytest.param(
|
|
"foo", 0, 0, TypeError, "Hue must be an integer", id="hue invalid type"
|
|
),
|
|
pytest.param(
|
|
0, -1, 0, ValueError, "Invalid saturation", id="saturation out of range"
|
|
),
|
|
pytest.param(
|
|
0, 101, 0, ValueError, "Invalid saturation", id="saturation out of range"
|
|
),
|
|
pytest.param(
|
|
0,
|
|
0.5,
|
|
0,
|
|
TypeError,
|
|
"Saturation must be an integer",
|
|
id="saturation invalid type",
|
|
),
|
|
pytest.param(
|
|
0,
|
|
"foo",
|
|
0,
|
|
TypeError,
|
|
"Saturation must be an integer",
|
|
id="saturation invalid type",
|
|
),
|
|
pytest.param(
|
|
0, 0, -1, ValueError, "Invalid brightness", id="brightness out of range"
|
|
),
|
|
pytest.param(
|
|
0, 0, 101, ValueError, "Invalid brightness", id="brightness out of range"
|
|
),
|
|
pytest.param(
|
|
0,
|
|
0,
|
|
0.5,
|
|
TypeError,
|
|
"Brightness must be an integer",
|
|
id="brightness invalid type",
|
|
),
|
|
pytest.param(
|
|
0,
|
|
0,
|
|
"foo",
|
|
TypeError,
|
|
"Brightness must be an integer",
|
|
id="brightness invalid type",
|
|
),
|
|
],
|
|
)
|
|
async def test_invalid_hsv(
|
|
dev: Device,
|
|
turn_on: bool,
|
|
hue,
|
|
sat,
|
|
brightness,
|
|
exception_cls: type[Exception],
|
|
error: str,
|
|
) -> None:
|
|
light = dev.modules.get(Module.Light)
|
|
assert light
|
|
await handle_turn_on(dev, turn_on)
|
|
assert light.has_feature("hsv")
|
|
with pytest.raises(exception_cls, match=error):
|
|
await light.set_hsv(hue, sat, brightness)
|
|
|
|
|
|
@color_bulb
|
|
@pytest.mark.skip("requires color feature")
|
|
async def test_color_state_information(dev: Device) -> None:
|
|
light = dev.modules.get(Module.Light)
|
|
assert light
|
|
assert "HSV" in dev.state_information
|
|
assert dev.state_information["HSV"] == light.hsv
|
|
|
|
|
|
@non_color_bulb
|
|
async def test_hsv_on_non_color(dev: Device) -> None:
|
|
light = dev.modules.get(Module.Light)
|
|
assert light
|
|
assert not light.has_feature("hsv")
|
|
|
|
with pytest.raises(KasaException):
|
|
await light.set_hsv(0, 0, 0)
|
|
with pytest.raises(KasaException):
|
|
print(light.hsv)
|
|
|
|
|
|
@variable_temp
|
|
@pytest.mark.skip("requires colortemp module")
|
|
async def test_variable_temp_state_information(dev: Device) -> None:
|
|
light = dev.modules.get(Module.Light)
|
|
assert light
|
|
assert "Color temperature" in dev.state_information
|
|
assert dev.state_information["Color temperature"] == light.color_temp
|
|
|
|
|
|
@variable_temp
|
|
@turn_on
|
|
async def test_try_set_colortemp(dev: Device, turn_on: bool) -> None:
|
|
light = dev.modules.get(Module.Light)
|
|
assert light
|
|
await handle_turn_on(dev, turn_on)
|
|
await light.set_color_temp(2700)
|
|
await dev.update()
|
|
assert light.color_temp == 2700
|
|
|
|
|
|
@variable_temp
|
|
async def test_out_of_range_temperature(dev: Device) -> None:
|
|
light = dev.modules.get(Module.Light)
|
|
assert light
|
|
with pytest.raises(
|
|
ValueError, match=r"Temperature should be between \d+ and \d+, was 1000"
|
|
):
|
|
await light.set_color_temp(1000)
|
|
with pytest.raises(
|
|
ValueError, match=r"Temperature should be between \d+ and \d+, was 10000"
|
|
):
|
|
await light.set_color_temp(10000)
|
|
|
|
|
|
@non_variable_temp
|
|
async def test_non_variable_temp(dev: Device) -> None:
|
|
light = dev.modules.get(Module.Light)
|
|
assert light
|
|
with pytest.raises(KasaException):
|
|
await light.set_color_temp(2700)
|
|
|
|
with pytest.raises(KasaException):
|
|
print(light.color_temp)
|
|
|
|
|
|
@bulb
|
|
def test_device_type_bulb(dev: Device) -> None:
|
|
assert dev.device_type in {DeviceType.Bulb, DeviceType.LightStrip}
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("attribute", "use_msg", "use_fn"),
|
|
[
|
|
pytest.param(
|
|
"is_color",
|
|
'use has_feature("hsv") instead',
|
|
lambda device, mod: mod.has_feature("hsv"),
|
|
id="is_color",
|
|
),
|
|
pytest.param(
|
|
"is_dimmable",
|
|
'use has_feature("brightness") instead',
|
|
lambda device, mod: mod.has_feature("brightness"),
|
|
id="is_dimmable",
|
|
),
|
|
pytest.param(
|
|
"is_variable_color_temp",
|
|
'use has_feature("color_temp") instead',
|
|
lambda device, mod: mod.has_feature("color_temp"),
|
|
id="is_variable_color_temp",
|
|
),
|
|
pytest.param(
|
|
"has_effects",
|
|
"check `Module.LightEffect in device.modules` instead",
|
|
lambda device, mod: Module.LightEffect in device.modules,
|
|
id="has_effects",
|
|
),
|
|
],
|
|
)
|
|
@bulb
|
|
async def test_deprecated_light_is_has_attributes(
|
|
dev: Device, attribute: str, use_msg: str, use_fn: Callable[[Device, Module], bool]
|
|
) -> None:
|
|
light = dev.modules.get(Module.Light)
|
|
assert light
|
|
|
|
msg = f"{attribute} is deprecated, {use_msg}"
|
|
with pytest.deprecated_call(match=(re.escape(msg))):
|
|
result = getattr(light, attribute)
|
|
|
|
assert result == use_fn(dev, light)
|
|
|
|
|
|
@bulb
|
|
async def test_deprecated_light_valid_temperature_range(dev: Device) -> None:
|
|
light = dev.modules.get(Module.Light)
|
|
assert light
|
|
|
|
color_temp = light.has_feature("color_temp")
|
|
dep_msg = (
|
|
"valid_temperature_range is deprecated, use "
|
|
'get_feature("color_temp") minimum_value '
|
|
" and maximum_value instead"
|
|
)
|
|
exc_context = pytest.raises(KasaException, match="Color temperature not supported")
|
|
expected_context = nullcontext() if color_temp else exc_context
|
|
|
|
with (
|
|
expected_context,
|
|
pytest.deprecated_call(match=(re.escape(dep_msg))),
|
|
):
|
|
assert light.valid_temperature_range # type: ignore[attr-defined]
|