Integrations h* - i*: Rename HomeAssistantType to HomeAssistant. (#49587)

This commit is contained in:
jan iversen 2021-04-23 10:11:58 +02:00 committed by GitHub
parent a396619251
commit 9685cefba4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
26 changed files with 143 additions and 156 deletions

View File

@ -20,7 +20,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect, async_dispatcher_connect,
async_dispatcher_send, async_dispatcher_send,
) )
from homeassistant.helpers.typing import ConfigType, HomeAssistantType from homeassistant.helpers.typing import ConfigType
from .const import ( from .const import (
CONF_INSTANCE_CLIENTS, CONF_INSTANCE_CLIENTS,
@ -290,16 +290,12 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
return True return True
async def _async_entry_updated( async def _async_entry_updated(hass: HomeAssistant, config_entry: ConfigEntry) -> None:
hass: HomeAssistantType, config_entry: ConfigEntry
) -> None:
"""Handle entry updates.""" """Handle entry updates."""
await hass.config_entries.async_reload(config_entry.entry_id) await hass.config_entries.async_reload(config_entry.entry_id)
async def async_unload_entry( async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
hass: HomeAssistantType, config_entry: ConfigEntry
) -> bool:
"""Unload a config entry.""" """Unload a config entry."""
unload_ok = all( unload_ok = all(
await asyncio.gather( await asyncio.gather(

View File

@ -19,12 +19,11 @@ from homeassistant.components.light import (
LightEntity, LightEntity,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import ( from homeassistant.helpers.dispatcher import (
async_dispatcher_connect, async_dispatcher_connect,
async_dispatcher_send, async_dispatcher_send,
) )
from homeassistant.helpers.typing import HomeAssistantType
import homeassistant.util.color as color_util import homeassistant.util.color as color_util
from . import ( from . import (
@ -81,7 +80,7 @@ ICON_EXTERNAL_SOURCE = "mdi:television-ambient-light"
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, config_entry: ConfigEntry, async_add_entities: Callable hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: Callable
) -> bool: ) -> bool:
"""Set up a Hyperion platform from config entry.""" """Set up a Hyperion platform from config entry."""

View File

@ -26,12 +26,11 @@ from hyperion.const import (
from homeassistant.components.switch import SwitchEntity from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import ( from homeassistant.helpers.dispatcher import (
async_dispatcher_connect, async_dispatcher_connect,
async_dispatcher_send, async_dispatcher_send,
) )
from homeassistant.helpers.typing import HomeAssistantType
from homeassistant.util import slugify from homeassistant.util import slugify
from . import ( from . import (
@ -82,7 +81,7 @@ def _component_to_switch_name(component: str, instance_name: str) -> str:
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, config_entry: ConfigEntry, async_add_entities: Callable hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: Callable
) -> bool: ) -> bool:
"""Set up a Hyperion platform from config entry.""" """Set up a Hyperion platform from config entry."""
entry_data = hass.data[DOMAIN][config_entry.entry_id] entry_data = hass.data[DOMAIN][config_entry.entry_id]

View File

@ -27,6 +27,7 @@ from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers.aiohttp_client import async_get_clientsession from homeassistant.helpers.aiohttp_client import async_get_clientsession
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
@ -36,7 +37,7 @@ from homeassistant.helpers.dispatcher import (
) )
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import Entity
from homeassistant.helpers.event import async_track_time_interval from homeassistant.helpers.event import async_track_time_interval
from homeassistant.helpers.typing import ConfigType, HomeAssistantType from homeassistant.helpers.typing import ConfigType
from .const import DOMAIN, UPDATE_INTERVAL from .const import DOMAIN, UPDATE_INTERVAL
@ -58,7 +59,7 @@ CONFIG_SCHEMA = vol.Schema(
) )
async def async_setup(hass: HomeAssistantType, config: ConfigType) -> None: async def async_setup(hass: HomeAssistant, config: ConfigType) -> None:
"""Set up the Aqualink component.""" """Set up the Aqualink component."""
conf = config.get(DOMAIN) conf = config.get(DOMAIN)
@ -74,7 +75,7 @@ async def async_setup(hass: HomeAssistantType, config: ConfigType) -> None:
return True return True
async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> None: async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Set up Aqualink from a config entry.""" """Set up Aqualink from a config entry."""
username = entry.data[CONF_USERNAME] username = entry.data[CONF_USERNAME]
password = entry.data[CONF_PASSWORD] password = entry.data[CONF_PASSWORD]
@ -157,7 +158,7 @@ async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> None
return True return True
async def async_unload_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool: async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry.""" """Unload a config entry."""
forward_unload = hass.config_entries.async_forward_entry_unload forward_unload = hass.config_entries.async_forward_entry_unload

View File

@ -5,7 +5,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity, BinarySensorEntity,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from . import AqualinkEntity from . import AqualinkEntity
from .const import DOMAIN as AQUALINK_DOMAIN from .const import DOMAIN as AQUALINK_DOMAIN
@ -14,7 +14,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, config_entry: ConfigEntry, async_add_entities hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities
) -> None: ) -> None:
"""Set up discovered binary sensors.""" """Set up discovered binary sensors."""
devs = [] devs = []

View File

@ -20,7 +20,7 @@ from homeassistant.components.climate.const import (
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from . import AqualinkEntity, refresh_system from . import AqualinkEntity, refresh_system
from .const import CLIMATE_SUPPORTED_MODES, DOMAIN as AQUALINK_DOMAIN from .const import CLIMATE_SUPPORTED_MODES, DOMAIN as AQUALINK_DOMAIN
@ -31,7 +31,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, config_entry: ConfigEntry, async_add_entities hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities
) -> None: ) -> None:
"""Set up discovered switches.""" """Set up discovered switches."""
devs = [] devs = []

View File

@ -10,7 +10,7 @@ from homeassistant.components.light import (
LightEntity, LightEntity,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from . import AqualinkEntity, refresh_system from . import AqualinkEntity, refresh_system
from .const import DOMAIN as AQUALINK_DOMAIN from .const import DOMAIN as AQUALINK_DOMAIN
@ -19,7 +19,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, config_entry: ConfigEntry, async_add_entities hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities
) -> None: ) -> None:
"""Set up discovered lights.""" """Set up discovered lights."""
devs = [] devs = []

View File

@ -4,7 +4,7 @@ from __future__ import annotations
from homeassistant.components.sensor import DOMAIN, SensorEntity from homeassistant.components.sensor import DOMAIN, SensorEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT from homeassistant.const import DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from . import AqualinkEntity from . import AqualinkEntity
from .const import DOMAIN as AQUALINK_DOMAIN from .const import DOMAIN as AQUALINK_DOMAIN
@ -13,7 +13,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, config_entry: ConfigEntry, async_add_entities hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities
) -> None: ) -> None:
"""Set up discovered sensors.""" """Set up discovered sensors."""
devs = [] devs = []

View File

@ -1,7 +1,7 @@
"""Support for Aqualink pool feature switches.""" """Support for Aqualink pool feature switches."""
from homeassistant.components.switch import DOMAIN, SwitchEntity from homeassistant.components.switch import DOMAIN, SwitchEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from . import AqualinkEntity, refresh_system from . import AqualinkEntity, refresh_system
from .const import DOMAIN as AQUALINK_DOMAIN from .const import DOMAIN as AQUALINK_DOMAIN
@ -10,7 +10,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, config_entry: ConfigEntry, async_add_entities hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities
) -> None: ) -> None:
"""Set up discovered switches.""" """Set up discovered switches."""
devs = [] devs = []

View File

@ -5,8 +5,9 @@ import voluptuous as vol
from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.typing import ConfigType, HomeAssistantType, ServiceDataType from homeassistant.helpers.typing import ConfigType, ServiceDataType
from homeassistant.util import slugify from homeassistant.util import slugify
from .account import IcloudAccount from .account import IcloudAccount
@ -86,7 +87,7 @@ CONFIG_SCHEMA = vol.Schema(
) )
async def async_setup(hass: HomeAssistantType, config: ConfigType) -> bool: async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up iCloud from legacy config file.""" """Set up iCloud from legacy config file."""
conf = config.get(DOMAIN) conf = config.get(DOMAIN)
@ -103,7 +104,7 @@ async def async_setup(hass: HomeAssistantType, config: ConfigType) -> bool:
return True return True
async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool: async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up an iCloud account from a config entry.""" """Set up an iCloud account from a config entry."""
hass.data.setdefault(DOMAIN, {}) hass.data.setdefault(DOMAIN, {})
@ -221,7 +222,7 @@ async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool
return True return True
async def async_unload_entry(hass: HomeAssistantType, entry: ConfigEntry): async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
"""Unload a config entry.""" """Unload a config entry."""
unload_ok = all( unload_ok = all(
await asyncio.gather( await asyncio.gather(

View File

@ -16,11 +16,11 @@ from pyicloud.services.findmyiphone import AppleDevice
from homeassistant.components.zone import async_active_zone from homeassistant.components.zone import async_active_zone
from homeassistant.config_entries import SOURCE_REAUTH, ConfigEntry from homeassistant.config_entries import SOURCE_REAUTH, ConfigEntry
from homeassistant.const import ATTR_ATTRIBUTION, CONF_USERNAME from homeassistant.const import ATTR_ATTRIBUTION, CONF_USERNAME
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers.dispatcher import dispatcher_send from homeassistant.helpers.dispatcher import dispatcher_send
from homeassistant.helpers.event import track_point_in_utc_time from homeassistant.helpers.event import track_point_in_utc_time
from homeassistant.helpers.storage import Store from homeassistant.helpers.storage import Store
from homeassistant.helpers.typing import HomeAssistantType
from homeassistant.util import slugify from homeassistant.util import slugify
from homeassistant.util.async_ import run_callback_threadsafe from homeassistant.util.async_ import run_callback_threadsafe
from homeassistant.util.dt import utcnow from homeassistant.util.dt import utcnow
@ -76,7 +76,7 @@ class IcloudAccount:
def __init__( def __init__(
self, self,
hass: HomeAssistantType, hass: HomeAssistant,
username: str, username: str,
password: str, password: str,
icloud_dir: Store, icloud_dir: Store,

View File

@ -4,9 +4,8 @@ from __future__ import annotations
from homeassistant.components.device_tracker import SOURCE_TYPE_GPS from homeassistant.components.device_tracker import SOURCE_TYPE_GPS
from homeassistant.components.device_tracker.config_entry import TrackerEntity from homeassistant.components.device_tracker.config_entry import TrackerEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.typing import HomeAssistantType
from .account import IcloudAccount, IcloudDevice from .account import IcloudAccount, IcloudDevice
from .const import ( from .const import (
@ -17,14 +16,12 @@ from .const import (
) )
async def async_setup_scanner( async def async_setup_scanner(hass: HomeAssistant, config, see, discovery_info=None):
hass: HomeAssistantType, config, see, discovery_info=None
):
"""Old way of setting up the iCloud tracker.""" """Old way of setting up the iCloud tracker."""
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, entry: ConfigEntry, async_add_entities hass: HomeAssistant, entry: ConfigEntry, async_add_entities
) -> None: ) -> None:
"""Set up device tracker for iCloud component.""" """Set up device tracker for iCloud component."""
account = hass.data[DOMAIN][entry.unique_id] account = hass.data[DOMAIN][entry.unique_id]

View File

@ -4,17 +4,16 @@ from __future__ import annotations
from homeassistant.components.sensor import SensorEntity from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import DEVICE_CLASS_BATTERY, PERCENTAGE from homeassistant.const import DEVICE_CLASS_BATTERY, PERCENTAGE
from homeassistant.core import callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.icon import icon_for_battery_level from homeassistant.helpers.icon import icon_for_battery_level
from homeassistant.helpers.typing import HomeAssistantType
from .account import IcloudAccount, IcloudDevice from .account import IcloudAccount, IcloudDevice
from .const import DOMAIN from .const import DOMAIN
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, entry: ConfigEntry, async_add_entities hass: HomeAssistant, entry: ConfigEntry, async_add_entities
) -> None: ) -> None:
"""Set up device tracker for iCloud component.""" """Set up device tracker for iCloud component."""
account = hass.data[DOMAIN][entry.unique_id] account = hass.data[DOMAIN][entry.unique_id]

View File

@ -26,9 +26,8 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity, BinarySensorEntity,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.event import async_track_point_in_utc_time from homeassistant.helpers.event import async_track_point_in_utc_time
from homeassistant.helpers.typing import HomeAssistantType
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
from .const import ( from .const import (
@ -60,7 +59,7 @@ DEVICE_PARENT_REQUIRED = [
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, hass: HomeAssistant,
entry: ConfigEntry, entry: ConfigEntry,
async_add_entities: Callable[[list], None], async_add_entities: Callable[[list], None],
) -> bool: ) -> bool:

View File

@ -34,7 +34,7 @@ from homeassistant.const import (
TEMP_CELSIUS, TEMP_CELSIUS,
TEMP_FAHRENHEIT, TEMP_FAHRENHEIT,
) )
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from .const import ( from .const import (
_LOGGER, _LOGGER,
@ -61,7 +61,7 @@ ISY_SUPPORTED_FEATURES = (
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, hass: HomeAssistant,
entry: ConfigEntry, entry: ConfigEntry,
async_add_entities: Callable[[list], None], async_add_entities: Callable[[list], None],
) -> bool: ) -> bool:

View File

@ -12,7 +12,7 @@ from homeassistant.components.cover import (
CoverEntity, CoverEntity,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from .const import ( from .const import (
_LOGGER, _LOGGER,
@ -27,7 +27,7 @@ from .helpers import migrate_old_unique_ids
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, hass: HomeAssistant,
entry: ConfigEntry, entry: ConfigEntry,
async_add_entities: Callable[[list], None], async_add_entities: Callable[[list], None],
) -> bool: ) -> bool:

View File

@ -8,7 +8,7 @@ from pyisy.constants import ISY_VALUE_UNKNOWN, PROTO_INSTEON
from homeassistant.components.fan import DOMAIN as FAN, SUPPORT_SET_SPEED, FanEntity from homeassistant.components.fan import DOMAIN as FAN, SUPPORT_SET_SPEED, FanEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from homeassistant.util.percentage import ( from homeassistant.util.percentage import (
int_states_in_range, int_states_in_range,
percentage_to_ranged_value, percentage_to_ranged_value,
@ -23,7 +23,7 @@ SPEED_RANGE = (1, 255) # off is not included
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, hass: HomeAssistant,
entry: ConfigEntry, entry: ConfigEntry,
async_add_entities: Callable[[list], None], async_add_entities: Callable[[list], None],
) -> bool: ) -> bool:

View File

@ -21,8 +21,8 @@ from homeassistant.components.fan import DOMAIN as FAN
from homeassistant.components.light import DOMAIN as LIGHT from homeassistant.components.light import DOMAIN as LIGHT
from homeassistant.components.sensor import DOMAIN as SENSOR from homeassistant.components.sensor import DOMAIN as SENSOR
from homeassistant.components.switch import DOMAIN as SWITCH from homeassistant.components.switch import DOMAIN as SWITCH
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_registry import async_get_registry from homeassistant.helpers.entity_registry import async_get_registry
from homeassistant.helpers.typing import HomeAssistantType
from .const import ( from .const import (
_LOGGER, _LOGGER,
@ -366,7 +366,7 @@ def _categorize_variables(
async def migrate_old_unique_ids( async def migrate_old_unique_ids(
hass: HomeAssistantType, platform: str, devices: list[Any] | None hass: HomeAssistant, platform: str, devices: list[Any] | None
) -> None: ) -> None:
"""Migrate to new controller-specific unique ids.""" """Migrate to new controller-specific unique ids."""
registry = await async_get_registry(hass) registry = await async_get_registry(hass)

View File

@ -11,8 +11,8 @@ from homeassistant.components.light import (
LightEntity, LightEntity,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.restore_state import RestoreEntity from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import HomeAssistantType
from .const import ( from .const import (
_LOGGER, _LOGGER,
@ -29,7 +29,7 @@ ATTR_LAST_BRIGHTNESS = "last_brightness"
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, hass: HomeAssistant,
entry: ConfigEntry, entry: ConfigEntry,
async_add_entities: Callable[[list], None], async_add_entities: Callable[[list], None],
) -> bool: ) -> bool:

View File

@ -5,7 +5,7 @@ from pyisy.constants import ISY_VALUE_UNKNOWN
from homeassistant.components.lock import DOMAIN as LOCK, LockEntity from homeassistant.components.lock import DOMAIN as LOCK, LockEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from .const import _LOGGER, DOMAIN as ISY994_DOMAIN, ISY994_NODES, ISY994_PROGRAMS from .const import _LOGGER, DOMAIN as ISY994_DOMAIN, ISY994_NODES, ISY994_PROGRAMS
from .entity import ISYNodeEntity, ISYProgramEntity from .entity import ISYNodeEntity, ISYProgramEntity
@ -15,7 +15,7 @@ VALUE_TO_STATE = {0: False, 100: True}
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, hass: HomeAssistant,
entry: ConfigEntry, entry: ConfigEntry,
async_add_entities: Callable[[list], None], async_add_entities: Callable[[list], None],
) -> bool: ) -> bool:

View File

@ -8,7 +8,7 @@ from pyisy.constants import ISY_VALUE_UNKNOWN
from homeassistant.components.sensor import DOMAIN as SENSOR, SensorEntity from homeassistant.components.sensor import DOMAIN as SENSOR, SensorEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from .const import ( from .const import (
_LOGGER, _LOGGER,
@ -26,7 +26,7 @@ from .helpers import convert_isy_value_to_hass, migrate_old_unique_ids
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistantType, hass: HomeAssistant,
entry: ConfigEntry, entry: ConfigEntry,
async_add_entities: Callable[[list], None], async_add_entities: Callable[[list], None],
) -> bool: ) -> bool:

View File

@ -11,8 +11,8 @@ from homeassistant.components.hyperion import get_hyperion_unique_id
from homeassistant.components.hyperion.const import CONF_PRIORITY, DOMAIN from homeassistant.components.hyperion.const import CONF_PRIORITY, DOMAIN
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PORT from homeassistant.const import CONF_HOST, CONF_PORT
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.typing import HomeAssistantType
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
@ -120,7 +120,7 @@ def create_mock_client() -> Mock:
def add_test_config_entry( def add_test_config_entry(
hass: HomeAssistantType, hass: HomeAssistant,
data: dict[str, Any] | None = None, data: dict[str, Any] | None = None,
options: dict[str, Any] | None = None, options: dict[str, Any] | None = None,
) -> ConfigEntry: ) -> ConfigEntry:
@ -142,7 +142,7 @@ def add_test_config_entry(
async def setup_test_config_entry( async def setup_test_config_entry(
hass: HomeAssistantType, hass: HomeAssistant,
config_entry: ConfigEntry | None = None, config_entry: ConfigEntry | None = None,
hyperion_client: Mock | None = None, hyperion_client: Mock | None = None,
options: dict[str, Any] | None = None, options: dict[str, Any] | None = None,
@ -173,7 +173,7 @@ def call_registered_callback(
def register_test_entity( def register_test_entity(
hass: HomeAssistantType, domain: str, type_name: str, entity_id: str hass: HomeAssistant, domain: str, type_name: str, entity_id: str
) -> None: ) -> None:
"""Register a test entity.""" """Register a test entity."""
unique_id = get_hyperion_unique_id(TEST_SYSINFO_ID, TEST_INSTANCE, type_name) unique_id = get_hyperion_unique_id(TEST_SYSINFO_ID, TEST_INSTANCE, type_name)

View File

@ -25,7 +25,7 @@ from homeassistant.const import (
CONF_TOKEN, CONF_TOKEN,
SERVICE_TURN_ON, SERVICE_TURN_ON,
) )
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from . import ( from . import (
TEST_AUTH_REQUIRED_RESP, TEST_AUTH_REQUIRED_RESP,
@ -98,7 +98,7 @@ TEST_SSDP_SERVICE_INFO = {
} }
async def _create_mock_entry(hass: HomeAssistantType) -> MockConfigEntry: async def _create_mock_entry(hass: HomeAssistant) -> MockConfigEntry:
"""Add a test Hyperion entity to hass.""" """Add a test Hyperion entity to hass."""
entry: MockConfigEntry = MockConfigEntry( # type: ignore[no-untyped-call] entry: MockConfigEntry = MockConfigEntry( # type: ignore[no-untyped-call]
entry_id=TEST_CONFIG_ENTRY_ID, entry_id=TEST_CONFIG_ENTRY_ID,
@ -125,7 +125,7 @@ async def _create_mock_entry(hass: HomeAssistantType) -> MockConfigEntry:
async def _init_flow( async def _init_flow(
hass: HomeAssistantType, hass: HomeAssistant,
source: str = SOURCE_USER, source: str = SOURCE_USER,
data: dict[str, Any] | None = None, data: dict[str, Any] | None = None,
) -> Any: ) -> Any:
@ -138,7 +138,7 @@ async def _init_flow(
async def _configure_flow( async def _configure_flow(
hass: HomeAssistantType, result: dict, user_input: dict[str, Any] | None = None hass: HomeAssistant, result: dict, user_input: dict[str, Any] | None = None
) -> Any: ) -> Any:
"""Provide input to a flow.""" """Provide input to a flow."""
user_input = user_input or {} user_input = user_input or {}
@ -156,7 +156,7 @@ async def _configure_flow(
return result return result
async def test_user_if_no_configuration(hass: HomeAssistantType) -> None: async def test_user_if_no_configuration(hass: HomeAssistant) -> None:
"""Check flow behavior when no configuration is present.""" """Check flow behavior when no configuration is present."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -165,7 +165,7 @@ async def test_user_if_no_configuration(hass: HomeAssistantType) -> None:
assert result["handler"] == DOMAIN assert result["handler"] == DOMAIN
async def test_user_existing_id_abort(hass: HomeAssistantType) -> None: async def test_user_existing_id_abort(hass: HomeAssistant) -> None:
"""Verify a duplicate ID results in an abort.""" """Verify a duplicate ID results in an abort."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -179,7 +179,7 @@ async def test_user_existing_id_abort(hass: HomeAssistantType) -> None:
assert result["reason"] == "already_configured" assert result["reason"] == "already_configured"
async def test_user_client_errors(hass: HomeAssistantType) -> None: async def test_user_client_errors(hass: HomeAssistant) -> None:
"""Verify correct behaviour with client errors.""" """Verify correct behaviour with client errors."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -205,7 +205,7 @@ async def test_user_client_errors(hass: HomeAssistantType) -> None:
assert result["reason"] == "auth_required_error" assert result["reason"] == "auth_required_error"
async def test_user_confirm_cannot_connect(hass: HomeAssistantType) -> None: async def test_user_confirm_cannot_connect(hass: HomeAssistant) -> None:
"""Test a failure to connect during confirmation.""" """Test a failure to connect during confirmation."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -224,7 +224,7 @@ async def test_user_confirm_cannot_connect(hass: HomeAssistantType) -> None:
assert result["reason"] == "cannot_connect" assert result["reason"] == "cannot_connect"
async def test_user_confirm_id_error(hass: HomeAssistantType) -> None: async def test_user_confirm_id_error(hass: HomeAssistant) -> None:
"""Test a failure fetching the server id during confirmation.""" """Test a failure fetching the server id during confirmation."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -240,7 +240,7 @@ async def test_user_confirm_id_error(hass: HomeAssistantType) -> None:
assert result["reason"] == "no_id" assert result["reason"] == "no_id"
async def test_user_noauth_flow_success(hass: HomeAssistantType) -> None: async def test_user_noauth_flow_success(hass: HomeAssistant) -> None:
"""Check a full flow without auth.""" """Check a full flow without auth."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -258,7 +258,7 @@ async def test_user_noauth_flow_success(hass: HomeAssistantType) -> None:
} }
async def test_user_auth_required(hass: HomeAssistantType) -> None: async def test_user_auth_required(hass: HomeAssistant) -> None:
"""Verify correct behaviour when auth is required.""" """Verify correct behaviour when auth is required."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -273,7 +273,7 @@ async def test_user_auth_required(hass: HomeAssistantType) -> None:
assert result["step_id"] == "auth" assert result["step_id"] == "auth"
async def test_auth_static_token_auth_required_fail(hass: HomeAssistantType) -> None: async def test_auth_static_token_auth_required_fail(hass: HomeAssistant) -> None:
"""Verify correct behaviour with a failed auth required call.""" """Verify correct behaviour with a failed auth required call."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -287,7 +287,7 @@ async def test_auth_static_token_auth_required_fail(hass: HomeAssistantType) ->
assert result["reason"] == "auth_required_error" assert result["reason"] == "auth_required_error"
async def test_auth_static_token_success(hass: HomeAssistantType) -> None: async def test_auth_static_token_success(hass: HomeAssistant) -> None:
"""Test a successful flow with a static token.""" """Test a successful flow with a static token."""
result = await _init_flow(hass) result = await _init_flow(hass)
assert result["step_id"] == "user" assert result["step_id"] == "user"
@ -312,7 +312,7 @@ async def test_auth_static_token_success(hass: HomeAssistantType) -> None:
} }
async def test_auth_static_token_login_connect_fail(hass: HomeAssistantType) -> None: async def test_auth_static_token_login_connect_fail(hass: HomeAssistant) -> None:
"""Test correct behavior with a static token that cannot connect.""" """Test correct behavior with a static token that cannot connect."""
result = await _init_flow(hass) result = await _init_flow(hass)
assert result["step_id"] == "user" assert result["step_id"] == "user"
@ -333,7 +333,7 @@ async def test_auth_static_token_login_connect_fail(hass: HomeAssistantType) ->
assert result["reason"] == "cannot_connect" assert result["reason"] == "cannot_connect"
async def test_auth_static_token_login_fail(hass: HomeAssistantType) -> None: async def test_auth_static_token_login_fail(hass: HomeAssistant) -> None:
"""Test correct behavior with a static token that cannot login.""" """Test correct behavior with a static token that cannot login."""
result = await _init_flow(hass) result = await _init_flow(hass)
assert result["step_id"] == "user" assert result["step_id"] == "user"
@ -356,7 +356,7 @@ async def test_auth_static_token_login_fail(hass: HomeAssistantType) -> None:
assert result["errors"]["base"] == "invalid_access_token" assert result["errors"]["base"] == "invalid_access_token"
async def test_auth_create_token_approval_declined(hass: HomeAssistantType) -> None: async def test_auth_create_token_approval_declined(hass: HomeAssistant) -> None:
"""Verify correct behaviour when a token request is declined.""" """Verify correct behaviour when a token request is declined."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -400,7 +400,7 @@ async def test_auth_create_token_approval_declined(hass: HomeAssistantType) -> N
async def test_auth_create_token_approval_declined_task_canceled( async def test_auth_create_token_approval_declined_task_canceled(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Verify correct behaviour when a token request is declined.""" """Verify correct behaviour when a token request is declined."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -461,7 +461,7 @@ async def test_auth_create_token_approval_declined_task_canceled(
async def test_auth_create_token_when_issued_token_fails( async def test_auth_create_token_when_issued_token_fails(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Verify correct behaviour when a token is granted by fails to authenticate.""" """Verify correct behaviour when a token is granted by fails to authenticate."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -506,7 +506,7 @@ async def test_auth_create_token_when_issued_token_fails(
assert result["reason"] == "cannot_connect" assert result["reason"] == "cannot_connect"
async def test_auth_create_token_success(hass: HomeAssistantType) -> None: async def test_auth_create_token_success(hass: HomeAssistant) -> None:
"""Verify correct behaviour when a token is successfully created.""" """Verify correct behaviour when a token is successfully created."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -552,7 +552,7 @@ async def test_auth_create_token_success(hass: HomeAssistantType) -> None:
async def test_auth_create_token_success_but_login_fail( async def test_auth_create_token_success_but_login_fail(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Verify correct behaviour when a token is successfully created but the login fails.""" """Verify correct behaviour when a token is successfully created but the login fails."""
result = await _init_flow(hass) result = await _init_flow(hass)
@ -592,7 +592,7 @@ async def test_auth_create_token_success_but_login_fail(
assert result["reason"] == "auth_new_token_not_work_error" assert result["reason"] == "auth_new_token_not_work_error"
async def test_ssdp_success(hass: HomeAssistantType) -> None: async def test_ssdp_success(hass: HomeAssistant) -> None:
"""Check an SSDP flow.""" """Check an SSDP flow."""
client = create_mock_client() client = create_mock_client()
@ -617,7 +617,7 @@ async def test_ssdp_success(hass: HomeAssistantType) -> None:
} }
async def test_ssdp_cannot_connect(hass: HomeAssistantType) -> None: async def test_ssdp_cannot_connect(hass: HomeAssistant) -> None:
"""Check an SSDP flow that cannot connect.""" """Check an SSDP flow that cannot connect."""
client = create_mock_client() client = create_mock_client()
@ -633,7 +633,7 @@ async def test_ssdp_cannot_connect(hass: HomeAssistantType) -> None:
assert result["reason"] == "cannot_connect" assert result["reason"] == "cannot_connect"
async def test_ssdp_missing_serial(hass: HomeAssistantType) -> None: async def test_ssdp_missing_serial(hass: HomeAssistant) -> None:
"""Check an SSDP flow where no id is provided.""" """Check an SSDP flow where no id is provided."""
client = create_mock_client() client = create_mock_client()
@ -650,7 +650,7 @@ async def test_ssdp_missing_serial(hass: HomeAssistantType) -> None:
assert result["reason"] == "no_id" assert result["reason"] == "no_id"
async def test_ssdp_failure_bad_port_json(hass: HomeAssistantType) -> None: async def test_ssdp_failure_bad_port_json(hass: HomeAssistant) -> None:
"""Check an SSDP flow with bad json port.""" """Check an SSDP flow with bad json port."""
client = create_mock_client() client = create_mock_client()
@ -668,7 +668,7 @@ async def test_ssdp_failure_bad_port_json(hass: HomeAssistantType) -> None:
assert result["data"][CONF_PORT] == const.DEFAULT_PORT_JSON assert result["data"][CONF_PORT] == const.DEFAULT_PORT_JSON
async def test_ssdp_failure_bad_port_ui(hass: HomeAssistantType) -> None: async def test_ssdp_failure_bad_port_ui(hass: HomeAssistant) -> None:
"""Check an SSDP flow with bad ui port.""" """Check an SSDP flow with bad ui port."""
client = create_mock_client() client = create_mock_client()
@ -703,7 +703,7 @@ async def test_ssdp_failure_bad_port_ui(hass: HomeAssistantType) -> None:
} }
async def test_ssdp_abort_duplicates(hass: HomeAssistantType) -> None: async def test_ssdp_abort_duplicates(hass: HomeAssistant) -> None:
"""Check an SSDP flow where no id is provided.""" """Check an SSDP flow where no id is provided."""
client = create_mock_client() client = create_mock_client()
@ -723,7 +723,7 @@ async def test_ssdp_abort_duplicates(hass: HomeAssistantType) -> None:
assert result_2["reason"] == "already_in_progress" assert result_2["reason"] == "already_in_progress"
async def test_options_priority(hass: HomeAssistantType) -> None: async def test_options_priority(hass: HomeAssistant) -> None:
"""Check an options flow priority option.""" """Check an options flow priority option."""
config_entry = add_test_config_entry(hass) config_entry = add_test_config_entry(hass)
@ -761,7 +761,7 @@ async def test_options_priority(hass: HomeAssistantType) -> None:
assert client.async_send_set_color.call_args[1][CONF_PRIORITY] == new_priority assert client.async_send_set_color.call_args[1][CONF_PRIORITY] == new_priority
async def test_options_effect_show_list(hass: HomeAssistantType) -> None: async def test_options_effect_show_list(hass: HomeAssistant) -> None:
"""Check an options flow effect show list.""" """Check an options flow effect show list."""
config_entry = add_test_config_entry(hass) config_entry = add_test_config_entry(hass)
@ -795,7 +795,7 @@ async def test_options_effect_show_list(hass: HomeAssistantType) -> None:
) )
async def test_options_effect_hide_list_cannot_connect(hass: HomeAssistantType) -> None: async def test_options_effect_hide_list_cannot_connect(hass: HomeAssistant) -> None:
"""Check an options flow effect hide list with a failed connection.""" """Check an options flow effect hide list with a failed connection."""
config_entry = add_test_config_entry(hass) config_entry = add_test_config_entry(hass)
@ -814,7 +814,7 @@ async def test_options_effect_hide_list_cannot_connect(hass: HomeAssistantType)
assert result["reason"] == "cannot_connect" assert result["reason"] == "cannot_connect"
async def test_reauth_success(hass: HomeAssistantType) -> None: async def test_reauth_success(hass: HomeAssistant) -> None:
"""Check a reauth flow that succeeds.""" """Check a reauth flow that succeeds."""
config_data = { config_data = {
@ -848,7 +848,7 @@ async def test_reauth_success(hass: HomeAssistantType) -> None:
assert CONF_TOKEN in config_entry.data assert CONF_TOKEN in config_entry.data
async def test_reauth_cannot_connect(hass: HomeAssistantType) -> None: async def test_reauth_cannot_connect(hass: HomeAssistant) -> None:
"""Check a reauth flow that fails to connect.""" """Check a reauth flow that fails to connect."""
config_data = { config_data = {

View File

@ -39,8 +39,8 @@ from homeassistant.const import (
SERVICE_TURN_OFF, SERVICE_TURN_OFF,
SERVICE_TURN_ON, SERVICE_TURN_ON,
) )
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.typing import HomeAssistantType
from homeassistant.util import dt from homeassistant.util import dt
import homeassistant.util.color as color_util import homeassistant.util.color as color_util
@ -74,7 +74,7 @@ COLOR_BLACK = color_util.COLORS["black"]
def _get_config_entry_from_unique_id( def _get_config_entry_from_unique_id(
hass: HomeAssistantType, unique_id: str hass: HomeAssistant, unique_id: str
) -> ConfigEntry | None: ) -> ConfigEntry | None:
for entry in hass.config_entries.async_entries(domain=DOMAIN): for entry in hass.config_entries.async_entries(domain=DOMAIN):
if TEST_SYSINFO_ID == entry.unique_id: if TEST_SYSINFO_ID == entry.unique_id:
@ -82,14 +82,14 @@ def _get_config_entry_from_unique_id(
return None return None
async def test_setup_config_entry(hass: HomeAssistantType) -> None: async def test_setup_config_entry(hass: HomeAssistant) -> None:
"""Test setting up the component via config entries.""" """Test setting up the component via config entries."""
await setup_test_config_entry(hass, hyperion_client=create_mock_client()) await setup_test_config_entry(hass, hyperion_client=create_mock_client())
assert hass.states.get(TEST_ENTITY_ID_1) is not None assert hass.states.get(TEST_ENTITY_ID_1) is not None
async def test_setup_config_entry_not_ready_connect_fail( async def test_setup_config_entry_not_ready_connect_fail(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test the component not being ready.""" """Test the component not being ready."""
client = create_mock_client() client = create_mock_client()
@ -99,7 +99,7 @@ async def test_setup_config_entry_not_ready_connect_fail(
async def test_setup_config_entry_not_ready_switch_instance_fail( async def test_setup_config_entry_not_ready_switch_instance_fail(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test the component not being ready.""" """Test the component not being ready."""
client = create_mock_client() client = create_mock_client()
@ -110,7 +110,7 @@ async def test_setup_config_entry_not_ready_switch_instance_fail(
async def test_setup_config_entry_not_ready_load_state_fail( async def test_setup_config_entry_not_ready_load_state_fail(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test the component not being ready.""" """Test the component not being ready."""
client = create_mock_client() client = create_mock_client()
@ -126,7 +126,7 @@ async def test_setup_config_entry_not_ready_load_state_fail(
assert hass.states.get(TEST_ENTITY_ID_1) is None assert hass.states.get(TEST_ENTITY_ID_1) is None
async def test_setup_config_entry_dynamic_instances(hass: HomeAssistantType) -> None: async def test_setup_config_entry_dynamic_instances(hass: HomeAssistant) -> None:
"""Test dynamic changes in the instance configuration.""" """Test dynamic changes in the instance configuration."""
registry = er.async_get(hass) registry = er.async_get(hass)
@ -241,7 +241,7 @@ async def test_setup_config_entry_dynamic_instances(hass: HomeAssistantType) ->
assert hass.states.get(TEST_ENTITY_ID_3) is not None assert hass.states.get(TEST_ENTITY_ID_3) is not None
async def test_light_basic_properies(hass: HomeAssistantType) -> None: async def test_light_basic_properies(hass: HomeAssistant) -> None:
"""Test the basic properties.""" """Test the basic properties."""
client = create_mock_client() client = create_mock_client()
await setup_test_config_entry(hass, hyperion_client=client) await setup_test_config_entry(hass, hyperion_client=client)
@ -262,7 +262,7 @@ async def test_light_basic_properies(hass: HomeAssistantType) -> None:
) )
async def test_light_async_turn_on(hass: HomeAssistantType) -> None: async def test_light_async_turn_on(hass: HomeAssistant) -> None:
"""Test turning the light on.""" """Test turning the light on."""
client = create_mock_client() client = create_mock_client()
await setup_test_config_entry(hass, hyperion_client=client) await setup_test_config_entry(hass, hyperion_client=client)
@ -507,7 +507,7 @@ async def test_light_async_turn_on(hass: HomeAssistantType) -> None:
async def test_light_async_turn_on_fail_async_send_set_component( async def test_light_async_turn_on_fail_async_send_set_component(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test set_component failure when turning the light on.""" """Test set_component failure when turning the light on."""
client = create_mock_client() client = create_mock_client()
@ -523,7 +523,7 @@ async def test_light_async_turn_on_fail_async_send_set_component(
async def test_light_async_turn_on_fail_async_send_set_component_source( async def test_light_async_turn_on_fail_async_send_set_component_source(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test async_send_set_component failure when selecting the source.""" """Test async_send_set_component failure when selecting the source."""
client = create_mock_client() client = create_mock_client()
@ -546,7 +546,7 @@ async def test_light_async_turn_on_fail_async_send_set_component_source(
async def test_light_async_turn_on_fail_async_send_clear_source( async def test_light_async_turn_on_fail_async_send_clear_source(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test async_send_clear failure when turning the light on.""" """Test async_send_clear failure when turning the light on."""
client = create_mock_client() client = create_mock_client()
@ -566,7 +566,7 @@ async def test_light_async_turn_on_fail_async_send_clear_source(
async def test_light_async_turn_on_fail_async_send_clear_effect( async def test_light_async_turn_on_fail_async_send_clear_effect(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test async_send_clear failure when turning on an effect.""" """Test async_send_clear failure when turning on an effect."""
client = create_mock_client() client = create_mock_client()
@ -583,7 +583,7 @@ async def test_light_async_turn_on_fail_async_send_clear_effect(
async def test_light_async_turn_on_fail_async_send_set_effect( async def test_light_async_turn_on_fail_async_send_set_effect(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test async_send_set_effect failure when turning on the light.""" """Test async_send_set_effect failure when turning on the light."""
client = create_mock_client() client = create_mock_client()
@ -603,7 +603,7 @@ async def test_light_async_turn_on_fail_async_send_set_effect(
async def test_light_async_turn_on_fail_async_send_set_color( async def test_light_async_turn_on_fail_async_send_set_color(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test async_send_set_color failure when turning on the light.""" """Test async_send_set_color failure when turning on the light."""
client = create_mock_client() client = create_mock_client()
@ -623,7 +623,7 @@ async def test_light_async_turn_on_fail_async_send_set_color(
async def test_light_async_turn_off_fail_async_send_set_component( async def test_light_async_turn_off_fail_async_send_set_component(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test async_send_set_component failure when turning off the light.""" """Test async_send_set_component failure when turning off the light."""
client = create_mock_client() client = create_mock_client()
@ -642,7 +642,7 @@ async def test_light_async_turn_off_fail_async_send_set_component(
async def test_priority_light_async_turn_off_fail_async_send_clear( async def test_priority_light_async_turn_off_fail_async_send_clear(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test async_send_clear failure when turning off a priority light.""" """Test async_send_clear failure when turning off a priority light."""
client = create_mock_client() client = create_mock_client()
@ -662,7 +662,7 @@ async def test_priority_light_async_turn_off_fail_async_send_clear(
assert client.method_calls[-1] == call.async_send_clear(priority=180) assert client.method_calls[-1] == call.async_send_clear(priority=180)
async def test_light_async_turn_off(hass: HomeAssistantType) -> None: async def test_light_async_turn_off(hass: HomeAssistant) -> None:
"""Test turning the light off.""" """Test turning the light off."""
client = create_mock_client() client = create_mock_client()
await setup_test_config_entry(hass, hyperion_client=client) await setup_test_config_entry(hass, hyperion_client=client)
@ -705,7 +705,7 @@ async def test_light_async_turn_off(hass: HomeAssistantType) -> None:
async def test_light_async_updates_from_hyperion_client( async def test_light_async_updates_from_hyperion_client(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test receiving a variety of Hyperion client callbacks.""" """Test receiving a variety of Hyperion client callbacks."""
client = create_mock_client() client = create_mock_client()
@ -825,7 +825,7 @@ async def test_light_async_updates_from_hyperion_client(
assert entity_state.state == "on" assert entity_state.state == "on"
async def test_full_state_loaded_on_start(hass: HomeAssistantType) -> None: async def test_full_state_loaded_on_start(hass: HomeAssistant) -> None:
"""Test receiving a variety of Hyperion client callbacks.""" """Test receiving a variety of Hyperion client callbacks."""
client = create_mock_client() client = create_mock_client()
@ -848,7 +848,7 @@ async def test_full_state_loaded_on_start(hass: HomeAssistantType) -> None:
assert entity_state.attributes["hs_color"] == (180.0, 100.0) assert entity_state.attributes["hs_color"] == (180.0, 100.0)
async def test_unload_entry(hass: HomeAssistantType) -> None: async def test_unload_entry(hass: HomeAssistant) -> None:
"""Test unload.""" """Test unload."""
client = create_mock_client() client = create_mock_client()
await setup_test_config_entry(hass, hyperion_client=client) await setup_test_config_entry(hass, hyperion_client=client)
@ -862,7 +862,7 @@ async def test_unload_entry(hass: HomeAssistantType) -> None:
assert client.async_client_disconnect.call_count == 2 assert client.async_client_disconnect.call_count == 2
async def test_version_log_warning(caplog, hass: HomeAssistantType) -> None: # type: ignore[no-untyped-def] async def test_version_log_warning(caplog, hass: HomeAssistant) -> None: # type: ignore[no-untyped-def]
"""Test warning on old version.""" """Test warning on old version."""
client = create_mock_client() client = create_mock_client()
client.async_sysinfo_version = AsyncMock(return_value="2.0.0-alpha.7") client.async_sysinfo_version = AsyncMock(return_value="2.0.0-alpha.7")
@ -871,7 +871,7 @@ async def test_version_log_warning(caplog, hass: HomeAssistantType) -> None: #
assert "Please consider upgrading" in caplog.text assert "Please consider upgrading" in caplog.text
async def test_version_no_log_warning(caplog, hass: HomeAssistantType) -> None: # type: ignore[no-untyped-def] async def test_version_no_log_warning(caplog, hass: HomeAssistant) -> None: # type: ignore[no-untyped-def]
"""Test no warning on acceptable version.""" """Test no warning on acceptable version."""
client = create_mock_client() client = create_mock_client()
client.async_sysinfo_version = AsyncMock(return_value="2.0.0-alpha.9") client.async_sysinfo_version = AsyncMock(return_value="2.0.0-alpha.9")
@ -880,7 +880,7 @@ async def test_version_no_log_warning(caplog, hass: HomeAssistantType) -> None:
assert "Please consider upgrading" not in caplog.text assert "Please consider upgrading" not in caplog.text
async def test_setup_entry_no_token_reauth(hass: HomeAssistantType) -> None: async def test_setup_entry_no_token_reauth(hass: HomeAssistant) -> None:
"""Verify a reauth flow when auth is required but no token provided.""" """Verify a reauth flow when auth is required but no token provided."""
client = create_mock_client() client = create_mock_client()
config_entry = add_test_config_entry(hass) config_entry = add_test_config_entry(hass)
@ -903,7 +903,7 @@ async def test_setup_entry_no_token_reauth(hass: HomeAssistantType) -> None:
assert config_entry.state == ENTRY_STATE_SETUP_ERROR assert config_entry.state == ENTRY_STATE_SETUP_ERROR
async def test_setup_entry_bad_token_reauth(hass: HomeAssistantType) -> None: async def test_setup_entry_bad_token_reauth(hass: HomeAssistant) -> None:
"""Verify a reauth flow when a bad token is provided.""" """Verify a reauth flow when a bad token is provided."""
client = create_mock_client() client = create_mock_client()
config_entry = add_test_config_entry( config_entry = add_test_config_entry(
@ -932,7 +932,7 @@ async def test_setup_entry_bad_token_reauth(hass: HomeAssistantType) -> None:
async def test_priority_light_async_updates( async def test_priority_light_async_updates(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test receiving a variety of Hyperion client callbacks to a HyperionPriorityLight.""" """Test receiving a variety of Hyperion client callbacks to a HyperionPriorityLight."""
priority_template = { priority_template = {
@ -1094,7 +1094,7 @@ async def test_priority_light_async_updates(
async def test_priority_light_async_updates_off_sets_black( async def test_priority_light_async_updates_off_sets_black(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test turning the HyperionPriorityLight off.""" """Test turning the HyperionPriorityLight off."""
client = create_mock_client() client = create_mock_client()
@ -1142,7 +1142,7 @@ async def test_priority_light_async_updates_off_sets_black(
async def test_priority_light_prior_color_preserved_after_black( async def test_priority_light_prior_color_preserved_after_black(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test that color is preserved in an on->off->on cycle for a HyperionPriorityLight. """Test that color is preserved in an on->off->on cycle for a HyperionPriorityLight.
@ -1265,7 +1265,7 @@ async def test_priority_light_prior_color_preserved_after_black(
assert entity_state.attributes["hs_color"] == hs_color assert entity_state.attributes["hs_color"] == hs_color
async def test_priority_light_has_no_external_sources(hass: HomeAssistantType) -> None: async def test_priority_light_has_no_external_sources(hass: HomeAssistant) -> None:
"""Ensure a HyperionPriorityLight does not list external sources.""" """Ensure a HyperionPriorityLight does not list external sources."""
client = create_mock_client() client = create_mock_client()
client.priorities = [] client.priorities = []
@ -1283,7 +1283,7 @@ async def test_priority_light_has_no_external_sources(hass: HomeAssistantType) -
assert entity_state.attributes["effect_list"] == [hyperion_light.KEY_EFFECT_SOLID] assert entity_state.attributes["effect_list"] == [hyperion_light.KEY_EFFECT_SOLID]
async def test_light_option_effect_hide_list(hass: HomeAssistantType) -> None: async def test_light_option_effect_hide_list(hass: HomeAssistant) -> None:
"""Test the effect_hide_list option.""" """Test the effect_hide_list option."""
client = create_mock_client() client = create_mock_client()
client.effects = [{const.KEY_NAME: "One"}, {const.KEY_NAME: "Two"}] client.effects = [{const.KEY_NAME: "One"}, {const.KEY_NAME: "Two"}]
@ -1304,7 +1304,7 @@ async def test_light_option_effect_hide_list(hass: HomeAssistantType) -> None:
] ]
async def test_device_info(hass: HomeAssistantType) -> None: async def test_device_info(hass: HomeAssistant) -> None:
"""Verify device information includes expected details.""" """Verify device information includes expected details."""
client = create_mock_client() client = create_mock_client()
@ -1336,7 +1336,7 @@ async def test_device_info(hass: HomeAssistantType) -> None:
assert TEST_ENTITY_ID_1 in entities_from_device assert TEST_ENTITY_ID_1 in entities_from_device
async def test_lights_can_be_enabled(hass: HomeAssistantType) -> None: async def test_lights_can_be_enabled(hass: HomeAssistant) -> None:
"""Verify lights can be enabled.""" """Verify lights can be enabled."""
client = create_mock_client() client = create_mock_client()
await setup_test_config_entry(hass, hyperion_client=client) await setup_test_config_entry(hass, hyperion_client=client)
@ -1369,7 +1369,7 @@ async def test_lights_can_be_enabled(hass: HomeAssistantType) -> None:
assert entity_state assert entity_state
async def test_deprecated_effect_names(caplog, hass: HomeAssistantType) -> None: # type: ignore[no-untyped-def] async def test_deprecated_effect_names(caplog, hass: HomeAssistant) -> None: # type: ignore[no-untyped-def]
"""Test deprecated effects function and issue a warning.""" """Test deprecated effects function and issue a warning."""
client = create_mock_client() client = create_mock_client()
client.async_send_clear = AsyncMock(return_value=True) client.async_send_clear = AsyncMock(return_value=True)
@ -1401,7 +1401,7 @@ async def test_deprecated_effect_names(caplog, hass: HomeAssistantType) -> None:
async def test_deprecated_effect_names_not_in_effect_list( async def test_deprecated_effect_names_not_in_effect_list(
hass: HomeAssistantType, hass: HomeAssistant,
) -> None: ) -> None:
"""Test deprecated effects are not in shown effect list.""" """Test deprecated effects are not in shown effect list."""
await setup_test_config_entry(hass) await setup_test_config_entry(hass)

View File

@ -20,8 +20,8 @@ from homeassistant.components.hyperion.const import (
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
from homeassistant.config_entries import RELOAD_AFTER_UPDATE_DELAY from homeassistant.config_entries import RELOAD_AFTER_UPDATE_DELAY
from homeassistant.const import ATTR_ENTITY_ID, SERVICE_TURN_OFF, SERVICE_TURN_ON from homeassistant.const import ATTR_ENTITY_ID, SERVICE_TURN_OFF, SERVICE_TURN_ON
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.typing import HomeAssistantType
from homeassistant.util import dt, slugify from homeassistant.util import dt, slugify
from . import ( from . import (
@ -52,7 +52,7 @@ TEST_SWITCH_COMPONENT_BASE_ENTITY_ID = "switch.test_instance_1_component"
TEST_SWITCH_COMPONENT_ALL_ENTITY_ID = f"{TEST_SWITCH_COMPONENT_BASE_ENTITY_ID}_all" TEST_SWITCH_COMPONENT_ALL_ENTITY_ID = f"{TEST_SWITCH_COMPONENT_BASE_ENTITY_ID}_all"
async def test_switch_turn_on_off(hass: HomeAssistantType) -> None: async def test_switch_turn_on_off(hass: HomeAssistant) -> None:
"""Test turning the light on.""" """Test turning the light on."""
client = create_mock_client() client = create_mock_client()
client.async_send_set_component = AsyncMock(return_value=True) client.async_send_set_component = AsyncMock(return_value=True)
@ -121,7 +121,7 @@ async def test_switch_turn_on_off(hass: HomeAssistantType) -> None:
assert entity_state.state == "on" assert entity_state.state == "on"
async def test_switch_has_correct_entities(hass: HomeAssistantType) -> None: async def test_switch_has_correct_entities(hass: HomeAssistant) -> None:
"""Test that the correct switch entities are created.""" """Test that the correct switch entities are created."""
client = create_mock_client() client = create_mock_client()
client.components = TEST_COMPONENTS client.components = TEST_COMPONENTS
@ -144,7 +144,7 @@ async def test_switch_has_correct_entities(hass: HomeAssistantType) -> None:
assert entity_state, f"Couldn't find entity: {entity_id}" assert entity_state, f"Couldn't find entity: {entity_id}"
async def test_device_info(hass: HomeAssistantType) -> None: async def test_device_info(hass: HomeAssistant) -> None:
"""Verify device information includes expected details.""" """Verify device information includes expected details."""
client = create_mock_client() client = create_mock_client()
client.components = TEST_COMPONENTS client.components = TEST_COMPONENTS
@ -184,7 +184,7 @@ async def test_device_info(hass: HomeAssistantType) -> None:
assert entity_id in entities_from_device assert entity_id in entities_from_device
async def test_switches_can_be_enabled(hass: HomeAssistantType) -> None: async def test_switches_can_be_enabled(hass: HomeAssistant) -> None:
"""Verify switches can be enabled.""" """Verify switches can be enabled."""
client = create_mock_client() client = create_mock_client()
client.components = TEST_COMPONENTS client.components = TEST_COMPONENTS

View File

@ -20,7 +20,7 @@ from homeassistant.components.icloud.const import (
) )
from homeassistant.config_entries import SOURCE_IMPORT, SOURCE_REAUTH, SOURCE_USER from homeassistant.config_entries import SOURCE_IMPORT, SOURCE_REAUTH, SOURCE_USER
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.helpers.typing import HomeAssistantType from homeassistant.core import HomeAssistant
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
@ -157,7 +157,7 @@ def mock_controller_service_validate_verification_code_failed():
yield service_mock yield service_mock
async def test_user(hass: HomeAssistantType, service: MagicMock): async def test_user(hass: HomeAssistant, service: MagicMock):
"""Test user config.""" """Test user config."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": SOURCE_USER}, data=None DOMAIN, context={"source": SOURCE_USER}, data=None
@ -175,9 +175,7 @@ async def test_user(hass: HomeAssistantType, service: MagicMock):
assert result["step_id"] == CONF_TRUSTED_DEVICE assert result["step_id"] == CONF_TRUSTED_DEVICE
async def test_user_with_cookie( async def test_user_with_cookie(hass: HomeAssistant, service_authenticated: MagicMock):
hass: HomeAssistantType, service_authenticated: MagicMock
):
"""Test user config with presence of a cookie.""" """Test user config with presence of a cookie."""
# test with all provided # test with all provided
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
@ -199,7 +197,7 @@ async def test_user_with_cookie(
assert result["data"][CONF_GPS_ACCURACY_THRESHOLD] == DEFAULT_GPS_ACCURACY_THRESHOLD assert result["data"][CONF_GPS_ACCURACY_THRESHOLD] == DEFAULT_GPS_ACCURACY_THRESHOLD
async def test_import(hass: HomeAssistantType, service: MagicMock): async def test_import(hass: HomeAssistant, service: MagicMock):
"""Test import step.""" """Test import step."""
# import with required # import with required
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
@ -227,7 +225,7 @@ async def test_import(hass: HomeAssistantType, service: MagicMock):
async def test_import_with_cookie( async def test_import_with_cookie(
hass: HomeAssistantType, service_authenticated: MagicMock hass: HomeAssistant, service_authenticated: MagicMock
): ):
"""Test import step with presence of a cookie.""" """Test import step with presence of a cookie."""
# import with required # import with required
@ -268,7 +266,7 @@ async def test_import_with_cookie(
async def test_two_accounts_setup( async def test_two_accounts_setup(
hass: HomeAssistantType, service_authenticated: MagicMock hass: HomeAssistant, service_authenticated: MagicMock
): ):
"""Test to setup two accounts.""" """Test to setup two accounts."""
MockConfigEntry( MockConfigEntry(
@ -293,7 +291,7 @@ async def test_two_accounts_setup(
assert result["data"][CONF_GPS_ACCURACY_THRESHOLD] == DEFAULT_GPS_ACCURACY_THRESHOLD assert result["data"][CONF_GPS_ACCURACY_THRESHOLD] == DEFAULT_GPS_ACCURACY_THRESHOLD
async def test_already_setup(hass: HomeAssistantType): async def test_already_setup(hass: HomeAssistant):
"""Test we abort if the account is already setup.""" """Test we abort if the account is already setup."""
MockConfigEntry( MockConfigEntry(
domain=DOMAIN, domain=DOMAIN,
@ -320,7 +318,7 @@ async def test_already_setup(hass: HomeAssistantType):
assert result["reason"] == "already_configured" assert result["reason"] == "already_configured"
async def test_login_failed(hass: HomeAssistantType): async def test_login_failed(hass: HomeAssistant):
"""Test when we have errors during login.""" """Test when we have errors during login."""
with patch( with patch(
"homeassistant.components.icloud.config_flow.PyiCloudService.authenticate", "homeassistant.components.icloud.config_flow.PyiCloudService.authenticate",
@ -336,7 +334,7 @@ async def test_login_failed(hass: HomeAssistantType):
async def test_no_device( async def test_no_device(
hass: HomeAssistantType, service_authenticated_no_device: MagicMock hass: HomeAssistant, service_authenticated_no_device: MagicMock
): ):
"""Test when we have no devices.""" """Test when we have no devices."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
@ -348,7 +346,7 @@ async def test_no_device(
assert result["reason"] == "no_device" assert result["reason"] == "no_device"
async def test_trusted_device(hass: HomeAssistantType, service: MagicMock): async def test_trusted_device(hass: HomeAssistant, service: MagicMock):
"""Test trusted_device step.""" """Test trusted_device step."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
@ -361,7 +359,7 @@ async def test_trusted_device(hass: HomeAssistantType, service: MagicMock):
assert result["step_id"] == CONF_TRUSTED_DEVICE assert result["step_id"] == CONF_TRUSTED_DEVICE
async def test_trusted_device_success(hass: HomeAssistantType, service: MagicMock): async def test_trusted_device_success(hass: HomeAssistant, service: MagicMock):
"""Test trusted_device step success.""" """Test trusted_device step success."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
@ -377,7 +375,7 @@ async def test_trusted_device_success(hass: HomeAssistantType, service: MagicMoc
async def test_send_verification_code_failed( async def test_send_verification_code_failed(
hass: HomeAssistantType, service_send_verification_code_failed: MagicMock hass: HomeAssistant, service_send_verification_code_failed: MagicMock
): ):
"""Test when we have errors during send_verification_code.""" """Test when we have errors during send_verification_code."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
@ -394,7 +392,7 @@ async def test_send_verification_code_failed(
assert result["errors"] == {CONF_TRUSTED_DEVICE: "send_verification_code"} assert result["errors"] == {CONF_TRUSTED_DEVICE: "send_verification_code"}
async def test_verification_code(hass: HomeAssistantType, service: MagicMock): async def test_verification_code(hass: HomeAssistant, service: MagicMock):
"""Test verification_code step.""" """Test verification_code step."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
@ -410,7 +408,7 @@ async def test_verification_code(hass: HomeAssistantType, service: MagicMock):
assert result["step_id"] == CONF_VERIFICATION_CODE assert result["step_id"] == CONF_VERIFICATION_CODE
async def test_verification_code_success(hass: HomeAssistantType, service: MagicMock): async def test_verification_code_success(hass: HomeAssistant, service: MagicMock):
"""Test verification_code step success.""" """Test verification_code step success."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
@ -436,7 +434,7 @@ async def test_verification_code_success(hass: HomeAssistantType, service: Magic
async def test_validate_verification_code_failed( async def test_validate_verification_code_failed(
hass: HomeAssistantType, service_validate_verification_code_failed: MagicMock hass: HomeAssistant, service_validate_verification_code_failed: MagicMock
): ):
"""Test when we have errors during validate_verification_code.""" """Test when we have errors during validate_verification_code."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
@ -456,7 +454,7 @@ async def test_validate_verification_code_failed(
assert result["errors"] == {"base": "validate_verification_code"} assert result["errors"] == {"base": "validate_verification_code"}
async def test_2fa_code_success(hass: HomeAssistantType, service_2fa: MagicMock): async def test_2fa_code_success(hass: HomeAssistant, service_2fa: MagicMock):
"""Test 2fa step success.""" """Test 2fa step success."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
@ -481,7 +479,7 @@ async def test_2fa_code_success(hass: HomeAssistantType, service_2fa: MagicMock)
async def test_validate_2fa_code_failed( async def test_validate_2fa_code_failed(
hass: HomeAssistantType, service_validate_2fa_code_failed: MagicMock hass: HomeAssistant, service_validate_2fa_code_failed: MagicMock
): ):
"""Test when we have errors during validate_verification_code.""" """Test when we have errors during validate_verification_code."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
@ -499,9 +497,7 @@ async def test_validate_2fa_code_failed(
assert result["errors"] == {"base": "validate_verification_code"} assert result["errors"] == {"base": "validate_verification_code"}
async def test_password_update( async def test_password_update(hass: HomeAssistant, service_authenticated: MagicMock):
hass: HomeAssistantType, service_authenticated: MagicMock
):
"""Test that password reauthentication works successfully.""" """Test that password reauthentication works successfully."""
config_entry = MockConfigEntry( config_entry = MockConfigEntry(
domain=DOMAIN, data=MOCK_CONFIG, entry_id="test", unique_id=USERNAME domain=DOMAIN, data=MOCK_CONFIG, entry_id="test", unique_id=USERNAME
@ -525,7 +521,7 @@ async def test_password_update(
assert config_entry.data[CONF_PASSWORD] == PASSWORD_2 assert config_entry.data[CONF_PASSWORD] == PASSWORD_2
async def test_password_update_wrong_password(hass: HomeAssistantType): async def test_password_update_wrong_password(hass: HomeAssistant):
"""Test that during password reauthentication wrong password returns correct error.""" """Test that during password reauthentication wrong password returns correct error."""
config_entry = MockConfigEntry( config_entry = MockConfigEntry(
domain=DOMAIN, data=MOCK_CONFIG, entry_id="test", unique_id=USERNAME domain=DOMAIN, data=MOCK_CONFIG, entry_id="test", unique_id=USERNAME