home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-07-29 18:28:14 +02:00
Code Issues Packages Projects Releases Wiki Activity

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
1 changed files with 2 additions and 1 deletions
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(