From 3ccdbc342ad939f264d6cb15bf26a7be515b5fd1 Mon Sep 17 00:00:00 2001
From: Thomas Protzner <thomas.protzner@gmail.com>
Date: Mon, 12 Aug 2024 17:43:42 +0000
Subject: [PATCH] test TimeoutException

---
 tests/components/husqvarna_automower/test_init.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/tests/components/husqvarna_automower/test_init.py b/tests/components/husqvarna_automower/test_init.py
index df6e192c1de..50abd4c92fc 100644
--- a/tests/components/husqvarna_automower/test_init.py
+++ b/tests/components/husqvarna_automower/test_init.py
@@ -9,6 +9,7 @@ from aioautomower.exceptions import (
     ApiException,
     AuthException,
     HusqvarnaWSServerHandshakeError,
+    TimeoutException,
 )
 from freezegun.api import FrozenDateTimeFactory
 import pytest
@@ -158,10 +159,8 @@ async def test_websocket_not_available(
         await hass.async_block_till_done()
         assert mock_automower_client.start_listening.call_count == 1
 
-        # Test reconnection after another disconnect
-        mock_automower_client.auth.websocket_connect.side_effect = (
-            HusqvarnaWSServerHandshakeError("Boom")
-        )
+        # Simulate a start_listening TimeoutException
+        mock_automower_client.start_listening.side_effect = TimeoutException("Boom")
         await hass.async_block_till_done()
         assert (
             mock_automower_client.auth.websocket_connect.call_count