21 lines
730 B
Python
21 lines
730 B
Python
"""Unit tests for ZHA backup platform."""
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from homeassistant.components.zha.backup import async_post_backup, async_pre_backup
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
|
|
async def test_pre_backup(hass: HomeAssistant, setup_zha) -> None:
|
|
"""Test backup creation when `async_pre_backup` is called."""
|
|
with patch("zigpy.backups.BackupManager.create_backup", AsyncMock()) as backup_mock:
|
|
await setup_zha()
|
|
await async_pre_backup(hass)
|
|
|
|
backup_mock.assert_called_once_with(load_devices=True)
|
|
|
|
|
|
async def test_post_backup(hass: HomeAssistant, setup_zha) -> None:
|
|
"""Test no-op `async_post_backup`."""
|
|
await setup_zha()
|
|
await async_post_backup(hass)
|