From a42209065cffe7f5dc3a63ed18314aea47726c3d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Joakim=20S=C3=B8rensen?= Date: Wed, 28 Feb 2024 06:40:31 +0100 Subject: [PATCH] Pass kwargs to orig_async_create_task in test (#3529) --- tests/common.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/common.py b/tests/common.py index 7aabd5587e..35148ec93e 100644 --- a/tests/common.py +++ b/tests/common.py @@ -124,14 +124,14 @@ def async_add_executor_job(target, *args): return orig_async_add_executor_job(target, *args) - def async_create_task(coroutine, *args): + def async_create_task(coroutine, *args, **kwargs): """Create task.""" if isinstance(coroutine, Mock) and not isinstance(coroutine, AsyncMock): fut = asyncio.Future() fut.set_result(None) return fut - return orig_async_create_task(coroutine, *args) + return orig_async_create_task(coroutine, *args, **kwargs) hass.async_add_job = async_add_job hass.async_add_executor_job = async_add_executor_job