mirror of
https://github.com/home-assistant/core.git
synced 2026-03-01 06:16:29 +00:00
327 lines
11 KiB
Python
327 lines
11 KiB
Python
"""Test the Homeassistant repairs module."""
|
|
|
|
from typing import Any
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components.repairs import DOMAIN as REPAIRS_DOMAIN
|
|
from homeassistant.const import EVENT_HOMEASSISTANT_STARTED
|
|
from homeassistant.core import DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant
|
|
from homeassistant.helpers import issue_registry as ir
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import MockConfigEntry
|
|
from tests.components.repairs import (
|
|
async_process_repairs_platforms,
|
|
process_repair_fix_flow,
|
|
start_repair_fix_flow,
|
|
)
|
|
from tests.typing import ClientSessionGenerator
|
|
|
|
|
|
async def test_integration_not_found_confirm_step(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSessionGenerator,
|
|
issue_registry: ir.IssueRegistry,
|
|
) -> None:
|
|
"""Test the integration_not_found issue confirm step."""
|
|
assert await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
assert await async_setup_component(hass, REPAIRS_DOMAIN, {REPAIRS_DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
MockConfigEntry(domain="test1").add_to_hass(hass)
|
|
assert await async_setup_component(hass, "test1", {}) is False
|
|
await hass.async_block_till_done()
|
|
entry1 = MockConfigEntry(domain="test1")
|
|
entry1.add_to_hass(hass)
|
|
entry2 = MockConfigEntry(domain="test1")
|
|
entry2.add_to_hass(hass)
|
|
issue_id = "integration_not_found.test1"
|
|
|
|
await async_process_repairs_platforms(hass)
|
|
http_client = await hass_client()
|
|
|
|
issue = issue_registry.async_get_issue(HOMEASSISTANT_DOMAIN, issue_id)
|
|
assert issue is not None
|
|
assert issue.translation_placeholders == {"domain": "test1"}
|
|
|
|
data = await start_repair_fix_flow(http_client, HOMEASSISTANT_DOMAIN, issue_id)
|
|
|
|
flow_id = data["flow_id"]
|
|
assert data["step_id"] == "init"
|
|
assert data["description_placeholders"] == {"domain": "test1"}
|
|
|
|
data = await process_repair_fix_flow(http_client, flow_id)
|
|
|
|
assert data["type"] == "menu"
|
|
|
|
# Apply fix
|
|
data = await process_repair_fix_flow(
|
|
http_client, flow_id, json={"next_step_id": "confirm"}
|
|
)
|
|
|
|
assert data["type"] == "create_entry"
|
|
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.config_entries.async_get_entry(entry1.entry_id) is None
|
|
assert hass.config_entries.async_get_entry(entry2.entry_id) is None
|
|
|
|
# Assert the issue is resolved
|
|
assert not issue_registry.async_get_issue(HOMEASSISTANT_DOMAIN, issue_id)
|
|
|
|
|
|
async def test_integration_not_found_ignore_step(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSessionGenerator,
|
|
issue_registry: ir.IssueRegistry,
|
|
) -> None:
|
|
"""Test the integration_not_found issue ignore step."""
|
|
assert await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
assert await async_setup_component(hass, REPAIRS_DOMAIN, {REPAIRS_DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
MockConfigEntry(domain="test1").add_to_hass(hass)
|
|
assert await async_setup_component(hass, "test1", {}) is False
|
|
await hass.async_block_till_done()
|
|
entry1 = MockConfigEntry(domain="test1")
|
|
entry1.add_to_hass(hass)
|
|
issue_id = "integration_not_found.test1"
|
|
|
|
await async_process_repairs_platforms(hass)
|
|
http_client = await hass_client()
|
|
|
|
issue = issue_registry.async_get_issue(HOMEASSISTANT_DOMAIN, issue_id)
|
|
assert issue is not None
|
|
assert issue.translation_placeholders == {"domain": "test1"}
|
|
|
|
data = await start_repair_fix_flow(http_client, HOMEASSISTANT_DOMAIN, issue_id)
|
|
|
|
flow_id = data["flow_id"]
|
|
assert data["step_id"] == "init"
|
|
assert data["description_placeholders"] == {"domain": "test1"}
|
|
|
|
# Show menu
|
|
data = await process_repair_fix_flow(http_client, flow_id)
|
|
|
|
assert data["type"] == "menu"
|
|
|
|
# Apply fix
|
|
data = await process_repair_fix_flow(
|
|
http_client, flow_id, json={"next_step_id": "ignore"}
|
|
)
|
|
|
|
assert data["type"] == "abort"
|
|
assert data["reason"] == "issue_ignored"
|
|
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.config_entries.async_get_entry(entry1.entry_id)
|
|
|
|
# Assert the issue is resolved
|
|
issue = issue_registry.async_get_issue(HOMEASSISTANT_DOMAIN, issue_id)
|
|
assert issue is not None
|
|
assert issue.dismissed_version is not None
|
|
|
|
|
|
async def test_orphaned_config_entry_confirm_step(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSessionGenerator,
|
|
hass_storage: dict[str, Any],
|
|
issue_registry: ir.IssueRegistry,
|
|
) -> None:
|
|
"""Test the orphaned_config_entry issue confirm step."""
|
|
assert await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
assert await async_setup_component(hass, REPAIRS_DOMAIN, {REPAIRS_DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
await async_process_repairs_platforms(hass)
|
|
http_client = await hass_client()
|
|
|
|
entry = MockConfigEntry(domain="test_issued", source=config_entries.SOURCE_IGNORE)
|
|
entry_valid = MockConfigEntry(domain="test_valid")
|
|
issue_id = f"orphaned_ignored_entry.{entry.entry_id}"
|
|
|
|
hass_storage[config_entries.STORAGE_KEY] = {
|
|
"version": 1,
|
|
"minor_version": 5,
|
|
"data": {
|
|
"entries": [
|
|
{
|
|
"created_at": entry.created_at.isoformat(),
|
|
"data": {},
|
|
"disabled_by": None,
|
|
"discovery_keys": {},
|
|
"domain": "test_issued",
|
|
"entry_id": entry.entry_id,
|
|
"minor_version": 1,
|
|
"modified_at": entry.modified_at.isoformat(),
|
|
"options": {},
|
|
"pref_disable_new_entities": False,
|
|
"pref_disable_polling": False,
|
|
"source": "ignore",
|
|
"subentries": [],
|
|
"title": "Title probably no-one will read",
|
|
"unique_id": None,
|
|
"version": 1,
|
|
},
|
|
{
|
|
"created_at": entry_valid.created_at.isoformat(),
|
|
"data": {},
|
|
"disabled_by": None,
|
|
"discovery_keys": {},
|
|
"domain": "test_valid",
|
|
"entry_id": entry_valid.entry_id,
|
|
"minor_version": 1,
|
|
"modified_at": entry_valid.modified_at.isoformat(),
|
|
"options": {},
|
|
"pref_disable_new_entities": False,
|
|
"pref_disable_polling": False,
|
|
"source": "user",
|
|
"subentries": [],
|
|
"title": "Title probably no-one will read",
|
|
"unique_id": None,
|
|
"version": 1,
|
|
},
|
|
]
|
|
},
|
|
}
|
|
|
|
await hass.config_entries.async_initialize()
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
|
|
await hass.async_block_till_done()
|
|
|
|
issue = issue_registry.async_get_issue(HOMEASSISTANT_DOMAIN, issue_id)
|
|
assert issue is not None
|
|
assert issue.translation_placeholders == {"domain": "test_issued"}
|
|
|
|
data = await start_repair_fix_flow(http_client, HOMEASSISTANT_DOMAIN, issue_id)
|
|
|
|
flow_id = data["flow_id"]
|
|
assert data["step_id"] == "init"
|
|
assert data["description_placeholders"] == {
|
|
"entry_id": entry.entry_id,
|
|
"domain": "test_issued",
|
|
}
|
|
|
|
data = await process_repair_fix_flow(http_client, flow_id)
|
|
|
|
assert data["type"] == "menu"
|
|
|
|
# Apply fix
|
|
data = await process_repair_fix_flow(
|
|
http_client, flow_id, json={"next_step_id": "confirm"}
|
|
)
|
|
|
|
assert data["type"] == "create_entry"
|
|
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.config_entries.async_get_entry(entry.entry_id) is None
|
|
assert hass.config_entries.async_get_entry(entry_valid.entry_id) is not None
|
|
|
|
assert not issue_registry.async_get_issue(HOMEASSISTANT_DOMAIN, issue_id)
|
|
|
|
|
|
async def test_orphaned_config_entry_ignore_step(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSessionGenerator,
|
|
hass_storage: dict[str, Any],
|
|
issue_registry: ir.IssueRegistry,
|
|
) -> None:
|
|
"""Test the orphaned_config_entry issue ignore step."""
|
|
assert await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
assert await async_setup_component(hass, REPAIRS_DOMAIN, {REPAIRS_DOMAIN: {}})
|
|
await hass.async_block_till_done()
|
|
|
|
await async_process_repairs_platforms(hass)
|
|
http_client = await hass_client()
|
|
|
|
entry = MockConfigEntry(domain="test_issued", source=config_entries.SOURCE_IGNORE)
|
|
entry_valid = MockConfigEntry(domain="test_valid")
|
|
issue_id = f"orphaned_ignored_entry.{entry.entry_id}"
|
|
|
|
hass_storage[config_entries.STORAGE_KEY] = {
|
|
"version": 1,
|
|
"minor_version": 5,
|
|
"data": {
|
|
"entries": [
|
|
{
|
|
"created_at": entry.created_at.isoformat(),
|
|
"data": {},
|
|
"disabled_by": None,
|
|
"discovery_keys": {},
|
|
"domain": "test_issued",
|
|
"entry_id": entry.entry_id,
|
|
"minor_version": 1,
|
|
"modified_at": entry.modified_at.isoformat(),
|
|
"options": {},
|
|
"pref_disable_new_entities": False,
|
|
"pref_disable_polling": False,
|
|
"source": "ignore",
|
|
"subentries": [],
|
|
"title": "Title probably no-one will read",
|
|
"unique_id": None,
|
|
"version": 1,
|
|
},
|
|
{
|
|
"created_at": entry_valid.created_at.isoformat(),
|
|
"data": {},
|
|
"disabled_by": None,
|
|
"discovery_keys": {},
|
|
"domain": "test_valid",
|
|
"entry_id": entry_valid.entry_id,
|
|
"minor_version": 1,
|
|
"modified_at": entry_valid.modified_at.isoformat(),
|
|
"options": {},
|
|
"pref_disable_new_entities": False,
|
|
"pref_disable_polling": False,
|
|
"source": "user",
|
|
"subentries": [],
|
|
"title": "Title probably no-one will read",
|
|
"unique_id": None,
|
|
"version": 1,
|
|
},
|
|
]
|
|
},
|
|
}
|
|
|
|
await hass.config_entries.async_initialize()
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
|
|
await hass.async_block_till_done()
|
|
|
|
issue = issue_registry.async_get_issue(HOMEASSISTANT_DOMAIN, issue_id)
|
|
assert issue is not None
|
|
assert issue.translation_placeholders == {"domain": "test_issued"}
|
|
|
|
data = await start_repair_fix_flow(http_client, HOMEASSISTANT_DOMAIN, issue_id)
|
|
|
|
flow_id = data["flow_id"]
|
|
assert data["step_id"] == "init"
|
|
assert data["description_placeholders"] == {
|
|
"entry_id": entry.entry_id,
|
|
"domain": "test_issued",
|
|
}
|
|
|
|
data = await process_repair_fix_flow(http_client, flow_id)
|
|
|
|
assert data["type"] == "menu"
|
|
|
|
# Apply fix
|
|
data = await process_repair_fix_flow(
|
|
http_client, flow_id, json={"next_step_id": "ignore"}
|
|
)
|
|
|
|
assert data["type"] == "abort"
|
|
assert data["reason"] == "issue_ignored"
|
|
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.config_entries.async_get_entry(entry.entry_id)
|
|
|
|
# Assert the issue is resolved
|
|
issue = issue_registry.async_get_issue(HOMEASSISTANT_DOMAIN, issue_id)
|
|
assert issue is not None
|
|
assert issue.dismissed_version is not None
|