jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-09 10:59:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix backup tests typing warnings (#141274)

Browse Source
This commit is contained in:
Martin Hjelmare
2025-03-24 12:36:57 +01:00
committed by GitHub
parent 86ff540db9
commit 75cd32b742
2 changed files with 45 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
tests/components/backup/common.py
View File

@@ -2,9 +2,9 @@
from __future__ import annotations
from collections.abc import AsyncIterator, Callable, Coroutine, Iterable
from collections.abc import AsyncIterator, Buffer, Callable, Coroutine, Iterable
from pathlib import Path
from typing import Any
from typing import Any, cast
from unittest.mock import AsyncMock, Mock, patch
from homeassistant.components.backup import (
@@ -16,6 +16,7 @@ from homeassistant.components.backup import (
BackupNotFound,
Folder,
)
from homeassistant.components.backup.backup import CoreLocalBackupAgent
from homeassistant.components.backup.const import DATA_MANAGER
from homeassistant.core import HomeAssistant
from homeassistant.helpers.backup import async_initialize_backup
@@ -69,7 +70,7 @@ def mock_backup_agent(name: str, backups: list[AgentBackup] | None = None) -> Mo
async def delete_backup(backup_id: str, **kwargs: Any) -> None:
"""Mock delete."""
get_backup(backup_id)
await get_backup(backup_id)
async def download_backup(backup_id: str, **kwargs: Any) -> AsyncIterator[bytes]:
"""Mock download."""
@@ -77,7 +78,7 @@ def mock_backup_agent(name: str, backups: list[AgentBackup] | None = None) -> Mo
async def get_backup(backup_id: str, **kwargs: Any) -> AgentBackup:
"""Get a backup."""
backup = next((b for b in backups if b.backup_id == backup_id), None)
backup = next((b for b in _backups if b.backup_id == backup_id), None)
if backup is None:
raise BackupNotFound
return backup
@@ -89,15 +90,15 @@ def mock_backup_agent(name: str, backups: list[AgentBackup] | None = None) -> Mo
**kwargs: Any,
) -> None:
"""Upload a backup."""
backups.append(backup)
_backups.append(backup)
backup_stream = await open_stream()
backup_data = bytearray()
async for chunk in backup_stream:
backup_data += chunk
backups_data[backup.backup_id] = backup_data
backups = backups or []
backups_data: dict[str, bytes] = {}
_backups = backups or []
backups_data: dict[str, Buffer] = {}
mock_agent = Mock(spec=BackupAgent)
mock_agent.domain = TEST_DOMAIN
mock_agent.name = name
@@ -113,7 +114,7 @@ def mock_backup_agent(name: str, backups: list[AgentBackup] | None = None) -> Mo
side_effect=get_backup, spec_set=[BackupAgent.async_get_backup]
)
mock_agent.async_list_backups = AsyncMock(
return_value=backups, spec_set=[BackupAgent.async_list_backups]
return_value=_backups, spec_set=[BackupAgent.async_list_backups]
)
mock_agent.async_upload_backup = AsyncMock(
side_effect=upload_backup,
@@ -160,11 +161,18 @@ async def setup_backup_integration(
if LOCAL_AGENT_ID not in backups or with_hassio:
return remote_agents_dict
agent = hass.data[DATA_MANAGER].backup_agents[LOCAL_AGENT_ID]
local_agent = cast(
CoreLocalBackupAgent, hass.data[DATA_MANAGER].backup_agents[LOCAL_AGENT_ID]
)
for backup in backups[LOCAL_AGENT_ID]:
await agent.async_upload_backup(open_stream=None, backup=backup)
agent._loaded_backups = True
await local_agent.async_upload_backup(
open_stream=AsyncMock(
side_effect=RuntimeError("Local agent does not open stream")
),
backup=backup,
)
local_agent._loaded_backups = True
return remote_agents_dict