evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve performance of websocket_api dispatch (#88496)

Browse source 
Co-authored-by: Paulus Schoutsen <paulus@home-assistant.io>
This commit is contained in:
J. Nick Koston 2023-02-20 12:51:34 -06:00 • committed by GitHub
parent cc4a179ca8
commit ecf87ae979
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 21 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/components/websocket_api/test_connection.py
View file

@ -10,6 +10,8 @@ import voluptuous as vol
from homeassistant import exceptions
from homeassistant.components import websocket_api
from homeassistant.components.websocket_api.const import DOMAIN
from homeassistant.core import HomeAssistant
from tests.common import MockUser
@ -56,6 +58,7 @@ from tests.common import MockUser
],
)
async def test_exception_handling(
hass: HomeAssistant,
caplog: pytest.LogCaptureFixture,
exc: Exception,
code: str,
@ -67,6 +70,7 @@ async def test_exception_handling(
user = MockUser()
refresh_token = Mock()
current_request = AsyncMock()
hass.data[DOMAIN] = {}
def get_extra_info(key: str) -> Any:
if key == "sslcontext":
@ -89,7 +93,7 @@ async def test_exception_handling(
) as current_request:
current_request.get.return_value = mocked_request
conn = websocket_api.ActiveConnection(
logging.getLogger(__name__), None, send_messages.append, user, refresh_token
logging.getLogger(__name__), hass, send_messages.append, user, refresh_token
)
conn.async_handle_exception({"id": 5}, exc)