From a3ad2526ef714c7fdd8b9a83a8858e3effd52b92 Mon Sep 17 00:00:00 2001 From: denpamusic Date: Thu, 5 Oct 2023 20:42:57 +0300 Subject: [PATCH] Fix DeviceEntry mock. --- tests/test_init.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/test_init.py b/tests/test_init.py index 783bb4b2..78769423 100644 --- a/tests/test_init.py +++ b/tests/test_init.py @@ -13,7 +13,6 @@ ) from homeassistant.core import HomeAssistant from homeassistant.exceptions import ConfigEntryNotReady -from homeassistant.helpers.device_registry import DeviceEntry from homeassistant.util import dt as dt_util from pyplumio.const import AlertType from pyplumio.structures.alerts import Alert @@ -123,7 +122,7 @@ async def test_setup_events( from_dt=dt_util.parse_datetime(DATE_FROM), to_dt=dt_util.parse_datetime(DATE_TO), ) - mock_device_entry = Mock(spec=DeviceEntry) + mock_device_entry = Mock() with patch( "homeassistant.helpers.device_registry.DeviceRegistry.async_get_device",