evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Clean up exposed domains (#118753)

Browse source 
* Remove lock and script

* Add media player

* Fix tests
This commit is contained in:
Michael Hansen 2024-06-03 21:26:40 -05:00 • committed by GitHub
parent 35a1ecea27
commit 0257aa4839
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 27 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/components/homeassistant/test_exposed_entities.py
View file

@ -57,9 +57,12 @@ def entities_unique_id(entity_registry: er.EntityRegistry) -> dict[str, str]:
entry_sensor_temperature = entity_registry.async_get_or_create(
"sensor",
"test",
"unique2",
"unique3",
original_device_class="temperature",
)
entry_media_player = entity_registry.async_get_or_create(
"media_player", "test", "unique4", original_device_class="media_player"
)
return {
"blocked": entry_blocked.entity_id,
"lock": entry_lock.entity_id,
@ -67,6 +70,7 @@ def entities_unique_id(entity_registry: er.EntityRegistry) -> dict[str, str]:
"door_sensor": entry_binary_sensor_door.entity_id,
"sensor": entry_sensor.entity_id,
"temperature_sensor": entry_sensor_temperature.entity_id,
"media_player": entry_media_player.entity_id,
}
@ -78,10 +82,12 @@ def entities_no_unique_id(hass: HomeAssistant) -> dict[str, str]:
door_sensor = "binary_sensor.door"
sensor = "sensor.test"
sensor_temperature = "sensor.temperature"
media_player = "media_player.test"
hass.states.async_set(binary_sensor, "on", {})
hass.states.async_set(door_sensor, "on", {"device_class": "door"})
hass.states.async_set(sensor, "on", {})
hass.states.async_set(sensor_temperature, "on", {"device_class": "temperature"})
hass.states.async_set(media_player, "idle", {})
return {
"blocked": blocked,
"lock": lock,
@ -89,6 +95,7 @@ def entities_no_unique_id(hass: HomeAssistant) -> dict[str, str]:
"door_sensor": door_sensor,
"sensor": sensor,
"temperature_sensor": sensor_temperature,
"media_player": media_player,
}
@ -409,8 +416,8 @@ async def test_should_expose(
# Blocked entity is not exposed
assert async_should_expose(hass, "cloud.alexa", entities["blocked"]) is False
# Lock is exposed
assert async_should_expose(hass, "cloud.alexa", entities["lock"]) is True
# Lock is not exposed
assert async_should_expose(hass, "cloud.alexa", entities["lock"]) is False
# Binary sensor without device class is not exposed
assert async_should_expose(hass, "cloud.alexa", entities["binary_sensor"]) is False
@ -426,6 +433,9 @@ async def test_should_expose(
async_should_expose(hass, "cloud.alexa", entities["temperature_sensor"]) is True
)
# Media player is exposed
assert async_should_expose(hass, "cloud.alexa", entities["media_player"]) is True
# The second time we check, it should load it from storage
assert (
async_should_expose(hass, "cloud.alexa", entities["temperature_sensor"]) is True