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