From f41bc98fe2dad97e3008a6f5d955808900800d90 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Thu, 28 Nov 2024 12:40:34 +0100
Subject: [PATCH] Cleanup deprecated exception in websocket tests (#131808)

---
 tests/components/samsungtv/test_config_flow.py | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/tests/components/samsungtv/test_config_flow.py b/tests/components/samsungtv/test_config_flow.py
index 32e169ffb24..3a849c9d4b1 100644
--- a/tests/components/samsungtv/test_config_flow.py
+++ b/tests/components/samsungtv/test_config_flow.py
@@ -15,13 +15,8 @@ from samsungtvws.exceptions import (
 )
 from websockets import frames
 
-# WebSocketProtocolError was deprecated in websockets '14.0'
 # pylint: disable-next=no-name-in-module
-from websockets.exceptions import (
-    ConnectionClosedError,
-    WebSocketException,
-    WebSocketProtocolError,
-)
+from websockets.exceptions import ConnectionClosedError, WebSocketException
 
 from homeassistant import config_entries
 from homeassistant.components import dhcp, ssdp, zeroconf
@@ -404,7 +399,7 @@ async def test_user_websocket_not_supported(hass: HomeAssistant) -> None:
         ),
         patch(
             "homeassistant.components.samsungtv.bridge.SamsungTVWSAsyncRemote.open",
-            side_effect=WebSocketProtocolError("Boom"),
+            side_effect=WebSocketException("Boom"),
         ),
     ):
         # websocket device not supported
@@ -787,12 +782,12 @@ async def test_ssdp_websocket_cannot_connect(hass: HomeAssistant) -> None:
         ),
         patch(
             "homeassistant.components.samsungtv.bridge.SamsungTVEncryptedWSAsyncRemote.start_listening",
-            side_effect=WebSocketProtocolError("Boom"),
+            side_effect=WebSocketException("Boom"),
         ),
         patch(
             "homeassistant.components.samsungtv.bridge.SamsungTVWSAsyncRemote",
         ) as remotews,
-        patch.object(remotews, "open", side_effect=WebSocketProtocolError("Boom")),
+        patch.object(remotews, "open", side_effect=WebSocketException("Boom")),
     ):
         # device not supported
         result = await hass.config_entries.flow.async_init(
@@ -1742,7 +1737,7 @@ async def test_update_legacy_missing_mac_from_dhcp_no_unique_id(
         ),
         patch(
             "homeassistant.components.samsungtv.bridge.SamsungTVEncryptedWSAsyncRemote.start_listening",
-            side_effect=WebSocketProtocolError("Boom"),
+            side_effect=WebSocketException("Boom"),
         ),
     ):
         result = await hass.config_entries.flow.async_init(