Fix test RuntimeWarning for upb (#122325)

This commit is contained in:
Marc Mueller 2024-07-21 18:50:00 +02:00 committed by GitHub
parent 273dc0998f
commit 6de824e875
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,7 +1,7 @@
"""Test the UPB Control config flow."""
from asyncio import TimeoutError
from unittest.mock import AsyncMock, PropertyMock, patch
from unittest.mock import AsyncMock, MagicMock, PropertyMock, patch
from homeassistant import config_entries
from homeassistant.components.upb.const import DOMAIN
@ -18,6 +18,7 @@ def mocked_upb(sync_complete=True, config_ok=True):
upb_mock = AsyncMock()
type(upb_mock).network_id = PropertyMock(return_value="42")
type(upb_mock).config_ok = PropertyMock(return_value=config_ok)
type(upb_mock).disconnect = MagicMock()
if sync_complete:
upb_mock.async_connect.side_effect = _upb_lib_connect
return patch(