Add repair-issue that backup location setup is missing in Synology DSM (#138233)

* add missing backup location setup repair-issue

* add tests

* tweak translation strings

* add test for other fixable issues

* remove senseless abort reason no_file_station
This commit is contained in:
Michael 2025-02-17 13:03:31 +01:00 committed by GitHub
parent a7f63e3847
commit df6cb0b824
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 496 additions and 0 deletions

View File

@ -35,13 +35,17 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers import issue_registry as ir
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .const import (
CONF_BACKUP_PATH,
CONF_DEVICE_TOKEN,
DEFAULT_TIMEOUT,
DOMAIN,
EXCEPTION_DETAILS,
EXCEPTION_UNKNOWN,
ISSUE_MISSING_BACKUP_SETUP,
SYNOLOGY_CONNECTION_EXCEPTIONS,
)
@ -174,6 +178,19 @@ class SynoApi:
" permissions or no writable shared folders available"
)
if shares and not self._entry.options.get(CONF_BACKUP_PATH):
ir.async_create_issue(
self._hass,
DOMAIN,
f"{ISSUE_MISSING_BACKUP_SETUP}_{self._entry.unique_id}",
data={"entry_id": self._entry.entry_id},
is_fixable=True,
is_persistent=False,
severity=ir.IssueSeverity.WARNING,
translation_key=ISSUE_MISSING_BACKUP_SETUP,
translation_placeholders={"title": self._entry.title},
)
LOGGER.debug(
"State of File Station during setup of '%s': %s",
self._entry.unique_id,

View File

@ -35,6 +35,8 @@ PLATFORMS = [
EXCEPTION_DETAILS = "details"
EXCEPTION_UNKNOWN = "unknown"
ISSUE_MISSING_BACKUP_SETUP = "missing_backup_setup"
# Configuration
CONF_SERIAL = "serial"
CONF_VOLUMES = "volumes"

View File

@ -0,0 +1,125 @@
"""Repair flows for the Synology DSM integration."""
from __future__ import annotations
from contextlib import suppress
import logging
from typing import cast
from synology_dsm.api.file_station.models import SynoFileSharedFolder
import voluptuous as vol
from homeassistant import data_entry_flow
from homeassistant.components.repairs import ConfirmRepairFlow, RepairsFlow
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import issue_registry as ir
from homeassistant.helpers.selector import (
SelectOptionDict,
SelectSelector,
SelectSelectorConfig,
SelectSelectorMode,
)
from .const import (
CONF_BACKUP_PATH,
CONF_BACKUP_SHARE,
DOMAIN,
ISSUE_MISSING_BACKUP_SETUP,
SYNOLOGY_CONNECTION_EXCEPTIONS,
)
from .models import SynologyDSMData
LOGGER = logging.getLogger(__name__)
class MissingBackupSetupRepairFlow(RepairsFlow):
"""Handler for an issue fixing flow."""
def __init__(self, entry: ConfigEntry, issue_id: str) -> None:
"""Create flow."""
self.entry = entry
self.issue_id = issue_id
super().__init__()
async def async_step_init(
self, user_input: dict[str, str] | None = None
) -> data_entry_flow.FlowResult:
"""Handle the first step of a fix flow."""
return self.async_show_menu(
menu_options=["confirm", "ignore"],
description_placeholders={
"docs_url": "https://www.home-assistant.io/integrations/synology_dsm/#backup-location"
},
)
async def async_step_confirm(
self, user_input: dict[str, str] | None = None
) -> data_entry_flow.FlowResult:
"""Handle the confirm step of a fix flow."""
syno_data: SynologyDSMData = self.hass.data[DOMAIN][self.entry.unique_id]
if user_input is not None:
self.hass.config_entries.async_update_entry(
self.entry, options={**dict(self.entry.options), **user_input}
)
return self.async_create_entry(data={})
shares: list[SynoFileSharedFolder] | None = None
if syno_data.api.file_station:
with suppress(*SYNOLOGY_CONNECTION_EXCEPTIONS):
shares = await syno_data.api.file_station.get_shared_folders(
only_writable=True
)
if not shares:
return self.async_abort(reason="no_shares")
return self.async_show_form(
data_schema=vol.Schema(
{
vol.Required(
CONF_BACKUP_SHARE,
default=self.entry.options[CONF_BACKUP_SHARE],
): SelectSelector(
SelectSelectorConfig(
options=[
SelectOptionDict(value=s.path, label=s.name)
for s in shares
],
mode=SelectSelectorMode.DROPDOWN,
),
),
vol.Required(
CONF_BACKUP_PATH,
default=self.entry.options[CONF_BACKUP_PATH],
): str,
}
),
)
async def async_step_ignore(
self, _: dict[str, str] | None = None
) -> data_entry_flow.FlowResult:
"""Handle the confirm step of a fix flow."""
ir.async_ignore_issue(self.hass, DOMAIN, self.issue_id, True)
return self.async_abort(reason="ignored")
async def async_create_fix_flow(
hass: HomeAssistant,
issue_id: str,
data: dict[str, str | int | float | None] | None,
) -> RepairsFlow:
"""Create flow."""
entry = None
if data and (entry_id := data.get("entry_id")):
entry_id = cast(str, entry_id)
entry = hass.config_entries.async_get_entry(entry_id)
if entry and issue_id.startswith(ISSUE_MISSING_BACKUP_SETUP):
return MissingBackupSetupRepairFlow(entry, issue_id)
return ConfirmRepairFlow()

View File

@ -185,6 +185,37 @@
}
}
},
"issues": {
"missing_backup_setup": {
"title": "Backup location not configured for {title}",
"fix_flow": {
"step": {
"init": {
"description": "The backup location for {title} is not configured. Do you want to set it up now? Details can be found in the integration documentation under [Backup Location]({docs_url})",
"menu_options": {
"confirm": "Set up the backup location now",
"ignore": "Don't set it up now"
}
},
"confirm": {
"title": "[%key:component::synology_dsm::config::step::backup_share::title%]",
"data": {
"backup_share": "[%key:component::synology_dsm::config::step::backup_share::data::backup_share%]",
"backup_path": "[%key:component::synology_dsm::config::step::backup_share::data::backup_path%]"
},
"data_description": {
"backup_share": "[%key:component::synology_dsm::config::step::backup_share::data_description::backup_share%]",
"backup_path": "[%key:component::synology_dsm::config::step::backup_share::data_description::backup_path%]"
}
}
},
"abort": {
"no_shares": "There are no shared folders available for the user.\nPlease check the documentation.",
"ignored": "The backup location has not been configured.\nYou can still set it up later via the integration options."
}
}
}
},
"services": {
"reboot": {
"name": "Reboot",

View File

@ -0,0 +1,321 @@
"""Test repairs for synology dsm."""
from __future__ import annotations
from unittest.mock import AsyncMock, MagicMock, Mock, patch
import pytest
from synology_dsm.api.file_station.models import SynoFileSharedFolder
from homeassistant.components.repairs import DOMAIN as REPAIRS_DOMAIN
from homeassistant.components.synology_dsm.const import (
CONF_BACKUP_PATH,
CONF_BACKUP_SHARE,
DOMAIN,
)
from homeassistant.const import (
CONF_HOST,
CONF_MAC,
CONF_PASSWORD,
CONF_PORT,
CONF_SSL,
CONF_USERNAME,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import issue_registry as ir
from homeassistant.setup import async_setup_component
from .consts import HOST, MACS, PASSWORD, PORT, SERIAL, USE_SSL, USERNAME
from tests.common import ANY, MockConfigEntry
from tests.components.repairs import process_repair_fix_flow, start_repair_fix_flow
from tests.typing import ClientSessionGenerator, WebSocketGenerator
@pytest.fixture
def mock_dsm_with_filestation():
"""Mock a successful service with filestation support."""
with patch("homeassistant.components.synology_dsm.common.SynologyDSM") as dsm:
dsm.login = AsyncMock(return_value=True)
dsm.update = AsyncMock(return_value=True)
dsm.surveillance_station.update = AsyncMock(return_value=True)
dsm.upgrade.update = AsyncMock(return_value=True)
dsm.utilisation = Mock(cpu_user_load=1, update=AsyncMock(return_value=True))
dsm.network = Mock(update=AsyncMock(return_value=True), macs=MACS)
dsm.storage = Mock(
disks_ids=["sda", "sdb", "sdc"],
volumes_ids=["volume_1"],
update=AsyncMock(return_value=True),
)
dsm.information = Mock(serial=SERIAL)
dsm.file = AsyncMock(
get_shared_folders=AsyncMock(
return_value=[
SynoFileSharedFolder(
additional=None,
is_dir=True,
name="HA Backup",
path="/ha_backup",
)
]
),
)
dsm.logout = AsyncMock(return_value=True)
yield dsm
@pytest.fixture
async def setup_dsm_with_filestation(
hass: HomeAssistant,
mock_dsm_with_filestation: MagicMock,
):
"""Mock setup of synology dsm config entry."""
with (
patch(
"homeassistant.components.synology_dsm.common.SynologyDSM",
return_value=mock_dsm_with_filestation,
),
patch("homeassistant.components.synology_dsm.PLATFORMS", return_value=[]),
):
entry = MockConfigEntry(
domain=DOMAIN,
data={
CONF_HOST: HOST,
CONF_PORT: PORT,
CONF_SSL: USE_SSL,
CONF_USERNAME: USERNAME,
CONF_PASSWORD: PASSWORD,
CONF_MAC: MACS[0],
},
options={
CONF_BACKUP_PATH: None,
CONF_BACKUP_SHARE: None,
},
unique_id="my_serial",
)
entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(entry.entry_id)
assert await async_setup_component(hass, REPAIRS_DOMAIN, {})
await hass.async_block_till_done()
yield mock_dsm_with_filestation
async def test_create_issue(
hass: HomeAssistant,
setup_dsm_with_filestation: MagicMock,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test the issue is created."""
ws_client = await hass_ws_client(hass)
await ws_client.send_json({"id": 1, "type": "repairs/list_issues"})
msg = await ws_client.receive_json()
assert msg["success"]
assert len(msg["result"]["issues"]) == 1
issue = msg["result"]["issues"][0]
assert issue["breaks_in_ha_version"] is None
assert issue["domain"] == DOMAIN
assert issue["issue_id"] == "missing_backup_setup_my_serial"
assert issue["translation_key"] == "missing_backup_setup"
async def test_missing_backup_ignore(
hass: HomeAssistant,
setup_dsm_with_filestation: MagicMock,
hass_client: ClientSessionGenerator,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test missing backup location setup issue is ignored by the user."""
ws_client = await hass_ws_client(hass)
client = await hass_client()
# get repair issues
await ws_client.send_json({"id": 1, "type": "repairs/list_issues"})
msg = await ws_client.receive_json()
assert msg["success"]
assert len(msg["result"]["issues"]) == 1
issue = msg["result"]["issues"][0]
assert not issue["ignored"]
# start repair flow
data = await start_repair_fix_flow(client, DOMAIN, "missing_backup_setup_my_serial")
flow_id = data["flow_id"]
assert data["description_placeholders"] == {
"docs_url": "https://www.home-assistant.io/integrations/synology_dsm/#backup-location"
}
assert data["step_id"] == "init"
assert data["menu_options"] == ["confirm", "ignore"]
# seelct to ignore the flow
data = await process_repair_fix_flow(
client, flow_id, json={"next_step_id": "ignore"}
)
assert data["type"] == "abort"
assert data["reason"] == "ignored"
# check issue is ignored
await ws_client.send_json({"id": 2, "type": "repairs/list_issues"})
msg = await ws_client.receive_json()
assert msg["success"]
assert len(msg["result"]["issues"]) == 1
issue = msg["result"]["issues"][0]
assert issue["ignored"]
async def test_missing_backup_success(
hass: HomeAssistant,
setup_dsm_with_filestation: MagicMock,
hass_client: ClientSessionGenerator,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test the missing backup location setup repair flow is fully processed by the user."""
ws_client = await hass_ws_client(hass)
client = await hass_client()
entries = hass.config_entries.async_entries(DOMAIN)
assert len(entries) == 1
entry = entries[0]
assert entry.options == {"backup_path": None, "backup_share": None}
# get repair issues
await ws_client.send_json({"id": 1, "type": "repairs/list_issues"})
msg = await ws_client.receive_json()
assert msg["success"]
assert len(msg["result"]["issues"]) == 1
issue = msg["result"]["issues"][0]
assert not issue["ignored"]
# start repair flow
data = await start_repair_fix_flow(client, DOMAIN, "missing_backup_setup_my_serial")
flow_id = data["flow_id"]
assert data["description_placeholders"] == {
"docs_url": "https://www.home-assistant.io/integrations/synology_dsm/#backup-location"
}
assert data["step_id"] == "init"
assert data["menu_options"] == ["confirm", "ignore"]
# seelct to confirm the flow
data = await process_repair_fix_flow(
client, flow_id, json={"next_step_id": "confirm"}
)
assert data["step_id"] == "confirm"
assert data["type"] == "form"
# fill out the form and submit
data = await process_repair_fix_flow(
client,
flow_id,
json={"backup_share": "/ha_backup", "backup_path": "backup_ha_dev"},
)
assert data["type"] == "create_entry"
assert entry.options == {
"backup_path": "backup_ha_dev",
"backup_share": "/ha_backup",
}
async def test_missing_backup_no_shares(
hass: HomeAssistant,
setup_dsm_with_filestation: MagicMock,
hass_client: ClientSessionGenerator,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test the missing backup location setup repair flow errors out."""
ws_client = await hass_ws_client(hass)
client = await hass_client()
# get repair issues
await ws_client.send_json({"id": 1, "type": "repairs/list_issues"})
msg = await ws_client.receive_json()
assert msg["success"]
assert len(msg["result"]["issues"]) == 1
# start repair flow
data = await start_repair_fix_flow(client, DOMAIN, "missing_backup_setup_my_serial")
flow_id = data["flow_id"]
assert data["description_placeholders"] == {
"docs_url": "https://www.home-assistant.io/integrations/synology_dsm/#backup-location"
}
assert data["step_id"] == "init"
assert data["menu_options"] == ["confirm", "ignore"]
# inject error
setup_dsm_with_filestation.file.get_shared_folders.return_value = []
# select to confirm the flow
data = await process_repair_fix_flow(
client, flow_id, json={"next_step_id": "confirm"}
)
assert data["type"] == "abort"
assert data["reason"] == "no_shares"
@pytest.mark.parametrize(
"ignore_translations",
["component.synology_dsm.issues.other_issue.title"],
)
async def test_other_fixable_issues(
hass: HomeAssistant,
setup_dsm_with_filestation: MagicMock,
hass_client: ClientSessionGenerator,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test fixing another issue."""
ws_client = await hass_ws_client(hass)
client = await hass_client()
await ws_client.send_json({"id": 1, "type": "repairs/list_issues"})
msg = await ws_client.receive_json()
assert msg["success"]
issue = {
"breaks_in_ha_version": None,
"domain": DOMAIN,
"issue_id": "other_issue",
"is_fixable": True,
"severity": "error",
"translation_key": "other_issue",
}
ir.async_create_issue(
hass,
issue["domain"],
issue["issue_id"],
is_fixable=issue["is_fixable"],
severity=issue["severity"],
translation_key=issue["translation_key"],
)
await ws_client.send_json({"id": 2, "type": "repairs/list_issues"})
msg = await ws_client.receive_json()
assert msg["success"]
results = msg["result"]["issues"]
assert {
"breaks_in_ha_version": None,
"created": ANY,
"dismissed_version": None,
"domain": "synology_dsm",
"ignored": False,
"is_fixable": True,
"issue_domain": None,
"issue_id": "other_issue",
"learn_more_url": None,
"severity": "error",
"translation_key": "other_issue",
"translation_placeholders": None,
} in results
data = await start_repair_fix_flow(client, DOMAIN, "other_issue")
flow_id = data["flow_id"]
assert data["step_id"] == "confirm"
data = await process_repair_fix_flow(client, flow_id)
assert data["type"] == "create_entry"
await hass.async_block_till_done()