From 29b442c89a90c9dabab591377c9924c77900910b Mon Sep 17 00:00:00 2001
From: EnjoyingM <6302356+mtielen@users.noreply.github.com>
Date: Fri, 21 Feb 2025 22:04:06 +0000
Subject: [PATCH] Fix test

---
 tests/components/wolflink/test_sensor.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/tests/components/wolflink/test_sensor.py b/tests/components/wolflink/test_sensor.py
index 20941905b74..4684bbdca42 100644
--- a/tests/components/wolflink/test_sensor.py
+++ b/tests/components/wolflink/test_sensor.py
@@ -4,7 +4,12 @@ from unittest.mock import MagicMock
 
 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 (
     WolfLinkEnergy,
     WolfLinkHours,
@@ -49,6 +54,14 @@ async def mock_config_entry(
     )
     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:
     """Test WolflinkSensor initialization."""