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(