From 6de824e875d44dba0a49548c904edbdfbb84dea1 Mon Sep 17 00:00:00 2001
From: Marc Mueller <30130371+cdce8p@users.noreply.github.com>
Date: Sun, 21 Jul 2024 18:50:00 +0200
Subject: [PATCH] Fix test RuntimeWarning for upb (#122325)
---
tests/components/upb/test_config_flow.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/tests/components/upb/test_config_flow.py b/tests/components/upb/test_config_flow.py
index 54aeb00e89a..efa6d60c344 100644
--- a/tests/components/upb/test_config_flow.py
+++ b/tests/components/upb/test_config_flow.py
@@ -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(
--
GitLab