evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix test RuntimeWarning for upb (#122325)

Browse source
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
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/components/upb/test_config_flow.py
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(