1
0
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:
Marc Mueller
2023-08-15 15:36:05 +02:00
committed by GitHub
parent e2d2ec8817
commit a9ade1f84d
27 changed files with 77 additions and 96 deletions

View File

@@ -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()