From 25ba046ff1e0538955da15943fd6e2854aca8bd3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=81lvaro=20Fern=C3=A1ndez=20Rojas?= <noltari@gmail.com>
Date: Sun, 3 Mar 2024 10:56:14 +0000
Subject: [PATCH] Fix flakey airzone_cloud tests by avoiding creation of the
 websocket (#112102)

---
 tests/components/airzone_cloud/conftest.py        | 15 +++++++++++++++
 .../components/airzone_cloud/test_coordinator.py  |  3 ---
 2 files changed, 15 insertions(+), 3 deletions(-)
 create mode 100644 tests/components/airzone_cloud/conftest.py

diff --git a/tests/components/airzone_cloud/conftest.py b/tests/components/airzone_cloud/conftest.py
new file mode 100644
index 00000000000..d810c808fde
--- /dev/null
+++ b/tests/components/airzone_cloud/conftest.py
@@ -0,0 +1,15 @@
+"""Tests for the Airzone integration."""
+
+from unittest.mock import patch
+
+import pytest
+
+
+@pytest.fixture(autouse=True)
+def airzone_cloud_no_websockets():
+    """Fixture to completely disable Airzone Cloud WebSockets."""
+    with patch(
+        "homeassistant.components.airzone_cloud.AirzoneCloudApi._update_websockets",
+        return_value=False,
+    ), patch("aioairzone_cloud.websockets.AirzoneCloudIWS.connect", return_value=True):
+        yield
diff --git a/tests/components/airzone_cloud/test_coordinator.py b/tests/components/airzone_cloud/test_coordinator.py
index a2307b94335..40b6c937ed2 100644
--- a/tests/components/airzone_cloud/test_coordinator.py
+++ b/tests/components/airzone_cloud/test_coordinator.py
@@ -46,9 +46,6 @@ async def test_coordinator_client_connector_error(hass: HomeAssistant) -> None:
     ) as mock_webserver, patch(
         "homeassistant.components.airzone_cloud.AirzoneCloudApi.login",
         return_value=None,
-    ), patch(
-        "homeassistant.components.airzone_cloud.AirzoneCloudApi._update_websockets",
-        return_value=False,
     ):
         await hass.config_entries.async_setup(config_entry.entry_id)
         await hass.async_block_till_done()