Add model ID to unifiprotect (#125376)

This commit is contained in:
Paulus Schoutsen 2024-09-05 19:44:28 -04:00 committed by GitHub
parent aa619c5594
commit c3921f2112
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 2 deletions

View File

@ -278,7 +278,8 @@ class ProtectDeviceEntity(BaseProtectEntity):
self._attr_device_info = DeviceInfo(
name=self.device.display_name,
manufacturer=DEFAULT_BRAND,
model=self.device.type,
model=self.device.market_name or self.device.type,
model_id=self.device.type,
via_device=(DOMAIN, self.data.api.bootstrap.nvr.mac),
sw_version=self.device.firmware_version,
connections={(dr.CONNECTION_NETWORK_MAC, self.device.mac)},

View File

@ -32,7 +32,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.setup import async_setup_component
from .utils import (
@ -66,6 +66,14 @@ def validate_default_camera_entity(
assert entity.disabled is False
assert entity.unique_id == unique_id
device_registry = dr.async_get(hass)
device = device_registry.async_get(entity.device_id)
assert device
assert device.manufacturer == "Ubiquiti"
assert device.name == camera_obj.name
assert device.model == camera_obj.market_name or camera_obj.type
assert device.model_id == camera_obj.type
return entity_id