Fix implicit-return in websocket_api tests (#122779)

This commit is contained in:
epenet 2024-07-29 22:14:41 +02:00 committed by GitHub
parent 2102a104d2
commit 7b08e625b4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -77,13 +77,15 @@ async def test_exception_handling(
refresh_token = Mock() refresh_token = Mock()
hass.data[DOMAIN] = {} hass.data[DOMAIN] = {}
def get_extra_info(key: str) -> Any: def get_extra_info(key: str) -> Any | None:
if key == "sslcontext": if key == "sslcontext":
return True return True
if key == "peername": if key == "peername":
return ("127.0.0.42", 8123) return ("127.0.0.42", 8123)
return None
mocked_transport = Mock() mocked_transport = Mock()
mocked_transport.get_extra_info = get_extra_info mocked_transport.get_extra_info = get_extra_info
mocked_request = make_mocked_request( mocked_request = make_mocked_request(