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

Use asyncio.timeout [core] (#98447)

Browse source
This commit is contained in:
Marc Mueller 2023-08-15 15:36:05 +02:00 • committed by GitHub
parent e2d2ec8817
commit a9ade1f84d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
27 changed files with 77 additions and 96 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/components/websocket_api/test_commands.py
View file

@ -1,9 +1,9 @@
"""Tests for WebSocket API commands."""
import asyncio
from copy import deepcopy
import datetime
from unittest.mock import ANY, AsyncMock, Mock, patch
from async_timeout import timeout
import pytest
import voluptuous as vol
@ -497,7 +497,7 @@ async def test_subscribe_unsubscribe_events(
hass.bus.async_fire("test_event", {"hello": "world"})
hass.bus.async_fire("ignore_event")
async with timeout(3):
async with asyncio.timeout(3):
msg = await websocket_client.receive_json()
assert msg["id"] == 5
@ -712,7 +712,7 @@ async def test_subscribe_unsubscribe_events_whitelist(
hass.bus.async_fire("themes_updated")
async with timeout(3):
async with asyncio.timeout(3):
msg = await websocket_client.receive_json()
assert msg["id"] == 6
@ -1611,7 +1611,7 @@ async def test_subscribe_trigger(hass: HomeAssistant, websocket_client) -> None:
hass.bus.async_fire("test_event", {"hello": "world"}, context=context)
hass.bus.async_fire("ignore_event")
async with timeout(3):
async with asyncio.timeout(3):
msg = await websocket_client.receive_json()
assert msg["id"] == 5