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

Add leak sensor for UP Sense for UniFi Protect (#107762)

Browse source
This commit is contained in:
Christopher Bailey 2024-01-10 20:32:05 -05:00 • committed by GitHub
parent 28b5104cda
commit e595d24d78
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 25 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
homeassistant/components/unifiprotect/binary_sensor.py
View file

@ -259,6 +259,13 @@ SENSE_SENSORS: tuple[ProtectBinaryEntityDescription, ...] = (
ufp_value="is_opened",
ufp_enabled="is_contact_sensor_enabled",
),
ProtectBinaryEntityDescription(
key="leak",
name="Leak",
device_class=BinarySensorDeviceClass.MOISTURE,
ufp_value="is_leak_detected",
ufp_enabled="is_leak_sensor_enabled",
),
ProtectBinaryEntityDescription(
key="battery_low",
name="Battery low",

28
tests/components/unifiprotect/test_binary_sensor.py
View file

@ -78,11 +78,11 @@ async def test_binary_sensor_sensor_remove(
ufp.api.bootstrap.nvr.system_info.ustorage = None
await init_entry(hass, ufp, [sensor_all])
assert_entity_counts(hass, Platform.BINARY_SENSOR, 4, 4)
assert_entity_counts(hass, Platform.BINARY_SENSOR, 5, 5)
await remove_entities(hass, ufp, [sensor_all])
assert_entity_counts(hass, Platform.BINARY_SENSOR, 0, 0)
await adopt_devices(hass, ufp, [sensor_all])
assert_entity_counts(hass, Platform.BINARY_SENSOR, 4, 4)
assert_entity_counts(hass, Platform.BINARY_SENSOR, 5, 5)
async def test_binary_sensor_setup_light(
@ -201,11 +201,18 @@ async def test_binary_sensor_setup_sensor(
"""Test binary_sensor entity setup for sensor devices."""
await init_entry(hass, ufp, [sensor_all])
assert_entity_counts(hass, Platform.BINARY_SENSOR, 10, 10)
assert_entity_counts(hass, Platform.BINARY_SENSOR, 11, 11)
entity_registry = er.async_get(hass)
for description in SENSE_SENSORS_WRITE:
expected = [
STATE_OFF,
STATE_UNAVAILABLE,
STATE_OFF,
STATE_OFF,
STATE_OFF,
]
for index, description in enumerate(SENSE_SENSORS_WRITE):
unique_id, entity_id = ids_from_device_description(
Platform.BINARY_SENSOR, sensor_all, description
)
@ -216,24 +223,25 @@ async def test_binary_sensor_setup_sensor(
state = hass.states.get(entity_id)
assert state
assert state.state == STATE_OFF
assert state.state == expected[index]
assert state.attributes[ATTR_ATTRIBUTION] == DEFAULT_ATTRIBUTION
async def test_binary_sensor_setup_sensor_none(
async def test_binary_sensor_setup_sensor_leak(
hass: HomeAssistant, ufp: MockUFPFixture, sensor: Sensor
) -> None:
"""Test binary_sensor entity setup for sensor with most sensors disabled."""
"""Test binary_sensor entity setup for sensor with most leak mounting type."""
sensor.mount_type = MountType.LEAK
await init_entry(hass, ufp, [sensor])
assert_entity_counts(hass, Platform.BINARY_SENSOR, 10, 10)
assert_entity_counts(hass, Platform.BINARY_SENSOR, 11, 11)
entity_registry = er.async_get(hass)
expected = [
STATE_UNAVAILABLE,
STATE_OFF,
STATE_OFF,
STATE_UNAVAILABLE,
STATE_OFF,
]
@ -348,7 +356,7 @@ async def test_binary_sensor_update_mount_type_window(
"""Test binary_sensor motion entity."""
await init_entry(hass, ufp, [sensor_all])
assert_entity_counts(hass, Platform.BINARY_SENSOR, 10, 10)
assert_entity_counts(hass, Platform.BINARY_SENSOR, 11, 11)
_, entity_id = ids_from_device_description(
Platform.BINARY_SENSOR, sensor_all, SENSE_SENSORS_WRITE[0]
@ -380,7 +388,7 @@ async def test_binary_sensor_update_mount_type_garage(
"""Test binary_sensor motion entity."""
await init_entry(hass, ufp, [sensor_all])
assert_entity_counts(hass, Platform.BINARY_SENSOR, 10, 10)
assert_entity_counts(hass, Platform.BINARY_SENSOR, 11, 11)
_, entity_id = ids_from_device_description(
Platform.BINARY_SENSOR, sensor_all, SENSE_SENSORS_WRITE[0]