diff --git a/homeassistant/components/ps4/__init__.py b/homeassistant/components/ps4/__init__.py index 11d271be543..51583b5f4bc 100644 --- a/homeassistant/components/ps4/__init__.py +++ b/homeassistant/components/ps4/__init__.py @@ -18,10 +18,9 @@ from homeassistant.const import ( CONF_REGION, CONF_TOKEN, ) -from homeassistant.core import split_entity_id +from homeassistant.core import HomeAssistant, split_entity_id from homeassistant.exceptions import HomeAssistantError from homeassistant.helpers import config_validation as cv, entity_registry -from homeassistant.helpers.typing import HomeAssistantType from homeassistant.util import location from homeassistant.util.json import load_json, save_json @@ -157,7 +156,7 @@ def format_unique_id(creds, mac_address): return f"{mac_address}_{suffix}" -def load_games(hass: HomeAssistantType, unique_id: str) -> dict: +def load_games(hass: HomeAssistant, unique_id: str) -> dict: """Load games for sources.""" g_file = hass.config.path(GAMES_FILE.format(unique_id)) try: @@ -176,7 +175,7 @@ def load_games(hass: HomeAssistantType, unique_id: str) -> dict: return games -def save_games(hass: HomeAssistantType, games: dict, unique_id: str): +def save_games(hass: HomeAssistant, games: dict, unique_id: str): """Save games to file.""" g_file = hass.config.path(GAMES_FILE.format(unique_id)) try: @@ -185,7 +184,7 @@ def save_games(hass: HomeAssistantType, games: dict, unique_id: str): _LOGGER.error("Could not save game list, %s", error) -def _reformat_data(hass: HomeAssistantType, games: dict, unique_id: str) -> dict: +def _reformat_data(hass: HomeAssistant, games: dict, unique_id: str) -> dict: """Reformat data to correct format.""" data_reformatted = False @@ -208,7 +207,7 @@ def _reformat_data(hass: HomeAssistantType, games: dict, unique_id: str) -> dict return games -def service_handle(hass: HomeAssistantType): +def service_handle(hass: HomeAssistant): """Handle for services.""" async def async_service_command(call): diff --git a/homeassistant/components/recorder/util.py b/homeassistant/components/recorder/util.py index e4e691bec91..9f99dc2bf45 100644 --- a/homeassistant/components/recorder/util.py +++ b/homeassistant/components/recorder/util.py @@ -11,7 +11,7 @@ import time from sqlalchemy.exc import OperationalError, SQLAlchemyError from sqlalchemy.orm.session import Session -from homeassistant.helpers.typing import HomeAssistantType +from homeassistant.core import HomeAssistant import homeassistant.util.dt as dt_util from .const import DATA_INSTANCE, SQLITE_URL_PREFIX @@ -37,7 +37,7 @@ MAX_RESTART_TIME = timedelta(minutes=10) @contextmanager def session_scope( - *, hass: HomeAssistantType | None = None, session: Session | None = None + *, hass: HomeAssistant | None = None, session: Session | None = None ) -> Generator[Session, None, None]: """Provide a transactional scope around a series of operations.""" if session is None and hass is not None: diff --git a/homeassistant/components/remote/__init__.py b/homeassistant/components/remote/__init__.py index 94c54dd323d..fef0da4dae6 100644 --- a/homeassistant/components/remote/__init__.py +++ b/homeassistant/components/remote/__init__.py @@ -17,6 +17,7 @@ from homeassistant.const import ( SERVICE_TURN_ON, STATE_ON, ) +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv from homeassistant.helpers.config_validation import ( # noqa: F401 PLATFORM_SCHEMA, @@ -25,7 +26,7 @@ from homeassistant.helpers.config_validation import ( # noqa: F401 ) from homeassistant.helpers.entity import ToggleEntity from homeassistant.helpers.entity_component import EntityComponent -from homeassistant.helpers.typing import ConfigType, HomeAssistantType +from homeassistant.helpers.typing import ConfigType from homeassistant.loader import bind_hass # mypy: allow-untyped-calls, allow-untyped-defs, no-check-untyped-defs @@ -69,12 +70,12 @@ REMOTE_SERVICE_ACTIVITY_SCHEMA = make_entity_service_schema( @bind_hass -def is_on(hass: HomeAssistantType, entity_id: str) -> bool: +def is_on(hass: HomeAssistant, entity_id: str) -> bool: """Return if the remote is on based on the statemachine.""" return hass.states.is_state(entity_id, STATE_ON) -async def async_setup(hass: HomeAssistantType, config: ConfigType) -> bool: +async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: """Track states and offer events for remotes.""" component = hass.data[DOMAIN] = EntityComponent( _LOGGER, DOMAIN, hass, SCAN_INTERVAL @@ -131,12 +132,12 @@ async def async_setup(hass: HomeAssistantType, config: ConfigType) -> bool: return True -async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool: +async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: """Set up a config entry.""" return await cast(EntityComponent, hass.data[DOMAIN]).async_setup_entry(entry) -async def async_unload_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool: +async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: """Unload a config entry.""" return await cast(EntityComponent, hass.data[DOMAIN]).async_unload_entry(entry) diff --git a/homeassistant/components/remote/group.py b/homeassistant/components/remote/group.py index 1636054663d..234883ffd5a 100644 --- a/homeassistant/components/remote/group.py +++ b/homeassistant/components/remote/group.py @@ -3,13 +3,12 @@ from homeassistant.components.group import GroupIntegrationRegistry from homeassistant.const import STATE_OFF, STATE_ON -from homeassistant.core import callback -from homeassistant.helpers.typing import HomeAssistantType +from homeassistant.core import HomeAssistant, callback @callback def async_describe_on_off_states( - hass: HomeAssistantType, registry: GroupIntegrationRegistry + hass: HomeAssistant, registry: GroupIntegrationRegistry ) -> None: """Describe group on off states.""" registry.on_off_states({STATE_ON}, STATE_OFF) diff --git a/homeassistant/components/remote/reproduce_state.py b/homeassistant/components/remote/reproduce_state.py index 24f748d4a02..cc9685dee2f 100644 --- a/homeassistant/components/remote/reproduce_state.py +++ b/homeassistant/components/remote/reproduce_state.py @@ -13,8 +13,7 @@ from homeassistant.const import ( STATE_OFF, STATE_ON, ) -from homeassistant.core import Context, State -from homeassistant.helpers.typing import HomeAssistantType +from homeassistant.core import Context, HomeAssistant, State from . import DOMAIN @@ -24,7 +23,7 @@ VALID_STATES = {STATE_ON, STATE_OFF} async def _async_reproduce_state( - hass: HomeAssistantType, + hass: HomeAssistant, state: State, *, context: Context | None = None, @@ -60,7 +59,7 @@ async def _async_reproduce_state( async def async_reproduce_states( - hass: HomeAssistantType, + hass: HomeAssistant, states: Iterable[State], *, context: Context | None = None, diff --git a/homeassistant/components/roku/__init__.py b/homeassistant/components/roku/__init__.py index 3a12de51b06..d7bf3059374 100644 --- a/homeassistant/components/roku/__init__.py +++ b/homeassistant/components/roku/__init__.py @@ -13,9 +13,9 @@ from homeassistant.components.media_player import DOMAIN as MEDIA_PLAYER_DOMAIN from homeassistant.components.remote import DOMAIN as REMOTE_DOMAIN from homeassistant.config_entries import ConfigEntry from homeassistant.const import ATTR_NAME, CONF_HOST +from homeassistant.core import HomeAssistant from homeassistant.helpers import config_validation as cv from homeassistant.helpers.aiohttp_client import async_get_clientsession -from homeassistant.helpers.typing import HomeAssistantType from homeassistant.helpers.update_coordinator import ( CoordinatorEntity, DataUpdateCoordinator, @@ -39,7 +39,7 @@ SCAN_INTERVAL = timedelta(seconds=15) _LOGGER = logging.getLogger(__name__) -async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool: +async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: """Set up Roku from a config entry.""" hass.data.setdefault(DOMAIN, {}) coordinator = hass.data[DOMAIN].get(entry.entry_id) @@ -57,7 +57,7 @@ async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool 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_ok = all( await asyncio.gather( @@ -95,7 +95,7 @@ class RokuDataUpdateCoordinator(DataUpdateCoordinator[Device]): def __init__( self, - hass: HomeAssistantType, + hass: HomeAssistant, *, host: str, ): diff --git a/homeassistant/components/roku/config_flow.py b/homeassistant/components/roku/config_flow.py index 8424850fe6c..d10e22cd1bc 100644 --- a/homeassistant/components/roku/config_flow.py +++ b/homeassistant/components/roku/config_flow.py @@ -15,9 +15,8 @@ from homeassistant.components.ssdp import ( ) from homeassistant.config_entries import CONN_CLASS_LOCAL_POLL, ConfigFlow from homeassistant.const import CONF_HOST, CONF_NAME -from homeassistant.core import callback +from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.aiohttp_client import async_get_clientsession -from homeassistant.helpers.typing import HomeAssistantType from .const import DOMAIN @@ -29,7 +28,7 @@ ERROR_UNKNOWN = "unknown" _LOGGER = logging.getLogger(__name__) -async def validate_input(hass: HomeAssistantType, data: dict) -> dict: +async def validate_input(hass: HomeAssistant, data: dict) -> dict: """Validate the user input allows us to connect. Data has the keys from DATA_SCHEMA with values provided by the user. diff --git a/homeassistant/components/roku/remote.py b/homeassistant/components/roku/remote.py index da578667578..a4f35294fd5 100644 --- a/homeassistant/components/roku/remote.py +++ b/homeassistant/components/roku/remote.py @@ -5,14 +5,14 @@ from typing import Callable from homeassistant.components.remote import ATTR_NUM_REPEATS, RemoteEntity from homeassistant.config_entries import ConfigEntry -from homeassistant.helpers.typing import HomeAssistantType +from homeassistant.core import HomeAssistant from . import RokuDataUpdateCoordinator, RokuEntity, roku_exception_handler from .const import DOMAIN async def async_setup_entry( - hass: HomeAssistantType, + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: Callable[[list, bool], None], ) -> bool: diff --git a/homeassistant/components/ruckus_unleashed/device_tracker.py b/homeassistant/components/ruckus_unleashed/device_tracker.py index 90a848b663b..a5bc266f045 100644 --- a/homeassistant/components/ruckus_unleashed/device_tracker.py +++ b/homeassistant/components/ruckus_unleashed/device_tracker.py @@ -4,10 +4,9 @@ from __future__ import annotations from homeassistant.components.device_tracker import SOURCE_TYPE_ROUTER from homeassistant.components.device_tracker.config_entry import ScannerEntity from homeassistant.config_entries import ConfigEntry -from homeassistant.core import callback +from homeassistant.core import HomeAssistant, callback from homeassistant.helpers import entity_registry from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC -from homeassistant.helpers.typing import HomeAssistantType from homeassistant.helpers.update_coordinator import CoordinatorEntity from .const import ( @@ -22,7 +21,7 @@ from .const import ( async def async_setup_entry( - hass: HomeAssistantType, entry: ConfigEntry, async_add_entities + hass: HomeAssistant, entry: ConfigEntry, async_add_entities ) -> None: """Set up device tracker for Ruckus Unleashed component.""" coordinator = hass.data[DOMAIN][entry.entry_id][COORDINATOR] diff --git a/homeassistant/components/screenlogic/services.py b/homeassistant/components/screenlogic/services.py index 7ca2bb69129..31e35788f44 100644 --- a/homeassistant/components/screenlogic/services.py +++ b/homeassistant/components/screenlogic/services.py @@ -5,11 +5,10 @@ import logging from screenlogicpy import ScreenLogicError import voluptuous as vol -from homeassistant.core import ServiceCall, callback +from homeassistant.core import HomeAssistant, ServiceCall, callback from homeassistant.exceptions import HomeAssistantError import homeassistant.helpers.config_validation as cv from homeassistant.helpers.service import async_extract_config_entry_ids -from homeassistant.helpers.typing import HomeAssistantType from .const import ( ATTR_COLOR_MODE, @@ -28,7 +27,7 @@ SET_COLOR_MODE_SCHEMA = cv.make_entity_service_schema( @callback -def async_load_screenlogic_services(hass: HomeAssistantType): +def async_load_screenlogic_services(hass: HomeAssistant): """Set up services for the ScreenLogic integration.""" if hass.services.has_service(DOMAIN, SERVICE_SET_COLOR_MODE): # Integration-level services have already been added. Return. @@ -76,7 +75,7 @@ def async_load_screenlogic_services(hass: HomeAssistantType): @callback -def async_unload_screenlogic_services(hass: HomeAssistantType): +def async_unload_screenlogic_services(hass: HomeAssistant): """Unload services for the ScreenLogic integration.""" if hass.data[DOMAIN]: # There is still another config entry for this domain, don't remove services. diff --git a/homeassistant/components/shell_command/__init__.py b/homeassistant/components/shell_command/__init__.py index 089dc36b1a8..a86bf5b3566 100644 --- a/homeassistant/components/shell_command/__init__.py +++ b/homeassistant/components/shell_command/__init__.py @@ -6,10 +6,10 @@ import shlex import voluptuous as vol -from homeassistant.core import ServiceCall +from homeassistant.core import HomeAssistant, ServiceCall from homeassistant.exceptions import TemplateError from homeassistant.helpers import config_validation as cv, template -from homeassistant.helpers.typing import ConfigType, HomeAssistantType +from homeassistant.helpers.typing import ConfigType DOMAIN = "shell_command" @@ -22,7 +22,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 the shell_command component.""" conf = config.get(DOMAIN, {}) diff --git a/homeassistant/components/slack/notify.py b/homeassistant/components/slack/notify.py index f1e293773bd..c2ca834b565 100644 --- a/homeassistant/components/slack/notify.py +++ b/homeassistant/components/slack/notify.py @@ -21,14 +21,10 @@ from homeassistant.components.notify import ( BaseNotificationService, ) from homeassistant.const import ATTR_ICON, CONF_API_KEY, CONF_ICON, CONF_USERNAME -from homeassistant.core import callback +from homeassistant.core import HomeAssistant, callback from homeassistant.helpers import aiohttp_client, config_validation as cv import homeassistant.helpers.template as template -from homeassistant.helpers.typing import ( - ConfigType, - DiscoveryInfoType, - HomeAssistantType, -) +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType _LOGGER = logging.getLogger(__name__) @@ -109,7 +105,7 @@ class MessageT(TypedDict, total=False): async def async_get_service( - hass: HomeAssistantType, + hass: HomeAssistant, config: ConfigType, discovery_info: DiscoveryInfoType | None = None, ) -> SlackNotificationService | None: @@ -152,7 +148,7 @@ def _async_sanitize_channel_names(channel_list: list[str]) -> list[str]: @callback -def _async_templatize_blocks(hass: HomeAssistantType, value: Any) -> Any: +def _async_templatize_blocks(hass: HomeAssistant, value: Any) -> Any: """Recursive template creator helper function.""" if isinstance(value, list): return [_async_templatize_blocks(hass, item) for item in value] @@ -170,7 +166,7 @@ class SlackNotificationService(BaseNotificationService): def __init__( self, - hass: HomeAssistantType, + hass: HomeAssistant, client: WebClient, default_channel: str, username: str | None, diff --git a/homeassistant/components/smarthab/__init__.py b/homeassistant/components/smarthab/__init__.py index c259ef71aab..ba6e7a5e5e7 100644 --- a/homeassistant/components/smarthab/__init__.py +++ b/homeassistant/components/smarthab/__init__.py @@ -7,9 +7,9 @@ import voluptuous as vol from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry from homeassistant.const import CONF_EMAIL, CONF_PASSWORD +from homeassistant.core import HomeAssistant from homeassistant.exceptions import ConfigEntryNotReady import homeassistant.helpers.config_validation as cv -from homeassistant.helpers.typing import HomeAssistantType DOMAIN = "smarthab" DATA_HUB = "hub" @@ -50,7 +50,7 @@ async def async_setup(hass, config) -> bool: return True -async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry): +async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry): """Set up config entry for SmartHab integration.""" # Assign configuration variables @@ -77,7 +77,7 @@ async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry): return True -async def async_unload_entry(hass: HomeAssistantType, entry: ConfigEntry): +async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry): """Unload config entry from SmartHab integration.""" result = all( diff --git a/homeassistant/components/smartthings/__init__.py b/homeassistant/components/smartthings/__init__.py index d36739c9551..456857efc9b 100644 --- a/homeassistant/components/smartthings/__init__.py +++ b/homeassistant/components/smartthings/__init__.py @@ -18,6 +18,7 @@ from homeassistant.const import ( HTTP_FORBIDDEN, HTTP_UNAUTHORIZED, ) +from homeassistant.core import HomeAssistant from homeassistant.exceptions import ConfigEntryNotReady from homeassistant.helpers.aiohttp_client import async_get_clientsession from homeassistant.helpers.dispatcher import ( @@ -26,7 +27,7 @@ from homeassistant.helpers.dispatcher import ( ) from homeassistant.helpers.entity import Entity from homeassistant.helpers.event import async_track_time_interval -from homeassistant.helpers.typing import ConfigType, HomeAssistantType +from homeassistant.helpers.typing import ConfigType from .config_flow import SmartThingsFlowHandler # noqa: F401 from .const import ( @@ -55,13 +56,13 @@ from .smartapp import ( _LOGGER = logging.getLogger(__name__) -async def async_setup(hass: HomeAssistantType, config: ConfigType): +async def async_setup(hass: HomeAssistant, config: ConfigType): """Initialize the SmartThings platform.""" await setup_smartapp_endpoint(hass) return True -async def async_migrate_entry(hass: HomeAssistantType, entry: ConfigEntry): +async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry): """Handle migration of a previous version config entry. A config entry created under a previous version must go through the @@ -81,7 +82,7 @@ async def async_migrate_entry(hass: HomeAssistantType, entry: ConfigEntry): return False -async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry): +async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry): """Initialize config entry which represents an installed SmartApp.""" # For backwards compat if entry.unique_id is None: @@ -208,7 +209,7 @@ async def async_get_entry_scenes(entry: ConfigEntry, api): return [] -async def async_unload_entry(hass: HomeAssistantType, entry: ConfigEntry): +async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry): """Unload a config entry.""" broker = hass.data[DOMAIN][DATA_BROKERS].pop(entry.entry_id, None) if broker: @@ -221,7 +222,7 @@ async def async_unload_entry(hass: HomeAssistantType, entry: ConfigEntry): return all(await asyncio.gather(*tasks)) -async def async_remove_entry(hass: HomeAssistantType, entry: ConfigEntry) -> None: +async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None: """Perform clean-up when entry is being removed.""" api = SmartThings(async_get_clientsession(hass), entry.data[CONF_ACCESS_TOKEN]) @@ -270,7 +271,7 @@ class DeviceBroker: def __init__( self, - hass: HomeAssistantType, + hass: HomeAssistant, entry: ConfigEntry, token, smart_app, diff --git a/homeassistant/components/smartthings/smartapp.py b/homeassistant/components/smartthings/smartapp.py index 24d6e4ae18f..0225a17a62c 100644 --- a/homeassistant/components/smartthings/smartapp.py +++ b/homeassistant/components/smartthings/smartapp.py @@ -25,13 +25,13 @@ from pysmartthings import ( from homeassistant.components import webhook from homeassistant.const import CONF_WEBHOOK_ID +from homeassistant.core import HomeAssistant from homeassistant.helpers.aiohttp_client import async_get_clientsession from homeassistant.helpers.dispatcher import ( async_dispatcher_connect, async_dispatcher_send, ) from homeassistant.helpers.network import NoURLAvailableError, get_url -from homeassistant.helpers.typing import HomeAssistantType from .const import ( APP_NAME_PREFIX, @@ -60,7 +60,7 @@ def format_unique_id(app_id: str, location_id: str) -> str: return f"{app_id}_{location_id}" -async def find_app(hass: HomeAssistantType, api): +async def find_app(hass: HomeAssistant, api): """Find an existing SmartApp for this installation of hass.""" apps = await api.apps() for app in [app for app in apps if app.app_name.startswith(APP_NAME_PREFIX)]: @@ -92,7 +92,7 @@ async def validate_installed_app(api, installed_app_id: str): return installed_app -def validate_webhook_requirements(hass: HomeAssistantType) -> bool: +def validate_webhook_requirements(hass: HomeAssistant) -> bool: """Ensure Home Assistant is setup properly to receive webhooks.""" if hass.components.cloud.async_active_subscription(): return True @@ -101,7 +101,7 @@ def validate_webhook_requirements(hass: HomeAssistantType) -> bool: return get_webhook_url(hass).lower().startswith("https://") -def get_webhook_url(hass: HomeAssistantType) -> str: +def get_webhook_url(hass: HomeAssistant) -> str: """ Get the URL of the webhook. @@ -113,7 +113,7 @@ def get_webhook_url(hass: HomeAssistantType) -> str: return webhook.async_generate_url(hass, hass.data[DOMAIN][CONF_WEBHOOK_ID]) -def _get_app_template(hass: HomeAssistantType): +def _get_app_template(hass: HomeAssistant): try: endpoint = f"at {get_url(hass, allow_cloud=False, prefer_external=True)}" except NoURLAvailableError: @@ -135,7 +135,7 @@ def _get_app_template(hass: HomeAssistantType): } -async def create_app(hass: HomeAssistantType, api): +async def create_app(hass: HomeAssistant, api): """Create a SmartApp for this instance of hass.""" # Create app from template attributes template = _get_app_template(hass) @@ -163,7 +163,7 @@ async def create_app(hass: HomeAssistantType, api): return app, client -async def update_app(hass: HomeAssistantType, app): +async def update_app(hass: HomeAssistant, app): """Ensure the SmartApp is up-to-date and update if necessary.""" template = _get_app_template(hass) template.pop("app_name") # don't update this @@ -199,7 +199,7 @@ def setup_smartapp(hass, app): return smartapp -async def setup_smartapp_endpoint(hass: HomeAssistantType): +async def setup_smartapp_endpoint(hass: HomeAssistant): """ Configure the SmartApp webhook in hass. @@ -276,7 +276,7 @@ async def setup_smartapp_endpoint(hass: HomeAssistantType): ) -async def unload_smartapp_endpoint(hass: HomeAssistantType): +async def unload_smartapp_endpoint(hass: HomeAssistant): """Tear down the component configuration.""" if DOMAIN not in hass.data: return @@ -308,7 +308,7 @@ async def unload_smartapp_endpoint(hass: HomeAssistantType): async def smartapp_sync_subscriptions( - hass: HomeAssistantType, + hass: HomeAssistant, auth_token: str, location_id: str, installed_app_id: str, @@ -397,7 +397,7 @@ async def smartapp_sync_subscriptions( async def _continue_flow( - hass: HomeAssistantType, + hass: HomeAssistant, app_id: str, location_id: str, installed_app_id: str, @@ -429,7 +429,7 @@ async def _continue_flow( ) -async def smartapp_install(hass: HomeAssistantType, req, resp, app): +async def smartapp_install(hass: HomeAssistant, req, resp, app): """Handle a SmartApp installation and continue the config flow.""" await _continue_flow( hass, app.app_id, req.location_id, req.installed_app_id, req.refresh_token @@ -441,7 +441,7 @@ async def smartapp_install(hass: HomeAssistantType, req, resp, app): ) -async def smartapp_update(hass: HomeAssistantType, req, resp, app): +async def smartapp_update(hass: HomeAssistant, req, resp, app): """Handle a SmartApp update and either update the entry or continue the flow.""" entry = next( ( @@ -470,7 +470,7 @@ async def smartapp_update(hass: HomeAssistantType, req, resp, app): ) -async def smartapp_uninstall(hass: HomeAssistantType, req, resp, app): +async def smartapp_uninstall(hass: HomeAssistant, req, resp, app): """ Handle when a SmartApp is removed from a location by the user. @@ -496,7 +496,7 @@ async def smartapp_uninstall(hass: HomeAssistantType, req, resp, app): ) -async def smartapp_webhook(hass: HomeAssistantType, webhook_id: str, request): +async def smartapp_webhook(hass: HomeAssistant, webhook_id: str, request): """ Handle a smartapp lifecycle event callback from SmartThings. diff --git a/homeassistant/components/zha/core/store.py b/homeassistant/components/zha/core/store.py index c4bbca2567a..f1b2ee57aee 100644 --- a/homeassistant/components/zha/core/store.py +++ b/homeassistant/components/zha/core/store.py @@ -9,8 +9,7 @@ from typing import cast import attr -from homeassistant.core import callback -from homeassistant.helpers.typing import HomeAssistantType +from homeassistant.core import HomeAssistant, callback from homeassistant.loader import bind_hass from .typing import ZhaDeviceType @@ -35,9 +34,9 @@ class ZhaDeviceEntry: class ZhaStorage: """Class to hold a registry of zha devices.""" - def __init__(self, hass: HomeAssistantType) -> None: + def __init__(self, hass: HomeAssistant) -> None: """Initialize the zha device storage.""" - self.hass: HomeAssistantType = hass + self.hass: HomeAssistant = hass self.devices: MutableMapping[str, ZhaDeviceEntry] = {} self._store = hass.helpers.storage.Store(STORAGE_VERSION, STORAGE_KEY) @@ -130,7 +129,7 @@ class ZhaStorage: @bind_hass -async def async_get_registry(hass: HomeAssistantType) -> ZhaStorage: +async def async_get_registry(hass: HomeAssistant) -> ZhaStorage: """Return zha device storage instance.""" task = hass.data.get(DATA_REGISTRY) diff --git a/tests/components/recorder/common.py b/tests/components/recorder/common.py index 0bc4cfbfeb9..7414548c864 100644 --- a/tests/components/recorder/common.py +++ b/tests/components/recorder/common.py @@ -3,7 +3,7 @@ from datetime import timedelta from homeassistant import core as ha from homeassistant.components import recorder -from homeassistant.helpers.typing import HomeAssistantType +from homeassistant.core import HomeAssistant from homeassistant.util import dt as dt_util from tests.common import async_fire_time_changed, fire_time_changed @@ -11,7 +11,7 @@ from tests.common import async_fire_time_changed, fire_time_changed DEFAULT_PURGE_TASKS = 3 -def wait_recording_done(hass: HomeAssistantType) -> None: +def wait_recording_done(hass: HomeAssistant) -> None: """Block till recording is done.""" hass.block_till_done() trigger_db_commit(hass) @@ -20,12 +20,12 @@ def wait_recording_done(hass: HomeAssistantType) -> None: hass.block_till_done() -async def async_wait_recording_done_without_instance(hass: HomeAssistantType) -> None: +async def async_wait_recording_done_without_instance(hass: HomeAssistant) -> None: """Block till recording is done.""" await hass.loop.run_in_executor(None, wait_recording_done, hass) -def trigger_db_commit(hass: HomeAssistantType) -> None: +def trigger_db_commit(hass: HomeAssistant) -> None: """Force the recorder to commit.""" for _ in range(recorder.DEFAULT_COMMIT_INTERVAL): # We only commit on time change @@ -33,7 +33,7 @@ def trigger_db_commit(hass: HomeAssistantType) -> None: async def async_wait_recording_done( - hass: HomeAssistantType, + hass: HomeAssistant, instance: recorder.Recorder, ) -> None: """Async wait until recording is done.""" @@ -45,7 +45,7 @@ async def async_wait_recording_done( async def async_wait_purge_done( - hass: HomeAssistantType, instance: recorder.Recorder, max: int = None + hass: HomeAssistant, instance: recorder.Recorder, max: int = None ) -> None: """Wait for max number of purge events. @@ -61,14 +61,14 @@ async def async_wait_purge_done( @ha.callback -def async_trigger_db_commit(hass: HomeAssistantType) -> None: +def async_trigger_db_commit(hass: HomeAssistant) -> None: """Fore the recorder to commit. Async friendly.""" for _ in range(recorder.DEFAULT_COMMIT_INTERVAL): async_fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=1)) async def async_recorder_block_till_done( - hass: HomeAssistantType, + hass: HomeAssistant, instance: recorder.Recorder, ) -> None: """Non blocking version of recorder.block_till_done().""" diff --git a/tests/components/recorder/conftest.py b/tests/components/recorder/conftest.py index 6eadb1c62ed..6b8c61d4d7d 100644 --- a/tests/components/recorder/conftest.py +++ b/tests/components/recorder/conftest.py @@ -8,7 +8,8 @@ import pytest from homeassistant.components.recorder import Recorder from homeassistant.components.recorder.const import DATA_INSTANCE -from homeassistant.helpers.typing import ConfigType, HomeAssistantType +from homeassistant.core import HomeAssistant +from homeassistant.helpers.typing import ConfigType from .common import async_recorder_block_till_done @@ -45,7 +46,7 @@ async def async_setup_recorder_instance() -> AsyncGenerator[ """Yield callable to setup recorder instance.""" async def async_setup_recorder( - hass: HomeAssistantType, config: ConfigType | None = None + hass: HomeAssistant, config: ConfigType | None = None ) -> Recorder: """Setup and return recorder instance.""" # noqa: D401 await async_init_recorder_component(hass, config) diff --git a/tests/components/recorder/test_init.py b/tests/components/recorder/test_init.py index d3464088394..dddba971aad 100644 --- a/tests/components/recorder/test_init.py +++ b/tests/components/recorder/test_init.py @@ -32,7 +32,6 @@ from homeassistant.const import ( STATE_UNLOCKED, ) from homeassistant.core import Context, CoreState, HomeAssistant, callback -from homeassistant.helpers.typing import HomeAssistantType from homeassistant.setup import async_setup_component, setup_component from homeassistant.util import dt as dt_util @@ -116,7 +115,7 @@ async def test_state_gets_saved_when_set_before_start_event( async def test_saving_state( - hass: HomeAssistantType, async_setup_recorder_instance: SetupRecorderInstanceT + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT ): """Test saving and restoring a state.""" instance = await async_setup_recorder_instance(hass) @@ -139,7 +138,7 @@ async def test_saving_state( async def test_saving_many_states( - hass: HomeAssistantType, async_setup_recorder_instance: SetupRecorderInstanceT + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT ): """Test we expire after many commits.""" instance = await async_setup_recorder_instance(hass) @@ -165,7 +164,7 @@ async def test_saving_many_states( async def test_saving_state_with_intermixed_time_changes( - hass: HomeAssistantType, async_setup_recorder_instance: SetupRecorderInstanceT + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT ): """Test saving states with intermixed time changes.""" instance = await async_setup_recorder_instance(hass) diff --git a/tests/components/recorder/test_purge.py b/tests/components/recorder/test_purge.py index d1825663ccc..23164bd73f5 100644 --- a/tests/components/recorder/test_purge.py +++ b/tests/components/recorder/test_purge.py @@ -12,7 +12,8 @@ from homeassistant.components.recorder.models import Events, RecorderRuns, State from homeassistant.components.recorder.purge import purge_old_data from homeassistant.components.recorder.util import session_scope from homeassistant.const import EVENT_STATE_CHANGED -from homeassistant.helpers.typing import ConfigType, HomeAssistantType +from homeassistant.core import HomeAssistant +from homeassistant.helpers.typing import ConfigType from homeassistant.util import dt as dt_util from .common import ( @@ -25,7 +26,7 @@ from .conftest import SetupRecorderInstanceT async def test_purge_old_states( - hass: HomeAssistantType, async_setup_recorder_instance: SetupRecorderInstanceT + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT ): """Test deleting old states.""" instance = await async_setup_recorder_instance(hass) @@ -57,7 +58,7 @@ async def test_purge_old_states( async def test_purge_old_states_encouters_database_corruption( - hass: HomeAssistantType, async_setup_recorder_instance: SetupRecorderInstanceT + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT ): """Test database image image is malformed while deleting old states.""" instance = await async_setup_recorder_instance(hass) @@ -89,7 +90,7 @@ async def test_purge_old_states_encouters_database_corruption( async def test_purge_old_states_encounters_temporary_mysql_error( - hass: HomeAssistantType, + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT, caplog, ): @@ -122,7 +123,7 @@ async def test_purge_old_states_encounters_temporary_mysql_error( async def test_purge_old_states_encounters_operational_error( - hass: HomeAssistantType, + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT, caplog, ): @@ -150,7 +151,7 @@ async def test_purge_old_states_encounters_operational_error( async def test_purge_old_events( - hass: HomeAssistantType, async_setup_recorder_instance: SetupRecorderInstanceT + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT ): """Test deleting old events.""" instance = await async_setup_recorder_instance(hass) @@ -173,7 +174,7 @@ async def test_purge_old_events( async def test_purge_old_recorder_runs( - hass: HomeAssistantType, async_setup_recorder_instance: SetupRecorderInstanceT + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT ): """Test deleting old recorder runs keeps current run.""" instance = await async_setup_recorder_instance(hass) @@ -195,7 +196,7 @@ async def test_purge_old_recorder_runs( async def test_purge_method( - hass: HomeAssistantType, + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT, caplog, ): @@ -265,12 +266,12 @@ async def test_purge_method( async def test_purge_edge_case( - hass: HomeAssistantType, + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT, ): """Test states and events are purged even if they occurred shortly before purge_before.""" - async def _add_db_entries(hass: HomeAssistantType, timestamp: datetime) -> None: + async def _add_db_entries(hass: HomeAssistant, timestamp: datetime) -> None: with recorder.session_scope(hass=hass) as session: session.add( Events( @@ -322,7 +323,7 @@ async def test_purge_edge_case( async def test_purge_filtered_states( - hass: HomeAssistantType, + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT, ): """Test filtered states are purged.""" @@ -330,7 +331,7 @@ async def test_purge_filtered_states( instance = await async_setup_recorder_instance(hass, config) assert instance.entity_filter("sensor.excluded") is False - def _add_db_entries(hass: HomeAssistantType) -> None: + def _add_db_entries(hass: HomeAssistant) -> None: with recorder.session_scope(hass=hass) as session: # Add states and state_changed events that should be purged for days in range(1, 4): @@ -467,14 +468,14 @@ async def test_purge_filtered_states( async def test_purge_filtered_events( - hass: HomeAssistantType, + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT, ): """Test filtered events are purged.""" config: ConfigType = {"exclude": {"event_types": ["EVENT_PURGE"]}} instance = await async_setup_recorder_instance(hass, config) - def _add_db_entries(hass: HomeAssistantType) -> None: + def _add_db_entries(hass: HomeAssistant) -> None: with recorder.session_scope(hass=hass) as session: # Add events that should be purged for days in range(1, 4): @@ -548,7 +549,7 @@ async def test_purge_filtered_events( async def test_purge_filtered_events_state_changed( - hass: HomeAssistantType, + hass: HomeAssistant, async_setup_recorder_instance: SetupRecorderInstanceT, ): """Test filtered state_changed events are purged. This should also remove all states.""" @@ -557,7 +558,7 @@ async def test_purge_filtered_events_state_changed( # Assert entity_id is NOT excluded assert instance.entity_filter("sensor.excluded") is True - def _add_db_entries(hass: HomeAssistantType) -> None: + def _add_db_entries(hass: HomeAssistant) -> None: with recorder.session_scope(hass=hass) as session: # Add states and state_changed events that should be purged for days in range(1, 4): @@ -652,7 +653,7 @@ async def test_purge_filtered_events_state_changed( assert session.query(States).get(63).old_state_id == 62 # should have been kept -async def _add_test_states(hass: HomeAssistantType, instance: recorder.Recorder): +async def _add_test_states(hass: HomeAssistant, instance: recorder.Recorder): """Add multiple states to the db for testing.""" utcnow = dt_util.utcnow() five_days_ago = utcnow - timedelta(days=5) @@ -700,7 +701,7 @@ async def _add_test_states(hass: HomeAssistantType, instance: recorder.Recorder) old_state_id = state.state_id -async def _add_test_events(hass: HomeAssistantType, instance: recorder.Recorder): +async def _add_test_events(hass: HomeAssistant, instance: recorder.Recorder): """Add a few events for testing.""" utcnow = dt_util.utcnow() five_days_ago = utcnow - timedelta(days=5) @@ -733,7 +734,7 @@ async def _add_test_events(hass: HomeAssistantType, instance: recorder.Recorder) ) -async def _add_test_recorder_runs(hass: HomeAssistantType, instance: recorder.Recorder): +async def _add_test_recorder_runs(hass: HomeAssistant, instance: recorder.Recorder): """Add a few recorder_runs for testing.""" utcnow = dt_util.utcnow() five_days_ago = utcnow - timedelta(days=5) diff --git a/tests/components/roku/__init__.py b/tests/components/roku/__init__.py index 4ab2991bd43..e9f2d5c85bf 100644 --- a/tests/components/roku/__init__.py +++ b/tests/components/roku/__init__.py @@ -9,7 +9,7 @@ from homeassistant.components.ssdp import ( ATTR_UPNP_SERIAL, ) from homeassistant.const import CONF_HOST, CONF_ID, CONF_NAME -from homeassistant.helpers.typing import HomeAssistantType +from homeassistant.core import HomeAssistant from tests.common import MockConfigEntry, load_fixture from tests.test_util.aiohttp import AiohttpClientMocker @@ -162,7 +162,7 @@ def mock_connection_server_error( async def setup_integration( - hass: HomeAssistantType, + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, device: str = "roku3", app: str = "roku", diff --git a/tests/components/roku/test_config_flow.py b/tests/components/roku/test_config_flow.py index ab0072377cd..fde46b6621c 100644 --- a/tests/components/roku/test_config_flow.py +++ b/tests/components/roku/test_config_flow.py @@ -4,12 +4,12 @@ from unittest.mock import patch from homeassistant.components.roku.const import DOMAIN from homeassistant.config_entries import SOURCE_HOMEKIT, SOURCE_SSDP, SOURCE_USER from homeassistant.const import CONF_HOST, CONF_NAME, CONF_SOURCE +from homeassistant.core import HomeAssistant from homeassistant.data_entry_flow import ( RESULT_TYPE_ABORT, RESULT_TYPE_CREATE_ENTRY, RESULT_TYPE_FORM, ) -from homeassistant.helpers.typing import HomeAssistantType from homeassistant.setup import async_setup_component from tests.components.roku import ( @@ -26,7 +26,7 @@ from tests.test_util.aiohttp import AiohttpClientMocker async def test_duplicate_error( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test that errors are shown when duplicates are added.""" await setup_integration(hass, aioclient_mock, skip_entry_setup=True) @@ -57,9 +57,7 @@ async def test_duplicate_error( assert result["reason"] == "already_configured" -async def test_form( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker -) -> None: +async def test_form(hass: HomeAssistant, aioclient_mock: AiohttpClientMocker) -> None: """Test the user step.""" await async_setup_component(hass, "persistent_notification", {}) mock_connection(aioclient_mock) @@ -90,7 +88,7 @@ async def test_form( async def test_form_cannot_connect( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test we handle cannot connect roku error.""" mock_connection(aioclient_mock, error=True) @@ -107,7 +105,7 @@ async def test_form_cannot_connect( assert result["errors"] == {"base": "cannot_connect"} -async def test_form_unknown_error(hass: HomeAssistantType) -> None: +async def test_form_unknown_error(hass: HomeAssistant) -> None: """Test we handle unknown error.""" result = await hass.config_entries.flow.async_init( DOMAIN, context={CONF_SOURCE: SOURCE_USER} @@ -130,7 +128,7 @@ async def test_form_unknown_error(hass: HomeAssistantType) -> None: async def test_homekit_cannot_connect( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test we abort homekit flow on connection error.""" mock_connection( @@ -151,7 +149,7 @@ async def test_homekit_cannot_connect( async def test_homekit_unknown_error( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test we abort homekit flow on unknown error.""" discovery_info = MOCK_HOMEKIT_DISCOVERY_INFO.copy() @@ -170,7 +168,7 @@ async def test_homekit_unknown_error( async def test_homekit_discovery( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test the homekit discovery flow.""" mock_connection(aioclient_mock, device="rokutv", host=HOMEKIT_HOST) @@ -213,7 +211,7 @@ async def test_homekit_discovery( async def test_ssdp_cannot_connect( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test we abort SSDP flow on connection error.""" mock_connection(aioclient_mock, error=True) @@ -230,7 +228,7 @@ async def test_ssdp_cannot_connect( async def test_ssdp_unknown_error( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test we abort SSDP flow on unknown error.""" discovery_info = MOCK_SSDP_DISCOVERY_INFO.copy() @@ -249,7 +247,7 @@ async def test_ssdp_unknown_error( async def test_ssdp_discovery( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test the SSDP discovery flow.""" mock_connection(aioclient_mock) diff --git a/tests/components/roku/test_init.py b/tests/components/roku/test_init.py index a5f16c6071f..be9131d5f91 100644 --- a/tests/components/roku/test_init.py +++ b/tests/components/roku/test_init.py @@ -7,14 +7,14 @@ from homeassistant.config_entries import ( ENTRY_STATE_NOT_LOADED, ENTRY_STATE_SETUP_RETRY, ) -from homeassistant.helpers.typing import HomeAssistantType +from homeassistant.core import HomeAssistant from tests.components.roku import setup_integration from tests.test_util.aiohttp import AiohttpClientMocker async def test_config_entry_not_ready( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test the Roku configuration entry not ready.""" entry = await setup_integration(hass, aioclient_mock, error=True) @@ -23,7 +23,7 @@ async def test_config_entry_not_ready( async def test_unload_config_entry( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test the Roku configuration entry unloading.""" with patch( diff --git a/tests/components/roku/test_media_player.py b/tests/components/roku/test_media_player.py index dca336be076..0964343e453 100644 --- a/tests/components/roku/test_media_player.py +++ b/tests/components/roku/test_media_player.py @@ -61,8 +61,8 @@ from homeassistant.const import ( STATE_STANDBY, STATE_UNAVAILABLE, ) +from homeassistant.core import HomeAssistant from homeassistant.helpers import device_registry as dr, entity_registry as er -from homeassistant.helpers.typing import HomeAssistantType from homeassistant.util import dt as dt_util from tests.common import async_fire_time_changed @@ -80,9 +80,7 @@ TV_SERIAL = "YN00H5555555" TV_SW_VERSION = "9.2.0" -async def test_setup( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker -) -> None: +async def test_setup(hass: HomeAssistant, aioclient_mock: AiohttpClientMocker) -> None: """Test setup with basic config.""" await setup_integration(hass, aioclient_mock) @@ -96,7 +94,7 @@ async def test_setup( async def test_idle_setup( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test setup with idle device.""" await setup_integration(hass, aioclient_mock, power=False) @@ -106,7 +104,7 @@ async def test_idle_setup( async def test_tv_setup( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test Roku TV setup.""" await setup_integration( @@ -128,7 +126,7 @@ async def test_tv_setup( async def test_availability( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test entity availability.""" now = dt_util.utcnow() @@ -153,7 +151,7 @@ async def test_availability( async def test_supported_features( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test supported features.""" await setup_integration(hass, aioclient_mock) @@ -177,7 +175,7 @@ async def test_supported_features( async def test_tv_supported_features( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test supported features for Roku TV.""" await setup_integration( @@ -207,7 +205,7 @@ async def test_tv_supported_features( async def test_attributes( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test attributes.""" await setup_integration(hass, aioclient_mock) @@ -222,7 +220,7 @@ async def test_attributes( async def test_attributes_app( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test attributes for app.""" await setup_integration(hass, aioclient_mock, app="netflix") @@ -237,7 +235,7 @@ async def test_attributes_app( async def test_attributes_app_media_playing( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test attributes for app with playing media.""" await setup_integration(hass, aioclient_mock, app="pluto", media_state="play") @@ -254,7 +252,7 @@ async def test_attributes_app_media_playing( async def test_attributes_app_media_paused( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test attributes for app with paused media.""" await setup_integration(hass, aioclient_mock, app="pluto", media_state="pause") @@ -271,7 +269,7 @@ async def test_attributes_app_media_paused( async def test_attributes_screensaver( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test attributes for app with screensaver.""" await setup_integration(hass, aioclient_mock, app="screensaver") @@ -286,7 +284,7 @@ async def test_attributes_screensaver( async def test_tv_attributes( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test attributes for Roku TV.""" await setup_integration( @@ -310,7 +308,7 @@ async def test_tv_attributes( async def test_tv_device_registry( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test device registered for Roku TV in the device registry.""" await setup_integration( @@ -333,7 +331,7 @@ async def test_tv_device_registry( async def test_services( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test the different media player services.""" await setup_integration(hass, aioclient_mock) @@ -448,7 +446,7 @@ async def test_services( async def test_tv_services( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test the media player services related to Roku TV.""" await setup_integration( @@ -691,7 +689,7 @@ async def test_media_browse_internal(hass, aioclient_mock, hass_ws_client): async def test_integration_services( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test integration services.""" await setup_integration(hass, aioclient_mock) diff --git a/tests/components/roku/test_remote.py b/tests/components/roku/test_remote.py index 363c7134bad..5b1c0509e1f 100644 --- a/tests/components/roku/test_remote.py +++ b/tests/components/roku/test_remote.py @@ -7,8 +7,8 @@ from homeassistant.components.remote import ( SERVICE_SEND_COMMAND, ) from homeassistant.const import ATTR_ENTITY_ID, SERVICE_TURN_OFF, SERVICE_TURN_ON +from homeassistant.core import HomeAssistant from homeassistant.helpers import entity_registry as er -from homeassistant.helpers.typing import HomeAssistantType from tests.components.roku import UPNP_SERIAL, setup_integration from tests.test_util.aiohttp import AiohttpClientMocker @@ -18,16 +18,14 @@ MAIN_ENTITY_ID = f"{REMOTE_DOMAIN}.my_roku_3" # pylint: disable=redefined-outer-name -async def test_setup( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker -) -> None: +async def test_setup(hass: HomeAssistant, aioclient_mock: AiohttpClientMocker) -> None: """Test setup with basic config.""" await setup_integration(hass, aioclient_mock) assert hass.states.get(MAIN_ENTITY_ID) async def test_unique_id( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test unique id.""" await setup_integration(hass, aioclient_mock) @@ -39,7 +37,7 @@ async def test_unique_id( async def test_main_services( - hass: HomeAssistantType, aioclient_mock: AiohttpClientMocker + hass: HomeAssistant, aioclient_mock: AiohttpClientMocker ) -> None: """Test platform services.""" await setup_integration(hass, aioclient_mock) diff --git a/tests/components/slack/test_notify.py b/tests/components/slack/test_notify.py index 6c353cf8fc6..f10673cced4 100644 --- a/tests/components/slack/test_notify.py +++ b/tests/components/slack/test_notify.py @@ -22,7 +22,7 @@ from homeassistant.const import ( CONF_PLATFORM, CONF_USERNAME, ) -from homeassistant.helpers.typing import HomeAssistantType +from homeassistant.core import HomeAssistant from homeassistant.setup import async_setup_component MODULE_PATH = "homeassistant.components.slack.notify" @@ -47,7 +47,7 @@ def filter_log_records(caplog: LogCaptureFixture) -> list[logging.LogRecord]: ] -async def test_setup(hass: HomeAssistantType, caplog: LogCaptureFixture): +async def test_setup(hass: HomeAssistant, caplog: LogCaptureFixture): """Test setup slack notify.""" config = DEFAULT_CONFIG @@ -68,7 +68,7 @@ async def test_setup(hass: HomeAssistantType, caplog: LogCaptureFixture): client.auth_test.assert_called_once_with() -async def test_setup_clientError(hass: HomeAssistantType, caplog: LogCaptureFixture): +async def test_setup_clientError(hass: HomeAssistant, caplog: LogCaptureFixture): """Test setup slack notify with aiohttp.ClientError exception.""" config = copy.deepcopy(DEFAULT_CONFIG) config[notify.DOMAIN][0].update({CONF_USERNAME: "user", CONF_ICON: "icon"}) @@ -89,7 +89,7 @@ async def test_setup_clientError(hass: HomeAssistantType, caplog: LogCaptureFixt assert aiohttp.ClientError.__qualname__ in record.message -async def test_setup_slackApiError(hass: HomeAssistantType, caplog: LogCaptureFixture): +async def test_setup_slackApiError(hass: HomeAssistant, caplog: LogCaptureFixture): """Test setup slack notify with SlackApiError exception.""" config = DEFAULT_CONFIG