From bd2b72235e54564a9c0ef45cbebcd2ab889bf458 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Fri, 13 Sep 2024 06:42:32 +0200
Subject: [PATCH] Use root import in tests (#125862)

* Use root import in components

* One more
---
 tests/common.py                                          | 2 +-
 tests/components/cloud/test_tts.py                       | 2 +-
 tests/components/conftest.py                             | 2 +-
 tests/components/dlna_dms/test_media_source.py           | 6 +++---
 tests/components/google_assistant/test_helpers.py        | 2 +-
 .../homeassistant_hardware/test_config_flow.py           | 2 +-
 .../homeassistant_hardware/test_config_flow_failures.py  | 6 +-----
 .../test_silabs_multiprotocol_addon.py                   | 9 +++++++--
 .../homeassistant_sky_connect/test_config_flow.py        | 2 +-
 .../components/homeassistant_yellow/test_config_flow.py  | 7 +++++--
 tests/components/mqtt/test_config_flow.py                | 8 +++++---
 tests/components/script/test_blueprint.py                | 2 +-
 tests/components/zwave_js/test_config_flow.py            | 3 +--
 tests/components/zwave_js/test_init.py                   | 2 +-
 14 files changed, 30 insertions(+), 25 deletions(-)

diff --git a/tests/common.py b/tests/common.py
index c2d561551ca..21b5ee1e720 100644
--- a/tests/common.py
+++ b/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")
diff --git a/tests/components/cloud/test_tts.py b/tests/components/cloud/test_tts.py
index 52a9bc19ea2..50ea5e87d82 100644
--- a/tests/components/cloud/test_tts.py
+++ b/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
diff --git a/tests/components/conftest.py b/tests/components/conftest.py
index 39ff7071dc4..1e79248fbeb 100644
--- a/tests/components/conftest.py
+++ b/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
diff --git a/tests/components/dlna_dms/test_media_source.py b/tests/components/dlna_dms/test_media_source.py
index 641232e356a..ad290826075 100644
--- a/tests/components/dlna_dms/test_media_source.py
+++ b/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
diff --git a/tests/components/google_assistant/test_helpers.py b/tests/components/google_assistant/test_helpers.py
index 492f1be1829..8b46545d9c5 100644
--- a/tests/components/google_assistant/test_helpers.py
+++ b/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
diff --git a/tests/components/homeassistant_hardware/test_config_flow.py b/tests/components/homeassistant_hardware/test_config_flow.py
index a1842f4c4e6..b94238c1225 100644
--- a/tests/components/homeassistant_hardware/test_config_flow.py
+++ b/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,
diff --git a/tests/components/homeassistant_hardware/test_config_flow_failures.py b/tests/components/homeassistant_hardware/test_config_flow_failures.py
index 4c3ea7d28fa..a5c5f4d666a 100644
--- a/tests/components/homeassistant_hardware/test_config_flow_failures.py
+++ b/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,
diff --git a/tests/components/homeassistant_hardware/test_silabs_multiprotocol_addon.py b/tests/components/homeassistant_hardware/test_silabs_multiprotocol_addon.py
index 5718133cd24..65fab707c0b 100644
--- a/tests/components/homeassistant_hardware/test_silabs_multiprotocol_addon.py
+++ b/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
diff --git a/tests/components/homeassistant_sky_connect/test_config_flow.py b/tests/components/homeassistant_sky_connect/test_config_flow.py
index 0d4c517b07f..de9af6f204c 100644
--- a/tests/components/homeassistant_sky_connect/test_config_flow.py
+++ b/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,
 )
diff --git a/tests/components/homeassistant_yellow/test_config_flow.py b/tests/components/homeassistant_yellow/test_config_flow.py
index 949e58e61b6..c82c08314b0 100644
--- a/tests/components/homeassistant_yellow/test_config_flow.py
+++ b/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,
 )
diff --git a/tests/components/mqtt/test_config_flow.py b/tests/components/mqtt/test_config_flow.py
index d2f399899b1..70231cc6115 100644
--- a/tests/components/mqtt/test_config_flow.py
+++ b/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,
diff --git a/tests/components/script/test_blueprint.py b/tests/components/script/test_blueprint.py
index 160b330c109..86567d2f16f 100644
--- a/tests/components/script/test_blueprint.py
+++ b/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
diff --git a/tests/components/zwave_js/test_config_flow.py b/tests/components/zwave_js/test_config_flow.py
index a3affb6b977..d6081d24b18 100644
--- a/tests/components/zwave_js/test_config_flow.py
+++ b/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
diff --git a/tests/components/zwave_js/test_init.py b/tests/components/zwave_js/test_init.py
index 51aeee72c1d..5ec72b8a46a 100644
--- a/tests/components/zwave_js/test_init.py
+++ b/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