1
mirror of https://github.com/home-assistant/core synced 2024-10-07 10:13:38 +02:00

Fix homewizard_energy migration issues from #65594 (#65718)

This commit is contained in:
Duco Sebel 2022-02-04 23:45:06 +01:00 committed by GitHub
parent 2370dda1f0
commit 0d3bbfc9a7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 6 deletions

View File

@ -10,7 +10,7 @@ import async_timeout
from voluptuous import Required, Schema
from homeassistant import config_entries
from homeassistant.components import zeroconf
from homeassistant.components import persistent_notification, zeroconf
from homeassistant.const import CONF_IP_ADDRESS
from homeassistant.data_entry_flow import AbortFlow, FlowResult
@ -32,13 +32,14 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
"""Handle a flow initiated by older `homewizard_energy` component."""
_LOGGER.debug("config_flow async_step_import")
self.hass.components.persistent_notification.async_create(
(
persistent_notification.async_create(
self.hass,
title="HomeWizard Energy",
message=(
"The custom integration of HomeWizard Energy has been migrated to core. "
"You can safely remove the custom integration from the custom_integrations folder."
),
"HomeWizard Energy",
f"homewizard_energy_to_{DOMAIN}",
notification_id=f"homewizard_energy_to_{DOMAIN}",
)
return await self.async_step_user({CONF_IP_ADDRESS: import_config["host"]})

View File

@ -77,7 +77,7 @@ async def test_init_accepts_and_migrates_old_entry(aioclient_mock, hass):
# Add original entry
original_entry = MockConfigEntry(
domain=DOMAIN,
domain="homewizard_energy",
data={CONF_IP_ADDRESS: "1.2.3.4"},
entry_id="old_id",
)
@ -122,6 +122,9 @@ async def test_init_accepts_and_migrates_old_entry(aioclient_mock, hass):
)
imported_entry.add_to_hass(hass)
assert imported_entry.domain == DOMAIN
assert imported_entry.domain != original_entry.domain
# Add the entry_id to trigger migration
with patch(
"aiohwenergy.HomeWizardEnergy",