From 7b08e625b42e06560f69b77023043f140315b142 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Mon, 29 Jul 2024 22:14:41 +0200
Subject: [PATCH] Fix implicit-return in websocket_api tests (#122779)

---
 tests/components/websocket_api/test_connection.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/components/websocket_api/test_connection.py b/tests/components/websocket_api/test_connection.py
index 5740bb48019..343575e5b4a 100644
--- a/tests/components/websocket_api/test_connection.py
+++ b/tests/components/websocket_api/test_connection.py
@@ -77,13 +77,15 @@ async def test_exception_handling(
     refresh_token = Mock()
     hass.data[DOMAIN] = {}
 
-    def get_extra_info(key: str) -> Any:
+    def get_extra_info(key: str) -> Any | None:
         if key == "sslcontext":
             return True
 
         if key == "peername":
             return ("127.0.0.42", 8123)
 
+        return None
+
     mocked_transport = Mock()
     mocked_transport.get_extra_info = get_extra_info
     mocked_request = make_mocked_request(