From ddf89b336c630fac4dfa27ba11556ce691e4bc3c Mon Sep 17 00:00:00 2001 From: Josef Zweck Date: Sun, 9 Aug 2026 09:38:55 +0200 Subject: [PATCH] Async fixture for stiebel_eltron (#178580) --- tests/components/stiebel_eltron/conftest.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/components/stiebel_eltron/conftest.py b/tests/components/stiebel_eltron/conftest.py index 395d8ed90947..dcef3caf1a6d 100644 --- a/tests/components/stiebel_eltron/conftest.py +++ b/tests/components/stiebel_eltron/conftest.py @@ -1,6 +1,6 @@ """Common fixtures for the STIEBEL ELTRON tests.""" -from collections.abc import Generator +from collections.abc import AsyncGenerator, Generator from unittest.mock import AsyncMock, MagicMock, patch from modbus_connection.mock import MockModbusConnection @@ -32,10 +32,11 @@ def mock_get_controller_model() -> Generator[MagicMock]: @pytest.fixture(autouse=True) -def mock_connect_tcp( +async def mock_connect_tcp( mock_modbus_connection: MockModbusConnection, -) -> Generator[AsyncMock]: +) -> AsyncGenerator[AsyncMock]: """Patch connect_tcp to return the in-memory mock connection.""" + await mock_modbus_connection.connect() connect = AsyncMock(return_value=mock_modbus_connection) with ( patch("homeassistant.components.stiebel_eltron.connect_tcp", new=connect),