This commit is contained in:
EnjoyingM
2025-02-21 22:04:06 +00:00
parent 3f89a3bdfa
commit 29b442c89a

View File

@@ -4,7 +4,12 @@ from unittest.mock import MagicMock
import pytest import pytest
from homeassistant.components.wolflink.const import COORDINATOR, DEVICE_ID, DOMAIN from homeassistant.components.wolflink.const import (
COORDINATOR,
DEVICE_ID,
DOMAIN,
MANUFACTURER,
)
from homeassistant.components.wolflink.sensor import ( from homeassistant.components.wolflink.sensor import (
WolfLinkEnergy, WolfLinkEnergy,
WolfLinkHours, WolfLinkHours,
@@ -49,6 +54,14 @@ async def mock_config_entry(
) )
config_entry.add_to_hass(hass) config_entry.add_to_hass(hass)
device_id = device_registry.async_get_or_create(
config_entry_id=config_entry.entry_id,
identifiers={(DOMAIN, CONFIG[DEVICE_ID])},
configuration_url="https://www.wolf-smartset.com/",
manufacturer=MANUFACTURER,
).id
assert device_registry.async_get(device_id).identifiers == {(DOMAIN, "1234")}
def test_wolflink_sensor_initialization(mock_coordinator) -> None: def test_wolflink_sensor_initialization(mock_coordinator) -> None:
"""Test WolflinkSensor initialization.""" """Test WolflinkSensor initialization."""