diff --git a/homeassistant/components/homewizard/config_flow.py b/homeassistant/components/homewizard/config_flow.py index 45a912fefec..c9f7d19a96a 100644 --- a/homeassistant/components/homewizard/config_flow.py +++ b/homeassistant/components/homewizard/config_flow.py @@ -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"]}) diff --git a/tests/components/homewizard/test_init.py b/tests/components/homewizard/test_init.py index 87a02a446e9..02e0b5c0c23 100644 --- a/tests/components/homewizard/test_init.py +++ b/tests/components/homewizard/test_init.py @@ -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",