From e9247fb94b5d22da172ddcfcc682a945c53c1ec6 Mon Sep 17 00:00:00 2001
From: Martin Hjelmare <marhje52@gmail.com>
Date: Thu, 14 Nov 2024 13:14:36 +0100
Subject: [PATCH] Export relevant names from backup integration (#130596)

---
 homeassistant/components/backup/__init__.py | 9 ++++++---
 tests/components/backup/common.py           | 9 ++++++---
 tests/components/backup/test_manager.py     | 9 +++++----
 tests/components/backup/test_websocket.py   | 2 +-
 4 files changed, 18 insertions(+), 11 deletions(-)

diff --git a/homeassistant/components/backup/__init__.py b/homeassistant/components/backup/__init__.py
index d160b0bd88f..a7cceee05cd 100644
--- a/homeassistant/components/backup/__init__.py
+++ b/homeassistant/components/backup/__init__.py
@@ -8,16 +8,19 @@ from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers.hassio import is_hassio
 from homeassistant.helpers.typing import ConfigType
 
-from .agent import BackupAgent, UploadedBackup
+from .agent import BackupAgent, BackupAgentPlatformProtocol, UploadedBackup
 from .const import DOMAIN, LOGGER
 from .http import async_register_http_views
-from .manager import BackupManager
-from .models import BackupUploadMetadata
+from .manager import BackupManager, BackupPlatformProtocol
+from .models import BackupUploadMetadata, BaseBackup
 from .websocket import async_register_websocket_handlers
 
 __all__ = [
     "BackupAgent",
+    "BackupAgentPlatformProtocol",
+    "BackupPlatformProtocol",
     "BackupUploadMetadata",
+    "BaseBackup",
     "UploadedBackup",
 ]
 
diff --git a/tests/components/backup/common.py b/tests/components/backup/common.py
index 18f9f28520d..5e2a7c7bf53 100644
--- a/tests/components/backup/common.py
+++ b/tests/components/backup/common.py
@@ -6,10 +6,13 @@ from pathlib import Path
 from typing import Any
 from unittest.mock import patch
 
-from homeassistant.components.backup import DOMAIN
-from homeassistant.components.backup.agent import BackupAgent, UploadedBackup
+from homeassistant.components.backup import (
+    DOMAIN,
+    BackupAgent,
+    BackupUploadMetadata,
+    UploadedBackup,
+)
 from homeassistant.components.backup.manager import Backup
-from homeassistant.components.backup.models import BackupUploadMetadata
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.typing import ConfigType
 from homeassistant.setup import async_setup_component
diff --git a/tests/components/backup/test_manager.py b/tests/components/backup/test_manager.py
index a15002d217e..ac856ad7d72 100644
--- a/tests/components/backup/test_manager.py
+++ b/tests/components/backup/test_manager.py
@@ -10,12 +10,13 @@ import aiohttp
 from multidict import CIMultiDict, CIMultiDictProxy
 import pytest
 
-from homeassistant.components.backup import BackupManager, BackupUploadMetadata
-from homeassistant.components.backup.agent import BackupAgentPlatformProtocol
-from homeassistant.components.backup.manager import (
+from homeassistant.components.backup import (
+    BackupAgentPlatformProtocol,
+    BackupManager,
     BackupPlatformProtocol,
-    BackupProgress,
+    BackupUploadMetadata,
 )
+from homeassistant.components.backup.manager import BackupProgress
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import HomeAssistantError
 from homeassistant.setup import async_setup_component
diff --git a/tests/components/backup/test_websocket.py b/tests/components/backup/test_websocket.py
index 8e130b1bceb..bc8b126aa27 100644
--- a/tests/components/backup/test_websocket.py
+++ b/tests/components/backup/test_websocket.py
@@ -8,9 +8,9 @@ from freezegun.api import FrozenDateTimeFactory
 import pytest
 from syrupy import SnapshotAssertion
 
+from homeassistant.components.backup import BaseBackup
 from homeassistant.components.backup.const import DATA_MANAGER
 from homeassistant.components.backup.manager import NewBackup
-from homeassistant.components.backup.models import BaseBackup
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import HomeAssistantError