mirror of
https://github.com/home-assistant/core.git
synced 2025-12-24 21:06:19 +00:00
Use asyncio.timeout [core] (#98447)
This commit is contained in:
@@ -14,7 +14,6 @@ import time
|
||||
from typing import Any
|
||||
from unittest.mock import MagicMock, Mock, PropertyMock, patch
|
||||
|
||||
import async_timeout
|
||||
import pytest
|
||||
import voluptuous as vol
|
||||
|
||||
@@ -235,7 +234,7 @@ async def test_async_get_hass_can_be_called(hass: HomeAssistant) -> None:
|
||||
assert can_call_async_get_hass()
|
||||
|
||||
hass.async_create_task(_async_create_task(), "create_task")
|
||||
async with async_timeout.timeout(1):
|
||||
async with asyncio.timeout(1):
|
||||
await task_finished.wait()
|
||||
task_finished.clear()
|
||||
|
||||
@@ -246,7 +245,7 @@ async def test_async_get_hass_can_be_called(hass: HomeAssistant) -> None:
|
||||
task_finished.set()
|
||||
|
||||
hass.async_add_job(_add_job)
|
||||
async with async_timeout.timeout(1):
|
||||
async with asyncio.timeout(1):
|
||||
await task_finished.wait()
|
||||
task_finished.clear()
|
||||
|
||||
@@ -263,7 +262,7 @@ async def test_async_get_hass_can_be_called(hass: HomeAssistant) -> None:
|
||||
hass.async_add_job(_callback)
|
||||
|
||||
_schedule_callback_from_callback()
|
||||
async with async_timeout.timeout(1):
|
||||
async with asyncio.timeout(1):
|
||||
await task_finished.wait()
|
||||
task_finished.clear()
|
||||
|
||||
@@ -279,7 +278,7 @@ async def test_async_get_hass_can_be_called(hass: HomeAssistant) -> None:
|
||||
hass.async_add_job(_coroutine())
|
||||
|
||||
_schedule_coroutine_from_callback()
|
||||
async with async_timeout.timeout(1):
|
||||
async with asyncio.timeout(1):
|
||||
await task_finished.wait()
|
||||
task_finished.clear()
|
||||
|
||||
@@ -295,7 +294,7 @@ async def test_async_get_hass_can_be_called(hass: HomeAssistant) -> None:
|
||||
hass.async_add_job(_callback)
|
||||
|
||||
await _schedule_callback_from_coroutine()
|
||||
async with async_timeout.timeout(1):
|
||||
async with asyncio.timeout(1):
|
||||
await task_finished.wait()
|
||||
task_finished.clear()
|
||||
|
||||
@@ -310,7 +309,7 @@ async def test_async_get_hass_can_be_called(hass: HomeAssistant) -> None:
|
||||
await hass.async_create_task(_coroutine())
|
||||
|
||||
await _schedule_callback_from_coroutine()
|
||||
async with async_timeout.timeout(1):
|
||||
async with asyncio.timeout(1):
|
||||
await task_finished.wait()
|
||||
task_finished.clear()
|
||||
|
||||
@@ -326,7 +325,7 @@ async def test_async_get_hass_can_be_called(hass: HomeAssistant) -> None:
|
||||
hass.add_job(_async_add_job)
|
||||
|
||||
await hass.async_add_executor_job(_async_add_executor_job_add_job)
|
||||
async with async_timeout.timeout(1):
|
||||
async with asyncio.timeout(1):
|
||||
await task_finished.wait()
|
||||
task_finished.clear()
|
||||
|
||||
@@ -341,7 +340,7 @@ async def test_async_get_hass_can_be_called(hass: HomeAssistant) -> None:
|
||||
hass.create_task(_async_create_task())
|
||||
|
||||
await hass.async_add_executor_job(_async_add_executor_job_create_task)
|
||||
async with async_timeout.timeout(1):
|
||||
async with asyncio.timeout(1):
|
||||
await task_finished.wait()
|
||||
task_finished.clear()
|
||||
|
||||
@@ -359,7 +358,7 @@ async def test_async_get_hass_can_be_called(hass: HomeAssistant) -> None:
|
||||
|
||||
my_job_add_job = MyJobAddJob()
|
||||
my_job_add_job.start()
|
||||
async with async_timeout.timeout(1):
|
||||
async with asyncio.timeout(1):
|
||||
await task_finished.wait()
|
||||
task_finished.clear()
|
||||
my_job_add_job.join()
|
||||
@@ -377,7 +376,7 @@ async def test_async_get_hass_can_be_called(hass: HomeAssistant) -> None:
|
||||
|
||||
my_job_create_task = MyJobCreateTask()
|
||||
my_job_create_task.start()
|
||||
async with async_timeout.timeout(1):
|
||||
async with asyncio.timeout(1):
|
||||
await task_finished.wait()
|
||||
task_finished.clear()
|
||||
my_job_create_task.join()
|
||||
|
||||
Reference in New Issue
Block a user