jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-15 09:17:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Unregister Google sheets services during unload (#79314)

Browse Source 
* Unregister services during unload - Google Sheets

* uno mas
This commit is contained in:
Robert Hillis 2022-09-29 14:29:41 -04:00 committed by GitHub
parent ba6a81c565
commit d7d6637d79
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
homeassistant/components/google_sheets/__init__.py
View File

@ -10,7 +10,7 @@ from google.oauth2.credentials import Credentials
from gspread import Client from gspread import Client
import voluptuous as vol import voluptuous as vol
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry, ConfigEntryState
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_TOKEN from homeassistant.const import CONF_ACCESS_TOKEN, CONF_TOKEN
from homeassistant.core import HomeAssistant, ServiceCall from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@ -69,6 +69,15 @@ def async_entry_has_scopes(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry.""" """Unload a config entry."""
hass.data[DOMAIN].pop(entry.entry_id) hass.data[DOMAIN].pop(entry.entry_id)
loaded_entries = [
entry
for entry in hass.config_entries.async_entries(DOMAIN)
if entry.state == ConfigEntryState.LOADED
]
if len(loaded_entries) == 1:
for service_name in hass.services.async_services()[DOMAIN]:
hass.services.async_remove(DOMAIN, service_name)
return True return True

1
tests/components/google_sheets/test_init.py
View File

@ -80,6 +80,7 @@ async def mock_setup_integration(
assert len(entries) == 1 assert len(entries) == 1
await hass.config_entries.async_unload(entries[0].entry_id) await hass.config_entries.async_unload(entries[0].entry_id)
await hass.async_block_till_done() await hass.async_block_till_done()
assert not len(hass.services.async_services().get(DOMAIN, {}))
assert not hass.data.get(DOMAIN) assert not hass.data.get(DOMAIN)
assert entries[0].state is ConfigEntryState.NOT_LOADED assert entries[0].state is ConfigEntryState.NOT_LOADED