evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve type hints in homekit and homekit_controller tests (#124213)

Browse source 
Add missing hass type hint in homekit and homekit_controller tests
This commit is contained in:
epenet 2024-08-20 12:47:41 +02:00 • committed by GitHub
parent be255613de
commit db92f29c00
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
24 changed files with 132 additions and 85 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/components/homekit_controller/test_light.py
View file

@ -3,8 +3,9 @@
from collections.abc import Callable
from unittest import mock
from aiohomekit.model import Accessory
from aiohomekit.model.characteristics import CharacteristicsTypes
from aiohomekit.model.services import ServicesTypes
from aiohomekit.model.services import Service, ServicesTypes
from aiohomekit.testing import FakeController
from homeassistant.components.homekit_controller.const import KNOWN_DEVICES
@ -23,7 +24,7 @@ LIGHT_BULB_NAME = "TestDevice"
LIGHT_BULB_ENTITY_ID = "light.testdevice"
def create_lightbulb_service(accessory):
def create_lightbulb_service(accessory: Accessory) -> Service:
"""Define lightbulb characteristics."""
service = accessory.add_service(ServicesTypes.LIGHTBULB, name=LIGHT_BULB_NAME)
@ -36,7 +37,7 @@ def create_lightbulb_service(accessory):
return service
def create_lightbulb_service_with_hs(accessory):
def create_lightbulb_service_with_hs(accessory: Accessory) -> Service:
"""Define a lightbulb service with hue + saturation."""
service = create_lightbulb_service(accessory)
@ -49,7 +50,7 @@ def create_lightbulb_service_with_hs(accessory):
return service
def create_lightbulb_service_with_color_temp(accessory):
def create_lightbulb_service_with_color_temp(accessory: Accessory) -> Service:
"""Define a lightbulb service with color temp."""
service = create_lightbulb_service(accessory)