Use root import in tests (#125862)

* Use root import in components

* One more

tests/common.py

@@ -419,7 +419,7 @@ def async_fire_mqtt_message(
     from paho.mqtt.client import MQTTMessage
 
     # pylint: disable-next=import-outside-toplevel
-    from homeassistant.components.mqtt.models import MqttData
+    from homeassistant.components.mqtt import MqttData
 
     if isinstance(payload, str):
         payload = payload.encode("utf-8")

tests/components/cloud/test_tts.py

@@ -23,8 +23,8 @@ from homeassistant.components.tts import (
     ATTR_MEDIA_PLAYER_ENTITY_ID,
     ATTR_MESSAGE,
     DOMAIN as TTS_DOMAIN,
+    get_engine_instance,
 )
-from homeassistant.components.tts.helper import get_engine_instance
 from homeassistant.config import async_process_ha_core_config
 from homeassistant.const import ATTR_ENTITY_ID, STATE_UNAVAILABLE, STATE_UNKNOWN
 from homeassistant.core import HomeAssistant

tests/components/conftest.py

@@ -14,7 +14,7 @@ from homeassistant.const import STATE_OFF, STATE_ON
 from homeassistant.core import HomeAssistant
 
 if TYPE_CHECKING:
-    from homeassistant.components.hassio.addon_manager import AddonManager
+    from homeassistant.components.hassio import AddonManager
 
     from .conversation import MockAgent
     from .device_tracker.common import MockScanner

tests/components/dlna_dms/test_media_source.py

@@ -13,11 +13,11 @@ from homeassistant.components.dlna_dms.media_source import (
     DmsMediaSource,
     async_get_media_source,
 )
-from homeassistant.components.media_player.errors import BrowseError
-from homeassistant.components.media_source.error import Unresolvable
-from homeassistant.components.media_source.models import (
+from homeassistant.components.media_player import BrowseError
+from homeassistant.components.media_source import (
     BrowseMediaSource,
     MediaSourceItem,
+    Unresolvable,
 )
 from homeassistant.const import CONF_DEVICE_ID, CONF_URL
 from homeassistant.core import HomeAssistant

tests/components/google_assistant/test_helpers.py

@@ -14,7 +14,7 @@ from homeassistant.components.google_assistant.const import (
     SOURCE_LOCAL,
     STORE_GOOGLE_LOCAL_WEBHOOK_ID,
 )
-from homeassistant.components.matter.models import MatterDeviceInfo
+from homeassistant.components.matter import MatterDeviceInfo
 from homeassistant.config import async_process_ha_core_config
 from homeassistant.core import HomeAssistant, State
 from homeassistant.helpers import device_registry as dr, entity_registry as er

tests/components/homeassistant_hardware/test_config_flow.py

@@ -9,7 +9,7 @@ from unittest.mock import AsyncMock, Mock, call, patch
 import pytest
 from universal_silabs_flasher.const import ApplicationType
 
-from homeassistant.components.hassio.addon_manager import AddonInfo, AddonState
+from homeassistant.components.hassio import AddonInfo, AddonState
 from homeassistant.components.homeassistant_hardware.firmware_config_flow import (
     STEP_PICK_FIRMWARE_THREAD,
     STEP_PICK_FIRMWARE_ZIGBEE,

tests/components/homeassistant_hardware/test_config_flow_failures.py

@@ -5,11 +5,7 @@ from unittest.mock import AsyncMock
 import pytest
 from universal_silabs_flasher.const import ApplicationType
 
-from homeassistant.components.hassio.addon_manager import (
-    AddonError,
-    AddonInfo,
-    AddonState,
-)
+from homeassistant.components.hassio import AddonError, AddonInfo, AddonState
 from homeassistant.components.homeassistant_hardware.firmware_config_flow import (
     STEP_PICK_FIRMWARE_THREAD,
     STEP_PICK_FIRMWARE_ZIGBEE,

tests/components/homeassistant_hardware/test_silabs_multiprotocol_addon.py

@@ -8,8 +8,13 @@ from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
 
-from homeassistant.components.hassio import AddonError, AddonInfo, AddonState, HassIO
-from homeassistant.components.hassio.handler import HassioAPIError
+from homeassistant.components.hassio import (
+    AddonError,
+    AddonInfo,
+    AddonState,
+    HassIO,
+    HassioAPIError,
+)
 from homeassistant.components.homeassistant_hardware import silabs_multiprotocol_addon
 from homeassistant.components.zha import DOMAIN as ZHA_DOMAIN
 from homeassistant.config_entries import ConfigEntry, ConfigFlow

tests/components/homeassistant_sky_connect/test_config_flow.py

@@ -5,7 +5,7 @@ from unittest.mock import Mock, patch
 import pytest
 
 from homeassistant.components import usb
-from homeassistant.components.hassio.addon_manager import AddonInfo, AddonState
+from homeassistant.components.hassio import AddonInfo, AddonState
 from homeassistant.components.homeassistant_hardware.firmware_config_flow import (
     STEP_PICK_FIRMWARE_ZIGBEE,
 )

tests/components/homeassistant_yellow/test_config_flow.py

@@ -5,8 +5,11 @@ from unittest.mock import Mock, patch
 
 import pytest
 
-from homeassistant.components.hassio import DOMAIN as HASSIO_DOMAIN
-from homeassistant.components.hassio.addon_manager import AddonInfo, AddonState
+from homeassistant.components.hassio import (
+    DOMAIN as HASSIO_DOMAIN,
+    AddonInfo,
+    AddonState,
+)
 from homeassistant.components.homeassistant_hardware.firmware_config_flow import (
     STEP_PICK_FIRMWARE_ZIGBEE,
 )

tests/components/mqtt/test_config_flow.py

@@ -13,9 +13,11 @@ import voluptuous as vol
 
 from homeassistant import config_entries
 from homeassistant.components import mqtt
-from homeassistant.components.hassio import HassioServiceInfo
-from homeassistant.components.hassio.addon_manager import AddonError
-from homeassistant.components.hassio.handler import HassioAPIError
+from homeassistant.components.hassio import (
+    AddonError,
+    HassioAPIError,
+    HassioServiceInfo,
+)
 from homeassistant.components.mqtt.config_flow import PWD_NOT_CHANGED
 from homeassistant.const import (
     CONF_CLIENT_ID,

tests/components/script/test_blueprint.py

@@ -9,7 +9,7 @@ from unittest.mock import patch
 import pytest
 
 from homeassistant.components import script
-from homeassistant.components.blueprint.models import Blueprint, DomainBlueprints
+from homeassistant.components.blueprint import Blueprint, DomainBlueprints
 from homeassistant.config_entries import ConfigEntryState
 from homeassistant.core import Context, HomeAssistant, callback
 from homeassistant.helpers import device_registry as dr, template

tests/components/zwave_js/test_config_flow.py

@@ -14,8 +14,7 @@ from zwave_js_server.version import VersionInfo
 
 from homeassistant import config_entries
 from homeassistant.components import usb
-from homeassistant.components.hassio import HassioServiceInfo
-from homeassistant.components.hassio.handler import HassioAPIError
+from homeassistant.components.hassio import HassioAPIError, HassioServiceInfo
 from homeassistant.components.zeroconf import ZeroconfServiceInfo
 from homeassistant.components.zwave_js.config_flow import SERVER_VERSION_TIMEOUT, TITLE
 from homeassistant.components.zwave_js.const import ADDON_SLUG, DOMAIN

tests/components/zwave_js/test_init.py

@@ -12,7 +12,7 @@ from zwave_js_server.exceptions import BaseZwaveJSServerError, InvalidServerVers
 from zwave_js_server.model.node import Node
 from zwave_js_server.model.version import VersionInfo
 
-from homeassistant.components.hassio.handler import HassioAPIError
+from homeassistant.components.hassio import HassioAPIError
 from homeassistant.components.logger import DOMAIN as LOGGER_DOMAIN, SERVICE_SET_LEVEL
 from homeassistant.components.persistent_notification import async_dismiss
 from homeassistant.components.zwave_js import DOMAIN