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

Remove old import logic for google_travel_time (#80018)

Browse Source 
Remove old import logic
This commit is contained in:
Kevin Stillhammer 2022-10-11 09:49:06 +02:00 committed by GitHub
parent 4e5b5dfb93
commit edad6d0f26
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 0 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2,23 +2,12 @@
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_registry import (
async_entries_for_config_entry,
async_get,
)
PLATFORMS = [Platform.SENSOR]
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Google Maps Travel Time from a config entry."""
if entry.unique_id is not None:
hass.config_entries.async_update_entry(entry, unique_id=None)
ent_reg = async_get(hass)
for entity in async_entries_for_config_entry(ent_reg, entry.entry_id):
ent_reg.async_update_entity(entity.entity_id, new_unique_id=entry.entry_id)
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True

21
tests/components/google_travel_time/test_init.py
View File

@ -1,21 +0,0 @@
"""Test Google Maps Travel Time initialization."""
from homeassistant.components.google_travel_time.const import DOMAIN
from homeassistant.helpers.entity_registry import async_get
from tests.common import MockConfigEntry
async def test_migration(hass, bypass_platform_setup):
"""Test migration logic for unique id."""
config_entry = MockConfigEntry(
domain=DOMAIN, version=1, entry_id="test", unique_id="test"
)
ent_reg = async_get(hass)
ent_entry = ent_reg.async_get_or_create(
"sensor", DOMAIN, unique_id="replaceable_unique_id", config_entry=config_entry
)
entity_id = ent_entry.entity_id
config_entry.add_to_hass(hass)
await hass.config_entries.async_setup(config_entry.entry_id)
assert config_entry.unique_id is None
assert ent_reg.async_get(entity_id).unique_id == config_entry.entry_id