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

Add test: warn entity_category assigned as str (#65142)

Browse Source
This commit is contained in:
Dave T 2022-01-28 20:46:57 +00:00 committed by GitHub
parent 7d949a7665
commit bf910229b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tests/helpers/test_entity_registry.py
View File

@ -9,6 +9,7 @@ from homeassistant.const import EVENT_HOMEASSISTANT_START, STATE_UNAVAILABLE
from homeassistant.core import CoreState, callback, valid_entity_id from homeassistant.core import CoreState, callback, valid_entity_id
from homeassistant.exceptions import MaxLengthExceeded from homeassistant.exceptions import MaxLengthExceeded
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.entity import EntityCategory
from tests.common import ( from tests.common import (
MockConfigEntry, MockConfigEntry,
@ -77,7 +78,7 @@ def test_get_or_create_updates_data(registry):
config_entry=orig_config_entry, config_entry=orig_config_entry,
device_id="mock-dev-id", device_id="mock-dev-id",
disabled_by=er.RegistryEntryDisabler.HASS, disabled_by=er.RegistryEntryDisabler.HASS,
entity_category="config", entity_category=EntityCategory.CONFIG,
original_device_class="mock-device-class", original_device_class="mock-device-class",
original_icon="initial-original_icon", original_icon="initial-original_icon",
original_name="initial-original_name", original_name="initial-original_name",
@ -95,7 +96,7 @@ def test_get_or_create_updates_data(registry):
device_class=None, device_class=None,
device_id="mock-dev-id", device_id="mock-dev-id",
disabled_by=er.RegistryEntryDisabler.HASS, disabled_by=er.RegistryEntryDisabler.HASS,
entity_category="config", entity_category=EntityCategory.CONFIG,
icon=None, icon=None,
id=orig_entry.id, id=orig_entry.id,
name=None, name=None,
@ -135,7 +136,7 @@ def test_get_or_create_updates_data(registry):
device_class=None, device_class=None,
device_id="new-mock-dev-id", device_id="new-mock-dev-id",
disabled_by=er.RegistryEntryDisabler.HASS, # Should not be updated disabled_by=er.RegistryEntryDisabler.HASS, # Should not be updated
entity_category="config", entity_category=EntityCategory.CONFIG,
icon=None, icon=None,
id=orig_entry.id, id=orig_entry.id,
name=None, name=None,
@ -189,7 +190,7 @@ async def test_loading_saving_data(hass, registry):
config_entry=mock_config, config_entry=mock_config,
device_id="mock-dev-id", device_id="mock-dev-id",
disabled_by=er.RegistryEntryDisabler.HASS, disabled_by=er.RegistryEntryDisabler.HASS,
entity_category="config", entity_category=EntityCategory.CONFIG,
original_device_class="mock-device-class", original_device_class="mock-device-class",
original_icon="hass:original-icon", original_icon="hass:original-icon",
original_name="Original Name", original_name="Original Name",
@ -1124,3 +1125,16 @@ async def test_deprecated_disabled_by_str(hass, registry, caplog):
assert entry.disabled_by is er.RegistryEntryDisabler.USER assert entry.disabled_by is er.RegistryEntryDisabler.USER
assert " str for entity registry disabled_by. This is deprecated " in caplog.text assert " str for entity registry disabled_by. This is deprecated " in caplog.text
async def test_deprecated_entity_category_str(hass, registry, caplog):
"""Test deprecated str use of entity_category converts to enum and logs a warning."""
entry = er.RegistryEntry(
"light",
"hue",
"5678",
entity_category="diagnostic",
)
assert entry.entity_category is EntityCategory.DIAGNOSTIC
assert " should be updated to use EntityCategory" in caplog.text