84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
|
"""Tests for the Backup integration."""
|
||
|
|
from collections.abc import Awaitable, Callable
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from aiohttp import ClientWebSocketResponse
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from homeassistant.core import HomeAssistant
|
||
|
|
|
||
|
|
from .common import TEST_BACKUP, setup_backup_integration
|
||
|
|
|
||
|
|
|
||
|
|
async def test_info(
|
||
|
|
hass: HomeAssistant,
|
||
|
|
hass_ws_client: Callable[[HomeAssistant], Awaitable[ClientWebSocketResponse]],
|
||
|
|
) -> None:
|
||
|
|
"""Test getting backup info."""
|
||
|
|
await setup_backup_integration(hass)
|
||
|
|
|
||
|
|
client = await hass_ws_client(hass)
|
||
|
|
await hass.async_block_till_done()
|
||
|
|
|
||
|
|
await client.send_json({"id": 1, "type": "backup/info"})
|
||
|
|
msg = await client.receive_json()
|
||
|
|
|
||
|
|
assert msg["id"] == 1
|
||
|
|
assert msg["success"]
|
||
|
|
assert msg["result"] == {"backing_up": False, "backups": []}
|
||
|
|
|
||
|
|
|
||
|
|
async def test_remove(
|
||
|
|
hass: HomeAssistant,
|
||
|
|
hass_ws_client: Callable[[HomeAssistant], Awaitable[ClientWebSocketResponse]],
|
||
|
|
caplog: pytest.LogCaptureFixture,
|
||
|
|
) -> None:
|
||
|
|
"""Test removing a backup file."""
|
||
|
|
await setup_backup_integration(hass)
|
||
|
|
|
||
|
|
client = await hass_ws_client(hass)
|
||
|
|
await hass.async_block_till_done()
|
||
|
|
|
||
|
|
with patch(
|
||
|
|
"homeassistant.components.backup.websocket.BackupManager._backups",
|
||
|
|
{TEST_BACKUP.slug: TEST_BACKUP},
|
||
|
|
), patch(
|
||
|
|
"homeassistant.components.backup.websocket.BackupManager._loaded",
|
||
|
|
True,
|
||
|
|
), patch(
|
||
|
|
"pathlib.Path.unlink"
|
||
|
|
), patch(
|
||
|
|
"pathlib.Path.exists", return_value=True
|
||
|
|
):
|
||
|
|
await client.send_json({"id": 1, "type": "backup/remove", "slug": "abc123"})
|
||
|
|
msg = await client.receive_json()
|
||
|
|
|
||
|
|
assert msg["id"] == 1
|
||
|
|
assert msg["success"]
|
||
|
|
assert f"Removed backup located at {TEST_BACKUP.path}" in caplog.text
|
||
|
|
|
||
|
|
|
||
|
|
async def test_generate(
|
||
|
|
hass: HomeAssistant,
|
||
|
|
hass_ws_client: Callable[[HomeAssistant], Awaitable[ClientWebSocketResponse]],
|
||
|
|
) -> None:
|
||
|
|
"""Test removing a backup file."""
|
||
|
|
await setup_backup_integration(hass)
|
||
|
|
|
||
|
|
client = await hass_ws_client(hass)
|
||
|
|
await hass.async_block_till_done()
|
||
|
|
|
||
|
|
with patch(
|
||
|
|
"homeassistant.components.backup.websocket.BackupManager._backups",
|
||
|
|
{TEST_BACKUP.slug: TEST_BACKUP},
|
||
|
|
), patch(
|
||
|
|
"homeassistant.components.backup.websocket.BackupManager.generate_backup",
|
||
|
|
return_value=TEST_BACKUP,
|
||
|
|
):
|
||
|
|
await client.send_json({"id": 1, "type": "backup/generate"})
|
||
|
|
msg = await client.receive_json()
|
||
|
|
|
||
|
|
assert msg["id"] == 1
|
||
|
|
assert msg["success"]
|
||
|
|
assert msg["result"] == TEST_BACKUP.as_dict()
|