mirror of
https://github.com/home-assistant/core.git
synced 2025-11-09 10:59:40 +00:00
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
"""Unit tests for ZHA backup platform."""
|
|
|
|
from collections.abc import Callable, Coroutine
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from zigpy.application import ControllerApplication
|
|
|
|
from homeassistant.components.zha.backup import async_post_backup, async_pre_backup
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
|
|
async def test_pre_backup(
|
|
hass: HomeAssistant,
|
|
zigpy_app_controller: ControllerApplication,
|
|
setup_zha: Callable[..., Coroutine[None]],
|
|
) -> None:
|
|
"""Test backup creation when `async_pre_backup` is called."""
|
|
await setup_zha()
|
|
|
|
zigpy_app_controller.backups.create_backup = AsyncMock()
|
|
await async_pre_backup(hass)
|
|
|
|
zigpy_app_controller.backups.create_backup.assert_called_once_with(
|
|
load_devices=True
|
|
)
|
|
|
|
|
|
@patch("homeassistant.components.zha.backup.get_zha_gateway", side_effect=ValueError())
|
|
async def test_pre_backup_no_gateway(
|
|
hass: HomeAssistant, setup_zha: Callable[..., Coroutine[None]]
|
|
) -> None:
|
|
"""Test graceful backup failure when no gateway exists."""
|
|
await setup_zha()
|
|
await async_pre_backup(hass)
|
|
|
|
|
|
async def test_post_backup(
|
|
hass: HomeAssistant, setup_zha: Callable[..., Coroutine[None]]
|
|
) -> None:
|
|
"""Test no-op `async_post_backup`."""
|
|
await setup_zha()
|
|
await async_post_backup(hass)
|