1
0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 12:59:34 +00:00

Add strict typing to abode (#57673)

This commit is contained in:
Robert Hillis
2022-01-10 09:54:09 -05:00
committed by GitHub
parent bc2f4e82e3
commit 7c51d2f159
25 changed files with 304 additions and 221 deletions

View File

@@ -13,6 +13,7 @@ from homeassistant.const import (
STATE_OFF,
STATE_ON,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from .common import setup_platform
@@ -23,7 +24,7 @@ DEVICE_ID = "switch.test_switch"
DEVICE_UID = "0012a4d3614cb7e2b8c9abea31d2fb2a"
async def test_entity_registry(hass):
async def test_entity_registry(hass: HomeAssistant) -> None:
"""Tests that the devices are registered in the entity registry."""
await setup_platform(hass, SWITCH_DOMAIN)
entity_registry = er.async_get(hass)
@@ -35,7 +36,7 @@ async def test_entity_registry(hass):
assert entry.unique_id == DEVICE_UID
async def test_attributes(hass):
async def test_attributes(hass: HomeAssistant) -> None:
"""Test the switch attributes are correct."""
await setup_platform(hass, SWITCH_DOMAIN)
@@ -43,7 +44,7 @@ async def test_attributes(hass):
assert state.state == STATE_OFF
async def test_switch_on(hass):
async def test_switch_on(hass: HomeAssistant) -> None:
"""Test the switch can be turned on."""
await setup_platform(hass, SWITCH_DOMAIN)
@@ -56,7 +57,7 @@ async def test_switch_on(hass):
mock_switch_on.assert_called_once()
async def test_switch_off(hass):
async def test_switch_off(hass: HomeAssistant) -> None:
"""Test the switch can be turned off."""
await setup_platform(hass, SWITCH_DOMAIN)
@@ -69,7 +70,7 @@ async def test_switch_off(hass):
mock_switch_off.assert_called_once()
async def test_automation_attributes(hass):
async def test_automation_attributes(hass: HomeAssistant) -> None:
"""Test the automation attributes are correct."""
await setup_platform(hass, SWITCH_DOMAIN)
@@ -78,7 +79,7 @@ async def test_automation_attributes(hass):
assert state.state == STATE_ON
async def test_turn_automation_off(hass):
async def test_turn_automation_off(hass: HomeAssistant) -> None:
"""Test the automation can be turned off."""
with patch("abodepy.AbodeAutomation.enable") as mock_trigger:
await setup_platform(hass, SWITCH_DOMAIN)
@@ -94,7 +95,7 @@ async def test_turn_automation_off(hass):
mock_trigger.assert_called_once_with(False)
async def test_turn_automation_on(hass):
async def test_turn_automation_on(hass: HomeAssistant) -> None:
"""Test the automation can be turned on."""
with patch("abodepy.AbodeAutomation.enable") as mock_trigger:
await setup_platform(hass, SWITCH_DOMAIN)
@@ -110,7 +111,7 @@ async def test_turn_automation_on(hass):
mock_trigger.assert_called_once_with(True)
async def test_trigger_automation(hass, requests_mock):
async def test_trigger_automation(hass: HomeAssistant) -> None:
"""Test the trigger automation service."""
await setup_platform(hass, SWITCH_DOMAIN)