Add model ID to roborock (#125366)

This commit is contained in:
Paulus Schoutsen 2024-09-05 16:52:45 -04:00 committed by GitHub
parent 56b4ddc6b4
commit 006b2da14e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 3 deletions

View File

@ -63,6 +63,7 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
identifiers={(DOMAIN, self.roborock_device_info.device.duid)},
manufacturer="Roborock",
model=self.roborock_device_info.product.model,
model_id=self.roborock_device_info.product.model,
sw_version=self.roborock_device_info.device.fv,
)
self.current_map: int | None = None

View File

@ -24,7 +24,7 @@ from homeassistant.components.vacuum import (
from homeassistant.const import ATTR_ENTITY_ID, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
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 .mock_data import PROP
@ -38,12 +38,17 @@ DEVICE_ID = "abc123"
async def test_registry_entries(
hass: HomeAssistant,
entity_registry: er.EntityRegistry,
device_registry: dr.DeviceRegistry,
bypass_api_fixture,
setup_entry: MockConfigEntry,
) -> None:
"""Tests devices are registered in the entity registry."""
entry = entity_registry.async_get(ENTITY_ID)
assert entry.unique_id == DEVICE_ID
entity_entry = entity_registry.async_get(ENTITY_ID)
assert entity_entry.unique_id == DEVICE_ID
device_entry = device_registry.async_get(entity_entry.device_id)
assert device_entry is not None
assert device_entry.model_id == "roborock.vacuum.a27"
@pytest.mark.parametrize(