mirror of
https://github.com/esphome/esphome.git
synced 2025-09-11 07:42:26 +01:00
130 lines
3.9 KiB
Python
130 lines
3.9 KiB
Python
"""Test duplicate entity object ID handling."""
|
|
|
|
import pytest
|
|
|
|
from esphome.core import CORE
|
|
from esphome.helpers import sanitize, snake_case
|
|
|
|
|
|
@pytest.fixture
|
|
def setup_test_device() -> None:
|
|
"""Set up test device configuration."""
|
|
CORE.name = "test-device"
|
|
CORE.friendly_name = "Test Device"
|
|
|
|
|
|
def test_unique_key_generation() -> None:
|
|
"""Test that unique keys are generated correctly."""
|
|
# Test with no device
|
|
key1: tuple[int, str, str] = (0, "binary_sensor", "temperature")
|
|
assert key1 == (0, "binary_sensor", "temperature")
|
|
|
|
# Test with device
|
|
key2: tuple[int, str, str] = (12345, "sensor", "humidity")
|
|
assert key2 == (12345, "sensor", "humidity")
|
|
|
|
|
|
def test_duplicate_tracking() -> None:
|
|
"""Test that duplicates are tracked correctly."""
|
|
# First occurrence
|
|
key: tuple[int, str, str] = (0, "sensor", "temperature")
|
|
assert key not in CORE.unique_ids
|
|
|
|
CORE.unique_ids[key] = 1
|
|
assert CORE.unique_ids[key] == 1
|
|
|
|
# Second occurrence
|
|
count: int = CORE.unique_ids[key] + 1
|
|
CORE.unique_ids[key] = count
|
|
assert CORE.unique_ids[key] == 2
|
|
|
|
|
|
def test_object_id_sanitization() -> None:
|
|
"""Test that object IDs are properly sanitized."""
|
|
# Test various inputs
|
|
assert sanitize(snake_case("Temperature Sensor")) == "temperature_sensor"
|
|
assert sanitize(snake_case("Living Room Light!")) == "living_room_light_"
|
|
assert sanitize(snake_case("Test-Device")) == "test-device"
|
|
assert sanitize(snake_case("")) == ""
|
|
|
|
|
|
def test_suffix_generation() -> None:
|
|
"""Test that suffixes are generated correctly."""
|
|
base_id: str = "temperature"
|
|
|
|
# No suffix for first occurrence
|
|
object_id_1: str = base_id
|
|
assert object_id_1 == "temperature"
|
|
|
|
# Add suffix for duplicates
|
|
count: int = 2
|
|
object_id_2: str = f"{base_id}_{count}"
|
|
assert object_id_2 == "temperature_2"
|
|
|
|
count = 3
|
|
object_id_3: str = f"{base_id}_{count}"
|
|
assert object_id_3 == "temperature_3"
|
|
|
|
|
|
def test_different_platforms_same_name() -> None:
|
|
"""Test that same name on different platforms doesn't conflict."""
|
|
# Simulate two entities with same name on different platforms
|
|
key1: tuple[int, str, str] = (0, "binary_sensor", "status")
|
|
key2: tuple[int, str, str] = (0, "text_sensor", "status")
|
|
|
|
# They should be different keys
|
|
assert key1 != key2
|
|
|
|
# Track them separately
|
|
CORE.unique_ids[key1] = 1
|
|
CORE.unique_ids[key2] = 1
|
|
|
|
# Both should be at count 1 (no conflict)
|
|
assert CORE.unique_ids[key1] == 1
|
|
assert CORE.unique_ids[key2] == 1
|
|
|
|
|
|
def test_different_devices_same_name_platform() -> None:
|
|
"""Test that same name+platform on different devices doesn't conflict."""
|
|
# Simulate two entities with same name and platform but different devices
|
|
key1: tuple[int, str, str] = (12345, "sensor", "temperature")
|
|
key2: tuple[int, str, str] = (67890, "sensor", "temperature")
|
|
|
|
# They should be different keys
|
|
assert key1 != key2
|
|
|
|
# Track them separately
|
|
CORE.unique_ids[key1] = 1
|
|
CORE.unique_ids[key2] = 1
|
|
|
|
# Both should be at count 1 (no conflict)
|
|
assert CORE.unique_ids[key1] == 1
|
|
assert CORE.unique_ids[key2] == 1
|
|
|
|
|
|
def test_empty_name_handling(setup_test_device: None) -> None:
|
|
"""Test handling of entities with empty names."""
|
|
# When name is empty, it should use the device name
|
|
empty_name: str = ""
|
|
base_id: str
|
|
if not empty_name:
|
|
if CORE.friendly_name:
|
|
base_id = sanitize(snake_case(CORE.friendly_name))
|
|
else:
|
|
base_id = sanitize(snake_case(CORE.name))
|
|
|
|
assert base_id == "test_device" # Uses friendly name
|
|
|
|
|
|
def test_reset_clears_unique_ids() -> None:
|
|
"""Test that CORE.reset() clears the unique_ids tracking."""
|
|
# Add some tracked IDs
|
|
CORE.unique_ids[(0, "sensor", "test")] = 2
|
|
CORE.unique_ids[(0, "binary_sensor", "test")] = 3
|
|
|
|
assert len(CORE.unique_ids) == 2
|
|
|
|
# Reset should clear them
|
|
CORE.reset()
|
|
assert len(CORE.unique_ids) == 0
|