Close DNS resolvers on setup failure and unload

Author: Phil-Rad

homeassistant/components/dnsip/__init__.py

@@ -57,6 +57,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: DnsIPConfigEntry) -> boo
             elif entry.data[CONF_IPV6]:
                 await resolver_ipv6.query(hostname, "AAAA")
     except (TimeoutError, DNSError) as err:
+        await resolver_ipv4.close()
+        await resolver_ipv6.close()
         raise ConfigEntryNotReady(f"DNS lookup failed for {hostname}: {err}") from err
 
     entry.runtime_data = DnsIPRuntimeData(
@@ -71,7 +73,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: DnsIPConfigEntry) -> boo
 async def async_unload_entry(hass: HomeAssistant, entry: DnsIPConfigEntry) -> bool:
     """Unload DNS IP config entry."""
 
-    return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
+    unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
+    if unload_ok:
+        await entry.runtime_data.resolver_ipv4.close()
+        await entry.runtime_data.resolver_ipv6.close()
+    return unload_ok
 
 
 async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: